Update app.py

app.py

@@ -1,37 +1,36 @@
 # app.py
 
 import gradio as gr
-from multi_inference import multi_query as deepseek_query
-
 from agents.philosopher import PhilosopherAgent
 from agents.historian import HistorianAgent
 from agents.hacker import HackerAgent
 from agents.comedian import ComedianAgent
+from agents.base_agent import ACPMessage
 
-# Initialize agent classes
-philosopher = PhilosopherAgent()
-historian = HistorianAgent()
-hacker = HackerAgent()
-comedian = ComedianAgent()
+# Instantiate agents
+agents = [
+    PhilosopherAgent(),
+    HistorianAgent(),
+    HackerAgent(),
+    ComedianAgent()
+]
 
-# Chat function
 def chat(prompt):
     responses = {}
-    responses["🧙‍♂️ Philosopher"] = philosopher.run(prompt, deepseek_query)
-    responses["👨‍🏫 Historian"] = historian.run(prompt, deepseek_query)
-    responses["💻 Hacker"] = hacker.run(prompt, deepseek_query)
-    responses["🎭 Comedian"] = comedian.run(prompt, deepseek_query)
+    for agent in agents:
+        try:
+            output = agent.generate([ACPMessage(role="user", content=prompt)])
+        except Exception as e:
+            output = f"[ERROR] {e}"
+        responses[agent.name] = output
     return responses
 
-# Gradio UI setup
-demo = gr.Interface(
+iface = gr.Interface(
     fn=chat,
-    inputs=gr.Textbox(label="Ask a Question"),
-    outputs=gr.JSON(label="Agent Responses"),
-    title="🧠 Multi-Agent AI Chatroom",
-    description="Ask anything. Each AI agent gives a unique answer!"
+    inputs=gr.Textbox(label="Ask Something", lines=3, placeholder="Type your message here..."),
+    outputs="json",
+    title="🤖 Multi-Agent Chatroom (Groq Edition)",
+    description="Talk to 4 AI agents at once using free Groq-powered LLaMA3 API."
 )
 
-# Only launch if run directly (used by Hugging Face too)
-if __name__ == "__main__":
-    demo.launch()
+iface.launch()