Update app.py

app.py

@@ -7,6 +7,7 @@ from datetime import timedelta
 from faker import Faker
 from langgraph.graph import StateGraph, START
 from typing import TypedDict, Optional
+import gradio as gr
 
 # Set your OpenAI API key from environment variable
 openai.api_key = os.getenv("OPENAI_API_KEY")
@@ -343,11 +344,18 @@ graph.add_edge("Query Optimization", "SQL Execution")
 
 compiled_pipeline = graph.compile()
 
-# ------------------ Example Execution ------------------
+# ------------------ 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)
 
+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."
+)
+
 if __name__ == "__main__":
-    # For testing outside of Hugging Face Spaces
-    print(run_multi_agent_query("Find the email_id of the top 5 customers who spent the most in 2024."))
+    iface.launch()