Update app.py

app.py

@@ -7,61 +7,56 @@ from smolagents import tool, CodeAgent, InferenceClientModel
 engine = create_engine("sqlite:///data.db")
 metadata_obj = MetaData()
 
-# --- Create receipts table ---
-receipts = Table(
-    "receipts",
+# --- Create students table ---
+students = Table(
+    "students",
     metadata_obj,
-    Column("receipt_id", Integer, primary_key=True),
-    Column("customer_name", String(16), primary_key=True),
-    Column("price", Float),
-    Column("tip", Float),
+    Column("student_id", Integer, primary_key=True),
+    Column("student_name", String(32), primary_key=True),
 )
 metadata_obj.create_all(engine)
 
-# Insert sample data
-rows = [
-    {"receipt_id": 1, "customer_name": "Alan Payne", "price": 12.06, "tip": 1.20},
-    {"receipt_id": 2, "customer_name": "Alex Mason", "price": 23.86, "tip": 0.24},
-    {"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},
+# Insert sample student data
+student_rows = [
+    {"student_id": 1, "student_name": "Alice Johnson"},
+    {"student_id": 2, "student_name": "Bob Smith"},
+    {"student_id": 3, "student_name": "Charlie Brown"},
+    {"student_id": 4, "student_name": "Diana Prince"},
 ]
-for row in rows:
-    stmt = insert(receipts).values(**row)
+for row in student_rows:
+    stmt = insert(students).values(**row)
     with engine.begin() as conn:
         conn.execute(stmt)
 
-# --- Create waiters table ---
-waiters = Table(
-    "waiters",
+# --- Create subjects table ---
+subjects = Table(
+    "subjects",
     metadata_obj,
-    Column("receipt_id", Integer, primary_key=True),
-    Column("waiter_name", String(16), primary_key=True),
+    Column("student_id", Integer, primary_key=True),
+    Column("subject_name", String(32), primary_key=True),
+    Column("marks", Float),
 )
 metadata_obj.create_all(engine)
 
-rows = [
-    {"receipt_id": 1, "waiter_name": "Corey Johnson"},
-    {"receipt_id": 2, "waiter_name": "Michael Watts"},
-    {"receipt_id": 3, "waiter_name": "Michael Watts"},
-    {"receipt_id": 4, "waiter_name": "Margaret James"},
+# Insert sample subject/marks data
+subject_rows = [
+    {"student_id": 1, "subject_name": "Math", "marks": 95.0},
+    {"student_id": 1, "subject_name": "English", "marks": 88.5},
+    {"student_id": 2, "subject_name": "Math", "marks": 72.0},
+    {"student_id": 2, "subject_name": "Science", "marks": 81.0},
+    {"student_id": 3, "subject_name": "Math", "marks": 85.0},
+    {"student_id": 3, "subject_name": "History", "marks": 90.0},
+    {"student_id": 4, "subject_name": "English", "marks": 78.0},
+    {"student_id": 4, "subject_name": "Science", "marks": 82.0},
 ]
-for row in rows:
-    stmt = insert(waiters).values(**row)
+for row in subject_rows:
+    stmt = insert(subjects).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.
-
-    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:
@@ -74,7 +69,7 @@ def sql_engine(query: str) -> list:
 # Dynamically describe tables
 updated_description = "Allows SQL queries on the following tables:\n"
 inspector = inspect(engine)
-for table in ["receipts", "waiters"]:
+for table in ["students", "subjects"]:
     columns_info = [(col["name"], col["type"]) for col in inspector.get_columns(table)]
     table_description = f"\n\nTable '{table}':\nColumns:\n" + "\n".join(
         [f" - {name}: {col_type}" for name, col_type in columns_info]
@@ -107,10 +102,10 @@ def ask_agent(question: str) -> str:
 
 demo = gr.Interface(
     fn=ask_agent,
-    inputs=gr.Textbox(label="Ask a question (e.g., Who got the biggest tip?)"),
+    inputs=gr.Textbox(label="Ask a question (e.g., Who scored highest in Math?)"),
     outputs=gr.Textbox(label="Agent response"),
     title="🧠 Text-to-SQL Agent",
-    description="Ask natural-language questions about a restaurant receipts database. Powered by smolagents + Meta-Llama-3.",
+    description="Ask natural-language questions about students, subjects, and marks. Powered by smolagents + Meta-Llama-3.",
 )
 
 demo.launch()