Update app.py

app.py

@@ -344,6 +344,11 @@ graph.add_edge("Query Optimization", "SQL Execution")
 
 compiled_pipeline = graph.compile()
 
+# ------------------ Function to Run the Multi-Agent Query ------------------
+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)
+
 # ------------------ Gradio Interface ------------------
 schema_description = """
 **Database Schema:**
@@ -364,9 +369,5 @@ iface = gr.Interface(
     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()
+    iface.launch()