Mustehson commited on
Commit
ece17f4
·
1 Parent(s): 2e06cf4

Revert "Summary Added"

Browse files

This reverts commit 7c2e7ac9fa4f44d4d37506cf2e17631d8293f85d.

Back to PP

Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -88,6 +88,7 @@ def get_data_df(schema):
88
  print('Getting Dataframe from the Database')
89
  return conn.sql(f"SELECT * FROM {schema} LIMIT 1000").df()
90
 
 
91
  def calcualte_stats(df):
92
  indev_stats = []
93
  cols = []
@@ -146,11 +147,20 @@ def df_summary(df):
146
 
147
  summary_df = pd.DataFrame(summary)
148
  return summary_df.reset_index(drop=True)
 
 
149
 
150
  def format_prompt(df):
151
- summary = df_summary(df)
 
 
 
 
 
 
 
152
  return prompt_autogenerate.format_prompt(data=df.head().to_json(orient='records'),
153
- summary=summary.to_json(orient='records'))
154
  def format_user_prompt(df):
155
  return prompt_user_input.format_prompt(data=df.head().to_json(orient='records'))
156
 
 
88
  print('Getting Dataframe from the Database')
89
  return conn.sql(f"SELECT * FROM {schema} LIMIT 1000").df()
90
 
91
+ <<<<<<< HEAD
92
  def calcualte_stats(df):
93
  indev_stats = []
94
  cols = []
 
147
 
148
  summary_df = pd.DataFrame(summary)
149
  return summary_df.reset_index(drop=True)
150
+ =======
151
+ >>>>>>> parent of 7c2e7ac (Summary Added)
152
 
153
  def format_prompt(df):
154
+ summary_df = pd.DataFrame({
155
+ "max": df.max(),
156
+ "min": df.min(),
157
+ "top": df.mode().iloc[0],
158
+ "nunique": df.nunique(),
159
+ "count": df.count(),
160
+ "dtype": df.dtypes.astype(str)
161
+ }).reset_index().rename(columns={"index": "column"})
162
  return prompt_autogenerate.format_prompt(data=df.head().to_json(orient='records'),
163
+ summary=summary_df.to_json(orient='records'))
164
  def format_user_prompt(df):
165
  return prompt_user_input.format_prompt(data=df.head().to_json(orient='records'))
166