Update app.py

app.py

@@ -3,11 +3,11 @@ import gradio as gr
 from sqlalchemy import create_engine, MetaData, Table, Column, String, Integer, Float, insert, text, inspect
 from smolagents import tool, CodeAgent, InferenceClientModel
 
-# --- Setup SQLite database (persistent file) ---
+# --- Setup SQLite database (persistent in Space) ---
 engine = create_engine("sqlite:///data.db")
 metadata_obj = MetaData()
 
-# Create receipts table
+# --- Create receipts table ---
 receipts = Table(
     "receipts",
     metadata_obj,
@@ -30,7 +30,7 @@ for row in rows:
     with engine.begin() as conn:
         conn.execute(stmt)
 
-# Create waiters table
+# --- Create waiters table ---
 waiters = Table(
     "waiters",
     metadata_obj,
@@ -52,7 +52,7 @@ for row in rows:
 
 # --- Define SQL tool for the agent ---
 @tool
-def sql_engine(query: str) -> str:
+def sql_engine(query: str) -> list:
     """
     Executes SQL queries on the available tables: receipts and waiters.
 
@@ -60,18 +60,16 @@ def sql_engine(query: str) -> str:
         query: SQL query string.
 
     Returns:
-        A string representing the query results, or an error message.
+        List of tuples with query results.
     """
     try:
-        print(f"🧩 Executing query: {query}")  # Debug log in Space console
+        print(f"🧩 Executing query: {query}")  # Debug log
         with engine.connect() as con:
             rows = con.execute(text(query))
-            results = [tuple(row) for row in rows]
-        if not results:
-            return "No results."
-        return "\n".join(str(r) for r in results)
+            results = [tuple(row) for row in rows]  # Keep results as tuples
+        return results or []
     except Exception as e:
-        return f"⚠️ SQL Error: {str(e)}"
+        return [f"⚠️ SQL Error: {str(e)}"]
 
 # Dynamically describe tables
 updated_description = "Allows SQL queries on the following tables:\n"
@@ -85,20 +83,26 @@ for table in ["receipts", "waiters"]:
 sql_engine.description = updated_description
 
 # --- Create the agent ---
-# Use the HF_TOKEN secret stored in the Space
 agent = CodeAgent(
     tools=[sql_engine],
     model=InferenceClientModel(
         "meta-llama/Meta-Llama-3-8B-Instruct",
-        api_key=os.environ.get("HF_TOKEN")  # 👈 Hugging Face secret
+        api_key=os.environ.get("HF_TOKEN")  # Use secret from Space
     ),
 )
 
 # --- Define Gradio interface ---
 def ask_agent(question: str) -> str:
-    """Ask the AI agent a question."""
-    result = agent.run(question)
-    return result
+    """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."
+        return str(result)
+    except Exception as e:
+        return f"⚠️ Error: {str(e)}"
 
 demo = gr.Interface(
     fn=ask_agent,