Update app.py

app.py

@@ -345,17 +345,28 @@ graph.add_edge("Query Optimization", "SQL Execution")
 compiled_pipeline = graph.compile()
 
 # ------------------ Gradio Interface ------------------
-def run_multi_agent_query(natural_language_query):
-    result = compiled_pipeline.invoke({"sql_query": natural_language_query})
-    return json.dumps(result.get("execution_result", {}), indent=2)
+schema_description = """
+**Database Schema:**
+
+- **customers**: customer_id, name, email, phone, address, created_at  
+- **products**: product_id, name, category, price, stock_quantity  
+- **orders**: order_id, customer_id, order_date, total_amount, status  
+- **order_items**: order_item_id, order_id, product_id, quantity, subtotal  
+- **payments**: payment_id, order_id, payment_date, amount, payment_method, status  
+- **shipment**: shipment_id, order_id, shipment_date, delivery_date, carrier, tracking_number, status  
+"""
 
 iface = gr.Interface(
     fn=run_multi_agent_query,
     inputs=gr.Textbox(lines=2, placeholder="Enter your natural language query here..."),
     outputs="text",
     title="Multi-Agent SQL Generator",
-    description="Enter a natural language query to generate and execute a SQL query."
+    description="Enter a natural language query to generate and execute a SQL query.\n\n" + schema_description
 )
 
+def run_multi_agent_query(natural_language_query):
+    result = compiled_pipeline.invoke({"sql_query": natural_language_query})
+    return json.dumps(result.get("execution_result", {}), indent=2)
+
 if __name__ == "__main__":
     iface.launch()