Update app.py

app.py

@@ -39,8 +39,22 @@ def ask_question(user_db, question):
     cursor = conn.cursor()
 
     # Create a Text-to-SQL prompt
-    prompt = f"Translate this question into an SQLite query:\nQuestion: {question}\nSQL:"
-    sql_query = sqlcoder_llm(prompt)
+    prompt = f"Translate this question into an SQLite query.\nReturn only SQL (no text):\nQuestion: {question}\nSQL:"
+
+    # Generate text
+    response = sqlcoder_llm(prompt)
+
+    # Some models return objects, ensure it's plain string
+    if isinstance(response, dict) and "text" in response:
+        response = response["text"]
+    elif isinstance(response, list):
+        response = response[0]["generated_text"]
+
+    # Extract SQL from response (clean any extra text)
+    sql_query = response.strip().split("SQL:")[-1].strip()
+    sql_query = sql_query.split("\n")[0].strip()
+    if not sql_query.endswith(";"):
+        sql_query += ";"
 
     try:
         cursor.execute(sql_query)
@@ -51,7 +65,7 @@ def ask_question(user_db, question):
         return sql_query, df
     except Exception as e:
         conn.close()
-        return f"❌ Error executing query: {e}", None
+        return f"❌ Error executing query: {e}\n\nGenerated SQL:\n{sql_query}", None
 
 # ============================================================
 # 🎨 Gradio UI