Update app.py

app.py

@@ -1,8 +1,6 @@
 import os
 import gradio as gr
-from sqlalchemy import (
-    create_engine, MetaData, Table, Column, String, Integer, Float, insert, text, inspect
-)
+from sqlalchemy import create_engine, MetaData, Table, Column, String, Integer, Float, insert, text, inspect
 from smolagents import tool, CodeAgent, InferenceClientModel
 
 # --- Setup SQLite database (persistent in Space) ---
@@ -27,9 +25,9 @@ rows = [
     {"receipt_id": 3, "customer_name": "Woodrow Wilson", "price": 53.43, "tip": 5.43},
     {"receipt_id": 4, "customer_name": "Margaret James", "price": 21.11, "tip": 1.00},
 ]
-with engine.begin() as conn:
-    for row in rows:
-        stmt = insert(receipts).values(**row)
+for row in rows:
+    stmt = insert(receipts).values(**row)
+    with engine.begin() as conn:
         conn.execute(stmt)
 
 # --- Create waiters table ---
@@ -47,20 +45,28 @@ rows = [
     {"receipt_id": 3, "waiter_name": "Michael Watts"},
     {"receipt_id": 4, "waiter_name": "Margaret James"},
 ]
-with engine.begin() as conn:
-    for row in rows:
-        stmt = insert(waiters).values(**row)
+for row in rows:
+    stmt = insert(waiters).values(**row)
+    with engine.begin() as conn:
         conn.execute(stmt)
 
 # --- Define SQL tool for the agent ---
 @tool
 def sql_engine(query: str) -> list:
-    """Executes SQL queries on the available tables: receipts and waiters."""
+    """
+    Executes SQL queries on the available tables: receipts and waiters.
+
+    Args:
+        query: SQL query string.
+
+    Returns:
+        List of tuples with query results.
+    """
     try:
         print(f"🧩 Executing query: {query}")  # Debug log
         with engine.connect() as con:
             rows = con.execute(text(query))
-            results = [tuple(row) for row in rows]
+            results = [tuple(row) for row in rows]  # Keep results as tuples
         return results or []
     except Exception as e:
         return [f"⚠️ SQL Error: {str(e)}"]
@@ -74,6 +80,7 @@ for table in ["receipts", "waiters"]:
         [f" - {name}: {col_type}" for name, col_type in columns_info]
     )
     updated_description += table_description
+
 sql_engine.description = updated_description
 
 # --- Create the agent ---
@@ -90,6 +97,7 @@ def ask_agent(question: str) -> str:
     """Ask the AI agent a question and return its answer."""
     try:
         result = agent.run(question)
+        # Convert tuples to readable string if result is a list of tuples
         if isinstance(result, list):
             result_str = "\n".join(str(r) for r in result)
             return result_str or "No results."