File size: 957 Bytes
89784d2
 
 
cb3bc65
89784d2
 
 
 
 
 
c363697
 
 
 
 
89784d2
c363697
 
 
 
 
 
 
5e2c798
89784d2
c363697
 
 
 
 
5e2c798
 
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
# 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

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

demo = gr.Interface(
    fn=chat,
    inputs=gr.Textbox(label="Ask a Question"),
    outputs=gr.JSON(label="Responses from Agents"),
    title="Multi-Agent AI Chatroom πŸ€–",
)

if __name__ == "__main__":
    demo.launch()