Create main.py

main.py

@@ -0,0 +1,77 @@
+# main.py
+from typing import Dict, Callable, Awaitable
+import asyncio
+import gradio as gr
+
+# We expect these callables to be injected by app.py (keeps separation/testability).
+# Signatures:
+#   run_agent(agent_id: int, message: str) -> Awaitable[str]
+#   run_allAgents(message: str) -> Awaitable[Dict[str, str]]
+
+def build_ui(
+    run_agent: Callable[[int, str], Awaitable[str]],
+    run_all_agents: Callable[[str], Awaitable[Dict[str, str]]],
+) -> gr.Blocks:
+    with gr.Blocks(title="GLM 4.5 Multi-Agent Control Panel") as demo:
+        gr.Markdown("# 🧭 GLM 4.5 Multi-Agent Control Panel")
+        gr.Markdown(
+            "Send a message to one agent or broadcast to all four. "
+            "API is available at `/api/...` endpoints."
+        )
+
+        with gr.Row():
+            agent = gr.Radio(
+                choices=["agent1", "agent2", "agent3", "agent4", "all"],
+                value="agent1",
+                label="Target",
+            )
+            msg = gr.Textbox(
+                label="Message",
+                placeholder="Type something for the agent(s)...",
+                lines=3,
+                show_label=True,
+            )
+
+        with gr.Row():
+            send = gr.Button("Send", variant="primary")
+            clear = gr.Button("Clear")
+
+        # Outputs: one for single agent, one for 'all'
+        single_out = gr.Chatbot(label="Agent Response", height=300)
+        all_out = gr.JSON(label="All Agents (JSON)")
+
+        def _reset():
+            return None, None
+
+        clear.click(_reset, outputs=[single_out, all_out])
+
+        async def handle_submit(agent_choice: str, message: str):
+            if not message.strip():
+                return gr.update(), gr.update(value={"error": "Message is empty."})
+            if agent_choice == "all":
+                data = await run_all_agents(message)
+                # Show JSON in the 'all_out' and clear the single chat.
+                return gr.update(value=[]), gr.update(value=data)
+            else:
+                agent_map = {"agent1": 1, "agent2": 2, "agent3": 3, "agent4": 4}
+                agent_id = agent_map[agent_choice]
+                resp = await run_agent(agent_id, message)
+                # Append to chatbot as (user, bot)
+                return gr.update(value=[[message, resp]]), gr.update(value=None)
+
+        send.click(
+            handle_submit,
+            inputs=[agent, msg],
+            outputs=[single_out, all_out],
+        )
+
+        gr.Markdown(
+            "### API\n"
+            "- `POST /api/agent1` `{ message: string }`\n"
+            "- `POST /api/agent2` `{ message: string }`\n"
+            "- `POST /api/agent3` `{ message: string }`\n"
+            "- `POST /api/agent4` `{ message: string }`\n"
+            "- `POST /api/run_all` `{ message: string }`"
+        )
+
+    return demo