Spaces:
Sleeping
Sleeping
Mustehson
commited on
Commit
·
ece17f4
1
Parent(s):
2e06cf4
Revert "Summary Added"
Browse filesThis reverts commit 7c2e7ac9fa4f44d4d37506cf2e17631d8293f85d.
Back to PP
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
152 |
return prompt_autogenerate.format_prompt(data=df.head().to_json(orient='records'),
|
153 |
-
summary=
|
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 |
|