improve table retrieval

app.py

@@ -8,6 +8,8 @@ import pandas as pd
 # Import the persistent database
 from database import engine, receipts
 
+import pandas as pd
+
 def get_receipts_table():
     """
     Fetches all data from the 'receipts' table and returns it as a Pandas DataFrame.
@@ -25,13 +27,11 @@ def get_receipts_table():
 
         # 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:
     """
@@ -122,26 +122,23 @@ agent = CodeAgent(
     model=HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct"),
 )
 
-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"],
-)
+with gr.Blocks() as demo:
+    gr.Markdown("## Natural Language to SQL Executor with Live Data")
+
+    with gr.Row():
+        with gr.Column(scale=1):  # Left: Query Interface
+            user_input = gr.Textbox(label="Enter your query in plain English")
+            query_output = gr.Textbox(label="Query Result")
+        
+        with gr.Column(scale=2):  # Right: Live Database Table
+            gr.Markdown("### Receipts Table (Live View)")
+            receipts_table = gr.Dataframe(value=get_receipts_table(), label="Receipts Table")
+
+    # Query handling function
+    user_input.change(fn=handle_query, inputs=user_input, outputs=query_output)
+
+    # Auto-refresh table every 5 seconds
+    demo.load(fn=get_receipts_table, outputs=receipts_table, every=5)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, share=True)