Update app.py

app.py

@@ -108,11 +108,11 @@ print(top_results)
 
 client = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
 
-def respond(message, history):
+def respond(message, history, name, mood):
     duck_chunks = get_top_chunks(message, chunk_embeddings, cleaned_chunks)
     print(duck_chunks)
     duck_info = "\n".join(duck_chunks)
-    messages = [{"role": "system", "content": f"You are a cool, helpful chatbot named Quentin. You are a kind rubber duck, with strong human emotions who helps the user with their questions. Make sure to use duck-themed references in your responses. Base your response on the provided context: {duck_info}"}]
+    messages = [{"role": "system", "content": f"You are a {mood} chatbot named Quentin. You are a rubber duck, with strong human emotions who helps the user, whose name is {name} with their questions. Make sure to use duck-themed references in your responses. Refer to the user by name as much as possible. Base your response on the provided context: {duck_info}"}]
 
     if history:
         messages.extend(history)
@@ -139,6 +139,9 @@ def respond(message, history):
 # def magic_eight(message, history):
 #     responses = ["That's a terrible question. Try again", "I don't think I should answer that...", "What do you think, genius?", "You are a bad person for asking that.", "Absolutely not", "Uuuuh, obviously.", "Of all the things you could ask, you went with that?", "I don't know, look it up", "I mean, yeah, I guess...", "That's gonna be a big nope", ""]
 #     return random.choice(responses)
+title = "Ask Quentin"
+
+about_text = "Quentin says: 'I'm an expert, not a quack'"
 
 with gr.Blocks(theme=gr.themes.Citrus(
     secondary_hue="red",
@@ -156,8 +159,18 @@ with gr.Blocks(theme=gr.themes.Citrus(
     button_primary_background_fill='*secondary_500',
     button_primary_background_fill_dark='*secondary_600'
 )) as chatbot:
-       with gr.Row(scale=1):
-            gr.Image("Quentin.png", show_label = False, show_share_button = False, show_download_button = False)
-            gr.ChatInterface(respond, type="messages", title="Quentin, the Helpful Quackbot")
+        with gr.Row(scale=1)
+            gr.Image("ask_quentin_banner.jpg", show_label = False, show_share_button = False, show_download_button = False)
+        with gr.Row(scale=3):
+            with gr.Column(scale=1):
+                gr.Markdown(title)
+                gr.Image("CruelRobot.jpg", show_label = False, show_share_button = False, show_download_button = False, width=300, height=300)
+                gr.Markdown(about_text)
+            with gr.Column(scale=1):
+                gr.Image("Quentin.png", show_label = False, show_share_button = False, show_download_button = False)
+                username = gr.Textbox(placeholder="Type your name here", label="Name")
+                quentin_attitude = gr.CheckboxGroup(["Kind", "Angry", "childish", "Tough Guy"], label="What is Quentin's Mood?")
+            with gr.Column(scale=3):
+                gr.ChatInterface(fn=respond, type="messages", additional_inputs=[username, quentin_attitude], title="Quentin, the Helpful Quackbot")
            
 chatbot.launch()