Spaces:
Running
Running
File size: 3,241 Bytes
db8fa4c ac9bc08 db8fa4c 189fa3e db8fa4c 3f23479 db8fa4c 6bab492 db8fa4c 01927e5 |
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 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 |
import gradio as gr
from poe_api_wrapper import PoeApi
from typing import Generator
def init_client(p_b: str, p_lat: str) -> PoeApi:
return PoeApi(token={"p-b": p_b, "p-lat": p_lat})
custom_css = """
#component-0 {
background: linear-gradient(135deg, #8B5CF6 0%, #FCD34D 100%);
padding: 20px;
border-radius: 15px;
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
}
.gradio-container {
background: linear-gradient(135deg, #8B5CF6 0%, #FCD34D 100%);
}
.chat-message {
padding: 15px;
margin: 5px;
border-radius: 10px;
background-color: rgba(255, 255, 255, 0.9);
}
.assistant-message {
border-left: 4px solid #8B5CF6;
}
.user-message {
border-left: 4px solid #FCD34D;
}
"""
def chat_stream(client: PoeApi, message: str, bot: str, history: list) -> Generator:
response = ""
for chunk in client.send_message(bot, message):
response += chunk["response"]
yield response
def create_app():
with gr.Blocks(css=custom_css) as app:
gr.Markdown("# 🤖 Poe Chat Interface")
with gr.Row():
with gr.Column():
p_b = gr.Textbox(label="P-B Token", type="password")
p_lat = gr.Textbox(label="P-LAT Token", type="password")
bot_select = gr.Dropdown(
choices=["Claude-3-Opus", "Claude-3-Sonnet", "GPT-4", "GPT-3.5-Turbo", "Claude-2", "Google-PaLM"],
label="Select Bot",
value="Claude-3.7-Sonnet"
)
connect_btn = gr.Button("Connect", variant="primary")
chatbot = gr.Chatbot(
[],
elem_classes=["chat-message", "assistant-message", "user-message"],
height=500
)
msg = gr.Textbox(
label="Type your message here...",
placeholder="Enter your message and press enter...",
lines=3
)
clear = gr.Button("Clear Chat")
client = {"instance": None}
def connect(p_b_val: str, p_lat_val: str) -> str:
try:
client["instance"] = init_client(p_b_val, p_lat_val)
return "✅ Successfully connected!"
except Exception as e:
# Log the error for debugging
print(f"Connection error: {e}")
return f"❌ Connection failed: {str(e)}"
def respond(message: str, chat_history: list) -> tuple:
if not client["instance"]:
return chat_history + [[message, "Please connect to Poe first!"]]
bot = bot_select.value.lower().replace("-", "").replace(" ", "")
history = chat_history + [[message, ""]]
for response in chat_stream(client["instance"], message, bot, history):
history[-1][1] = response
yield history
def clear_chat() -> tuple:
return [], []
connect_btn.click(connect, [p_b, p_lat], gr.Textbox(label="Connection Status"))
msg.submit(respond, [msg, chatbot], [chatbot])
clear.click(clear_chat, None, [chatbot, msg])
return app
demo = create_app()
if __name__ == "__main__":
demo.launch()
else:
app = create_app()
|