Update app.py

app.py

@@ -1,35 +1,40 @@
 # app.py
 
+from fastapi import FastAPI
 import gradio as gr
-# from inference import deepseek_query
-multi_inference import multi_query as deepseek_query in app.py
+from multi_inference import multi_query as deepseek_query  # ✅ Corrected import
 
 from agents.philosopher import PhilosopherAgent
 from agents.historian import HistorianAgent
 from agents.hacker import HackerAgent
 from agents.comedian import ComedianAgent
 
-# Load agents
-agents = [
-    PhilosopherAgent(),
-    HistorianAgent(),
-    HackerAgent(),
-    ComedianAgent()
-]
+app = FastAPI()
+
+philosopher = PhilosopherAgent()
+historian = HistorianAgent()
+hacker = HackerAgent()
+comedian = ComedianAgent()
+
 
-# Main chat function
 def chat(prompt):
-    responses = []
-    for agent in agents:
-        reply = agent.generate_response(prompt, deepseek_query)
-        responses.append(f"{agent.name}: {reply.strip()}")
-    return "\n\n".join(responses)
-
-# Gradio UI
-gr.Interface(
+    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)
+    return responses
+
+
+interface = gr.Interface(
     fn=chat,
-    inputs="text",
-    outputs="text",
-    title="🧠 Multi-Agent AI Chatroom",
-    description="Talk to a Philosopher, Historian, Hacker, and Comedian — all at once!"
-).launch()
+    inputs=gr.Textbox(label="Ask a Question"),
+    outputs=gr.JSON(label="Responses from Agents"),
+    title="Multi-Agent AI Chatroom 🤖",
+)
+
+@app.get("/")
+def read_root():
+    return {"message": "Welcome to the Multi-Agent AI Chatroom!"}
+
+app = gr.mount_gradio_app(app, interface, path="/chat")