add UI support for underlying flat file table

app.py

@@ -3,10 +3,35 @@ import gradio as gr
 from sqlalchemy import text
 from smolagents import tool, CodeAgent, HfApiModel
 import spaces
+import pandas as pd
 
 # Import the persistent database
 from database import engine, receipts
 
+def get_receipts_table():
+    """
+    Fetches all data from the 'receipts' table and returns it as a Pandas DataFrame.
+
+    Returns:
+        A Pandas DataFrame containing all receipt data.
+    """
+    try:
+        with engine.connect() as con:
+            result = con.execute(text("SELECT * FROM receipts"))
+            rows = result.fetchall()
+        
+        if not rows:
+            return pd.DataFrame(columns=["receipt_id", "customer_name", "price", "tip"])
+
+        # Convert rows into a DataFrame
+        df = pd.DataFrame(rows, columns=["receipt_id", "customer_name", "price", "tip"])
+
+        return df
+
+    except Exception as e:
+        return pd.DataFrame({"Error": [str(e)]})  # Return error message in DataFrame format
+
+
 @tool
 def sql_engine(query: str) -> str:
     """
@@ -97,14 +122,25 @@ agent = CodeAgent(
     model=HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct"),
 )
 
-# Define Gradio interface using handle_query instead of query_sql
-demo = gr.Interface(
-    fn=handle_query,  # Call handle_query to return the final SQL output
-    inputs=gr.Textbox(label="Enter your query in plain English"),
-    outputs=gr.Textbox(label="Query Result"),
-    title="Natural Language to SQL Executor",
-    description="Enter a plain English request, and the AI will generate an SQL query and return the results.",
-    flagging_mode="never",
+demo = gr.TabbedInterface(
+    [
+        gr.Interface(
+            fn=handle_query,
+            inputs=gr.Textbox(label="Enter your query in plain English"),
+            outputs=gr.Textbox(label="Query Result"),
+            title="Natural Language to SQL Executor",
+            description="Enter a plain English request, and the AI will generate an SQL query and return the results.",
+            flagging_mode="never",
+        ),
+        gr.Interface(
+            fn=get_receipts_table,
+            inputs=[],
+            outputs=gr.Dataframe(label="Receipts Table"),
+            title="Database Viewer",
+            description="This table shows all receipts stored in the database.",
+        ),
+    ],
+    tab_names=["SQL Query", "Receipts Table"],
 )
 
 if __name__ == "__main__":