Spaces:
Paused
Paused
File size: 2,304 Bytes
2dce015 |
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 |
# Standard library imports
import os
import time
# Related third party imports
import gradio as gr
from dotenv import load_dotenv
# Local application/library specific imports
from final_funcs import auth_page, SP_STATE
from final_agent import create_agent
from final_feedback import feedback_page
from final_msgs import INSTRUCTIONS, GREETING, CHAT_HEADER, WARNING
load_dotenv()
KEY = os.getenv("OPENAI_API_KEY")
def add_text(history, text):
history = history + [(text, None)]
return history, gr.update(value="", interactive=False)
def bot(history):
user_input = history[-1][0]
if len(history) == 1: # this is the first message from the user
response = GREETING
elif SP_STATE.value is None:
response = WARNING
elif user_input.strip() == '!help': # TODO: streaming !help message looks bad
response = INSTRUCTIONS
else:
response = agent_executor(user_input, include_run_info=True)
response = response["output"]
history[-1][1] = ""
for character in response:
history[-1][1] += character
gr.update(interactive=True)
time.sleep(0.0075)
yield history
with gr.Blocks() as chat_page:
gr.Markdown(CHAT_HEADER)
agent_executor = create_agent(KEY)
chatbot = gr.Chatbot([], elem_id="chatbot", height=400, label="Apollo 🎵")
with gr.Row():
txt = gr.Textbox(
show_label=False,
placeholder="What would you like to hear?",
container=False
)
txt_msg = txt.submit(add_text, [chatbot, txt], [chatbot, txt], queue=False).then(
bot, chatbot, chatbot
)
txt_msg.then(lambda: gr.update(interactive=True), None, [txt], queue=False)
gr.Examples(["Play CRAZY by AXL",
"I'm feeling great today, match my vibe",
"Make me a relaxing playlist of SZA-like songs"],
inputs=[txt], label="")
with gr.Accordion(label="Commands & Examples 📜", open=False):
gr.Markdown(INSTRUCTIONS)
apollo = gr.TabbedInterface([chat_page, auth_page, feedback_page],
["Music", "Authentication", "Feedback"],
theme = "finlaymacklon/boxy_violet")
apollo.queue()
apollo.launch()
|