Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,14 +55,15 @@ def clean_sql_for_duckdb(sql, df_columns):
|
|
| 55 |
|
| 56 |
# π¬ Main chatbot function
|
| 57 |
def chatbot_interface(file, question):
|
| 58 |
-
|
| 59 |
df = pd.read_excel(file)
|
| 60 |
sql = generate_sql_from_prompt(question, df)
|
| 61 |
cleaned_sql = clean_sql_for_duckdb(sql, df.columns)
|
| 62 |
result = duckdb.query(cleaned_sql).to_df()
|
| 63 |
return f"π SQL Query:\n```sql\n{sql}\n```", result
|
| 64 |
except Exception as e:
|
| 65 |
-
|
|
|
|
| 66 |
|
| 67 |
|
| 68 |
# ποΈ Gradio UI
|
|
|
|
| 55 |
|
| 56 |
# π¬ Main chatbot function
|
| 57 |
def chatbot_interface(file, question):
|
| 58 |
+
try:
|
| 59 |
df = pd.read_excel(file)
|
| 60 |
sql = generate_sql_from_prompt(question, df)
|
| 61 |
cleaned_sql = clean_sql_for_duckdb(sql, df.columns)
|
| 62 |
result = duckdb.query(cleaned_sql).to_df()
|
| 63 |
return f"π SQL Query:\n```sql\n{sql}\n```", result
|
| 64 |
except Exception as e:
|
| 65 |
+
return f"β Error: {str(e)}", pd.DataFrame()
|
| 66 |
+
|
| 67 |
|
| 68 |
|
| 69 |
# ποΈ Gradio UI
|