Update app.py

app.py

@@ -1,7 +1,103 @@
+import os
 import gradio as gr
+from zhipuai import ZhipuAI
 
-def greet(name):
-    return "Hello " + name + "!!"
+# === GLM setup ===
+api_key = os.getenv("ZHIPUAI_API_KEY")
+if not api_key:
+    print("❌ [Startup Error] Missing ZHIPUAI_API_KEY in environment variables.")
+else:
+    print("✅ [Startup] ZHIPUAI_API_KEY loaded successfully.")
+
+zhipu_client = ZhipuAI(api_key=api_key) if api_key else None
+
+# === Helper: call GLM ===
+def glm_call(prompt: str) -> str:
+    if not zhipu_client:
+        return "[GLM Error] API key not found. Please set ZHIPUAI_API_KEY."
+    try:
+        print(f"📡 Sending prompt to GLM: {prompt}")
+        resp = zhipu_client.chat.completions.create(
+            model="glm-4-0520",
+            messages=[{"role": "user", "content": prompt}]
+        )
+        reply = resp.choices[0].message["content"]
+        print(f"🤖 GLM replied: {reply}")
+        return reply
+    except Exception as e:
+        return f"[GLM Error] {e}"
+
+
+# === Agent Implementations ===
+
+# Autogen Agent (wired to GLM)
+def autogen_agent(prompt: str) -> str:
+    return f"[Autogen Agent] {glm_call(prompt)}"
+
+# LangGraph Agent (wired to GLM)
+def langgraph_agent(prompt: str) -> str:
+    return f"[LangGraph Agent] {glm_call(prompt)}"
+
+# CrewAI Agent (stub for now)
+def crewai_agent(prompt: str) -> str:
+    return "[CrewAI Agent] ❌ not connected to GLM (stubbed for now)"
+
+# OpenDevin Agent (stub for now)
+def opendevin_agent(prompt: str) -> str:
+    return "[OpenDevin Agent] ❌ not connected to GLM (stubbed for now)"
+
+
+# === Gradio Handlers ===
+
+def single_agent(agent: str, prompt: str) -> str:
+    if agent == "Autogen":
+        return autogen_agent(prompt)
+    elif agent == "LangGraph":
+        return langgraph_agent(prompt)
+    elif agent == "CrewAI":
+        return crewai_agent(prompt)
+    elif agent == "OpenDevin":
+        return opendevin_agent(prompt)
+    else:
+        return "[Error] Unknown agent selected."
+
+
+def all_agents(prompt: str) -> str:
+    outputs = []
+    outputs.append(autogen_agent(prompt))
+    outputs.append(langgraph_agent(prompt))
+    outputs.append(crewai_agent(prompt))
+    outputs.append(opendevin_agent(prompt))
+    return "\n\n".join(outputs)
+
+
+# === Gradio UI ===
+def build_ui():
+    with gr.Blocks() as demo:
+        gr.Markdown("## 🤖 Multi-Agent GLM Playground")
+
+        with gr.Tab("Single Agent"):
+            agent = gr.Dropdown(
+                ["Autogen", "LangGraph", "CrewAI", "OpenDevin"],
+                label="Choose an agent"
+            )
+            prompt = gr.Textbox(label="Your prompt")
+            output = gr.Textbox(label="Agent response")
+            run_btn = gr.Button("Run")
+
+            run_btn.click(fn=single_agent, inputs=[agent, prompt], outputs=output)
+
+        with gr.Tab("All Agents"):
+            prompt_all = gr.Textbox(label="Your prompt")
+            output_all = gr.Textbox(label="Responses")
+            run_all_btn = gr.Button("Run All")
+
+            run_all_btn.click(fn=all_agents, inputs=prompt_all, outputs=output_all)
+
+    return demo
+
+
+if __name__ == "__main__":
+    demo = build_ui()
+    demo.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", 7860)))
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="Have A Good Day")
-demo.launch()