Update main.py

main.py

@@ -1,80 +1,39 @@
-# main.py
-from typing import Dict, Callable, Awaitable
 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")
-
-        # ✅ Use OpenAI-style messages format
-        single_out = gr.Chatbot(label="Agent Response", height=300, type="messages")
-        all_out = gr.JSON(label="All Agents (JSON)")
-
-        def _reset():
-            return [], 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 using dict format
-                return gr.update(value=[
-                    {"role": "user", "content": message},
-                    {"role": "assistant", "content": 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 }`"
-        )
+import httpx
+
+def query_agent(agent: str, message: str):
+    url = f"http://localhost:7860/api/{agent}"  # works locally
+    try:
+        r = httpx.post(url, json={"text": message})
+        return r.json().get("response", "No response")
+    except Exception as e:
+        return str(e)
+
+def query_all(message: str):
+    url = "http://localhost:7860/api/run_all"
+    try:
+        r = httpx.post(url, json={"text": message})
+        return [resp["response"] for resp in r.json()["responses"]]
+    except Exception as e:
+        return [str(e)]
+
+def build_ui():
+    with gr.Blocks() as demo:
+        gr.Markdown("## Multi-Agent Control Panel")
+
+        with gr.Tab("Single Agent"):
+            msg = gr.Textbox(label="Message")
+            agent = gr.Dropdown(["agent1","agent2","agent3","agent4"], label="Choose Agent")
+            output = gr.Textbox(label="Agent Response")
+            btn = gr.Button("Send")
+            btn.click(fn=lambda m,a: query_agent(a,m), inputs=[msg,agent], outputs=output)
+
+        with gr.Tab("All Agents"):
+            msg_all = gr.Textbox(label="Message for All Agents")
+            outputs_all = gr.JSON(label="All Responses")
+            btn_all = gr.Button("Broadcast")
+            btn_all.click(fn=query_all, inputs=msg_all, outputs=outputs_all)
 
     return demo
 
+