File size: 1,925 Bytes
89784d2
 
 
 
 
b9b41a6
 
 
 
7337f9c
89784d2
f574446
b9b41a6
7337f9c
 
 
b9b41a6
 
 
 
 
7337f9c
c363697
f574446
b9b41a6
 
f574446
deb653c
b9b41a6
c363697
b9b41a6
dc6e3d1
7337f9c
 
 
dc6e3d1
b9b41a6
deb653c
c363697
f574446
deb653c
7337f9c
89784d2
b9b41a6
f574446
 
 
deb653c
f574446
b9b41a6
 
dd95794
deb653c
b9b41a6
 
dd95794
4993125
deb653c
 
c363697
 
deb653c
 
 
 
 
 
 
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import gradio as gr
from agents.philosopher import PhilosopherAgent
from agents.historian import HistorianAgent
from agents.hacker import HackerAgent
from agents.comedian import ComedianAgent
from agents.lawyer import LawyerAgent
from agents.scientist import ScientistAgent
from agents.journalist import JournalistAgent
from agents.trader import TraderAgent
from agents.base_agent import ACPMessage

# βœ… Initialize all 8 agents
all_agents = [
    PhilosopherAgent(),
    HistorianAgent(),
    HackerAgent(),
    ComedianAgent(),
    LawyerAgent(),
    ScientistAgent(),
    JournalistAgent(),
    TraderAgent()
]

# βœ… Map name to agent
agent_map = {agent.name: agent for agent in all_agents}


# βœ… Core chat function for API call
def chat(prompt, selected_agents):
    responses = {}
    for name in selected_agents:
        agent = agent_map.get(name)
        try:
            output = agent.generate([ACPMessage(role="user", content=prompt)])
        except Exception as e:
            output = f"[ERROR: {e}]"
        responses[name] = output
    return [responses]  # βœ… Must be inside a list for Gradio JSON API


# βœ… Gradio Interface
iface = gr.Interface(
    fn=chat,
    inputs=[
        gr.Textbox(
            label="Ask Anything",
            lines=4,
            placeholder="Ask your question here..."
        ),
        gr.CheckboxGroup(
            choices=list(agent_map.keys()),
            label="Choose Which Personalities to Ask",
            value=list(agent_map.keys())  # βœ… All selected by default
        )
    ],
    outputs=gr.JSON(label="Responses"),
    title="PerspectiveAI",
    description="Ask a question, get responses from 8 unique AI personas.",
    theme=gr.themes.Soft()
)

# βœ… Launch with API mode enabled (important!)
iface.launch(
    server_name="0.0.0.0",
    server_port=7860,
    show_api=True,
    ssr=False  # 🚨 This disables SvelteKit SSR and enables API mode
)