Update app.py

app.py

@@ -4,6 +4,7 @@ import random
 
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
+# List of motivational quotes
 quotes = [
     "When anger rises, think of the consequences.",
     "You cannot control the wind, but you can adjust your sails.",
@@ -11,51 +12,61 @@ quotes = [
     "Peace begins with a single thought.",
     "Don't let a moment of anger become a lifetime of regret."
 ]
-quote_of_the_day = random.choice(quotes)
-
-def respond(message, history):
-    system_message = "You are a calming and supportive emotional wellness chatbot named Aggro."
 
-    if not history:
-        return {
-            "role": "assistant",
-            "content": f"Hi, I'm Aggro! 🌱\nQuote of the Day: \"{quote_of_the_day}\"\nHow is your day going?"
-        }
+quote_of_the_day = random.choice(quotes)
 
+def respond(message, history, system_message, max_tokens, temperature, top_p):
     lowered = message.lower()
+
     if "angry" in lowered:
-        return {"role": "assistant", "content": "I'm sorry you're feeling angry. Let's take a deep breath. Want to try a calming exercise?"}
+        yield "I'm sorry you're feeling angry. Let's take a deep breath together. Would you like to try a calming exercise?"
+        return
     elif "sad" in lowered:
-        return {"role": "assistant", "content": "It's okay to feel sad. I'm here to support you. 💙"}
+        yield "It's okay to feel sad sometimes. I'm here to support you. 💙"
+        return
     elif "stressed" in lowered:
-        return {"role": "assistant", "content": "Let’s pause and breathe. I’m here with you. 🧘"}
+        yield "Let's pause and take a deep breath. I'm here for you."
+        return
     elif "calm" in lowered:
-        return {"role": "assistant", "content": "That's great to hear! 🌿 Keep holding on to that calm energy."}
+        yield "That's great to hear! Stay present and peaceful. 🌿"
+        return
 
     messages = [{"role": "system", "content": system_message}]
-    for m in history:
-        messages.append(m)
+    for user_msg, bot_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if bot_msg:
+            messages.append({"role": "assistant", "content": bot_msg})
+
     messages.append({"role": "user", "content": message})
 
     response = ""
     for msg in client.chat_completion(
         messages,
-        max_tokens=512,
+        max_tokens=max_tokens,
         stream=True,
-        temperature=0.7,
-        top_p=0.95,
+        temperature=temperature,
+        top_p=top_p,
     ):
         token = msg.choices[0].delta.content
         response += token
+        yield response
 
-    return {"role": "assistant", "content": response}
-
-# ✅ Build the interface
+# UI layout
 demo = gr.ChatInterface(
-    fn=respond,
-    chatbot=gr.Chatbot(label="Aggro 💬", type="messages"),
+    respond,
+    additional_inputs=[
+        gr.Textbox(
+            value="You are a calming and supportive emotional wellness assistant named Aggro.",
+            label="System message",
+            visible=False
+        ),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
+    ],
     title="💚 Meet Aggro - Your Emotion Companion",
-    description=f"A calming chatbot to help you reflect and feel supported.\n\n💬 _Quote of the Day:_ \"{quote_of_the_day}\""
+    description=f"💬 Quote of the Day: _{quote_of_the_day}_"
 )
 
 if __name__ == "__main__":