Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# SPOTIFY EMBED
 spotify_embed = """
 <iframe data-testid="embed-iframe" 
 style="border-radius:12px" 
@@ -15,10 +14,8 @@ clipboard-write;
 encrypted-media; 
 fullscreen; 
 picture-in-picture" 
-loading="lazy"></iframe>
-"""
+loading="lazy"></iframe>"""
 
-# THEME
 theme = gr.themes.Soft().set(
 
     body_background_fill="#fff7fb",
@@ -45,16 +42,12 @@ theme = gr.themes.Soft().set(
     link_text_color="#c77dff"
 )
 
-# LOAD KNOWLEDGE BASE
-with open("knowledge.txt", "r", encoding="utf-8") as f:
+with open("knowledge.txt" , "r", encoding="utf-8") as f:
     knowledge_base = f.read()
 
-# MODEL
 client = InferenceClient("Qwen/Qwen2.5-7B-Instruct")
 
-# SYSTEM PROMPTS
 SYSTEM_MESSAGES = {
-
     "wellness": (
         "You are a kind wellness chatbot. "
         "Give practical, supportive, and thoughtful advice "
@@ -69,18 +62,39 @@ SYSTEM_MESSAGES = {
 }
 
 
-# RESPONSE FUNCTION
 def respond(message, history, mode):
 
-    # NO MODE SELECTED
     if mode is None:
-        yield (
-            "Please select a mode first using the buttons above.",
-            mode
-        )
-        return
 
-    # SYSTEM MESSAGE
+        user_choice = message.lower().strip()
+
+        if user_choice in ["wellness", "wellness mode"]:
+            mode = "wellness"
+            yield (
+                "🌿 Wellness mode activated!\n\n"
+                "Tell me what's on your mind.",
+                mode
+            )
+            return
+
+        if user_choice in ["story", "storytelling", "creative"]:  #changed elif to if to let the user change choices anytime
+            mode = "story"
+            yield (
+                "📖 Creative Storytelling mode activated!\n\n"
+                "Give me a story idea!",
+                mode
+            )
+            return
+
+        else:
+            yield (
+                "Hi! How can I help you today?\n\n"
+                "• wellness\n"
+                "• story",
+                mode
+            )
+            return
+
     messages = [
         {
             "role": "system",
@@ -88,19 +102,16 @@ def respond(message, history, mode):
         }
     ]
 
-    # HISTORY
     if history:
         messages.extend(history)
 
-    # USER MESSAGE
     messages.append({
-        "role": "user",
-        "content": message
+    "role": "user",
+    "content": message
     })
-
+    
     response = ""
 
-    # STREAM RESPONSE
     for chunk in client.chat_completion(
         messages=messages,
         max_tokens=1024,
@@ -109,86 +120,36 @@ def respond(message, history, mode):
         stream=True,
     ):
 
-        token = chunk.choices[0].delta.content or ""
+        token = chunk.choices[0].delta.content or "" 
 
         if token:
             response += token
             yield response, mode
 
 
-# UI
 with gr.Blocks(theme=theme) as demo:
 
-    # MODE STATE
     mode_state = gr.State(None)
 
-    # BANNER
-    gr.Image(
-        "Website Banner.png",
-        show_label=False,
-        container=False
-    )
+    gr.Image("Website Banner.png", show_label=False, container=False)
 
-    # TITLE
     gr.Markdown("# ***Wellness and Storytelling ChatBot***")
 
     gr.Markdown(
-        "DISCLAIMER: While signing, please spell out individual letters in ASL instead of the distinct signs.\n\n"
-        "Choose a mode using the buttons below:"
+        "DISCLAIMER: While signing, please spell out individual letters in ASL instead of the distinct signs \n\n"
+        "Choose a mode by typing:\n\n"
+        "- `wellness`\n"
+        "- `story`"
     )
 
-    # BUTTONS
-    with gr.Row():
-
-        wellness_btn = gr.Button(
-            "🌿 Wellness",
-            variant="primary"
-        )
-
-        story_btn = gr.Button(
-            "📖 Story",
-            variant="secondary"
-        )
-    
-    # CHAT INTERFACE
     chatbot = gr.ChatInterface(
         fn=respond,
         additional_inputs=[mode_state],
-        additional_outputs=[mode_state]
-    )
-
-
-    # BUTTON ACTIONS
-    wellness_btn.click(
-        fn=lambda: (
-            [
-                {
-                    "role": "assistant",
-                    "content": "🌿 Wellness mode activated!"
-                }
-            ],
-            "wellness"
-        ),
-        outputs=[chatbot.chatbot, mode_state]
-    )
-
-    story_btn.click(
-        fn=lambda: (
-            [
-                {
-                    "role": "assistant",
-                    "content": "📖 Story mode activated!"
-                }
-            ],
-            "story"
-        ),
-        outputs=[chatbot.chatbot, mode_state]
+        additional_outputs=[mode_state],
     )
 
-    # MUSIC SECTION
     gr.Markdown("### 🎵 Music We Listened To While Building This")
 
     gr.HTML(spotify_embed)
 
-# LAUNCH
 demo.launch(debug=True)