Update app.py

app.py

@@ -10,17 +10,6 @@ def run_server(app_path: str, port: int):
     """Helper function to run a uvicorn server."""
     uvicorn.run(app_path, host="127.0.0.1", port=port, log_level="info")
 
-# --- Launch background MCP servers ---
-# This code runs once when the Gradio app starts.
-servers_to_run = {
-    "mock_mcp_servers.github_server:app": 8001,
-    "mock_mcp_servers.sandbox_server:app": 8002,
-}
-for app_path, port in servers_to_run.items():
-    thread = threading.Thread(target=run_server, args=(app_path, port), daemon=True)
-    thread.start()
-time.sleep(2) # Give servers a moment to start
-
 async def run_forge_agent(goal: str, hf_token: str, progress=gr.Progress(track_tqdm=True)):
     """
     The main function to be called by the Gradio interface.
@@ -80,4 +69,15 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     run_button.click(fn=run_forge_agent, inputs=[goal_input, hf_token_input], outputs=[chatbot])
 
 if __name__ == "__main__":
+    # --- Launch background MCP servers ---
+    # This code runs once when the Gradio app starts.
+    servers_to_run = {
+        "mock_mcp_servers.github_server:app": 8001,
+        "mock_mcp_servers.sandbox_server:app": 8002,
+    }
+    for app_path, port in servers_to_run.items():
+        thread = threading.Thread(target=run_server, args=(app_path, port), daemon=True)
+        thread.start()
+    time.sleep(2) # Give servers a moment to start
+
     demo.launch()