File size: 1,142 Bytes
89784d2 c363697 89784d2 c363697 89784d2 c363697 89784d2 c363697 89784d2 c363697 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
# app.py
from fastapi import FastAPI
import gradio as gr
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
app = FastAPI()
philosopher = PhilosopherAgent()
historian = HistorianAgent()
hacker = HackerAgent()
comedian = ComedianAgent()
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)
return responses
interface = gr.Interface(
fn=chat,
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")
|