Spaces:
Sleeping
Sleeping
File size: 4,337 Bytes
c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 14ca1f9 c8a3e31 | 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 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 | import os
import requests
import gradio as gr
# Groq API Configuration
groq_api_key = os.getenv("GROQ_API_KEY")
url = "https://api.groq.com/openai/v1/chat/completions"
headers = {
"Authorization": f"Bearer {groq_api_key}",
"Content-Type": "application/json"
}
# System role for the assistant
system_role = "You are a Product Manager and you will answer questions as a Product Manager."
# Simulated user database
user_db = {"admin@example.com": "password123"}
# Authentication logic
def authenticate(email, password, mode):
if mode == "Login":
if email in user_db and user_db[email] == password:
return gr.update(visible=True), gr.update(visible=False), "", "✅ Login successful"
else:
return gr.update(visible=False), gr.update(visible=True), "❌ Invalid email or password", ""
elif mode == "Sign Up":
if email in user_db:
return gr.update(visible=False), gr.update(visible=True), "❌ Email already exists", ""
user_db[email] = password
return gr.update(visible=True), gr.update(visible=False), "", "✅ Account created"
# Chat logic using OpenAI-style message format
def chat(user_input, history):
if not history:
history = []
messages = [{"role": "system", "content": system_role}] + history
messages.append({"role": "user", "content": user_input})
body = {
"model": "meta-llama/llama-4-maverick-17b-128e-instruct",
"messages": messages
}
response = requests.post(url, headers=headers, json=body)
if response.status_code == 200:
reply = response.json()['choices'][0]['message']['content']
history.append({"role": "user", "content": user_input})
history.append({"role": "assistant", "content": reply})
return history, history
else:
error = f"❌ Error: {response.status_code} - {response.text}"
history.append({"role": "user", "content": user_input})
history.append({"role": "assistant", "content": error})
return history, history
# Reset chat history
def reset():
return [], []
# Toggle password visibility
def toggle_password(view):
return gr.update(type="text" if view else "password")
# Gradio UI
with gr.Blocks(css="""
#chat-area {
height: 500px;
overflow-y: auto;
background: #F9FAFB;
border: 1px solid #D1D5DB;
border-radius: 8px;
padding: 10px;
}
#send-chat, #reset-chat, #auth-btn {
background-color: #2563EB !important;
color: white !important;
border-radius: 6px;
padding: 10px 16px;
font-weight: 600;
}
#send-chat:hover, #reset-chat:hover, #auth-btn:hover {
background-color: #1D4ED8 !important;
}
""") as demo:
gr.Markdown("## 💬 Product Manager Chatbot")
# Login/Signup UI
with gr.Column(visible=True) as login_section:
gr.Markdown("### 🔐 Login or Sign Up to Continue")
email = gr.Textbox(label="Email", placeholder="you@example.com")
password = gr.Textbox(label="Password", type="password")
show_password = gr.Checkbox(label="Show password")
show_password.change(fn=toggle_password, inputs=show_password, outputs=password)
mode = gr.Radio(["Login", "Sign Up"], value="Login", label="Mode")
auth_btn = gr.Button("Submit", elem_id="auth-btn")
auth_error = gr.Textbox(visible=True, interactive=False, show_label=False)
auth_success = gr.Textbox(visible=True, interactive=False, show_label=False)
# Chat UI
with gr.Column(visible=False) as chat_section:
chatbot = gr.Chatbot(label="Chat", elem_id="chat-area", type="messages")
with gr.Row():
user_input = gr.Textbox(show_label=False, placeholder="Ask a product question...", lines=1, scale=8)
send_btn = gr.Button("Send", elem_id="send-chat", scale=1)
reset_btn = gr.Button("Reset Chat", elem_id="reset-chat")
state = gr.State([])
# Button actions
auth_btn.click(fn=authenticate, inputs=[email, password, mode], outputs=[chat_section, login_section, auth_error, auth_success])
send_btn.click(fn=chat, inputs=[user_input, state], outputs=[chatbot, state])
user_input.submit(fn=chat, inputs=[user_input, state], outputs=[chatbot, state])
reset_btn.click(fn=reset, outputs=[chatbot, state])
# Launch app
demo.launch()
|