Update app.py

app.py

@@ -46,18 +46,21 @@ def generate_response(user_input):
     )
     response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
+    # Trim the output to remove extraneous content
+    response_text = response_text.split("Here's something")[0].strip()
+    
     # Add a suggestion for coping activity based on keywords in user input
-    if "angry" in user_input.lower() or "frustrated" in user_input.lower():
+    if "crying" in user_input.lower() or "overload" in user_input.lower():
         activity_suggestion = (
-            "When feeling angry or overwhelmed by work, it’s important to take a step back. "
-            "Deep breathing exercises can help calm your mind and regain some control. "
-            "Try inhaling for four counts, holding for four counts, and exhaling for four counts. "
-            "Doing this for a few minutes can make a world of difference."
+            "Work overload can be overwhelming, and it's completely understandable to feel like you're reaching your limit. "
+            "Sometimes, taking a short break can help you gain some clarity and calm your mind. "
+            "If possible, try stepping outside for a few minutes to breathe deeply or do some light stretching to release tension."
         )
-    elif "sad" in user_input.lower() or "lonely" in user_input.lower():
+    elif "angry" in user_input.lower() or "frustrated" in user_input.lower():
         activity_suggestion = (
-            "Writing about your feelings can be very therapeutic. Try journaling as a way to process and release some of your emotions. "
-            "You don’t need to worry about grammar or structure—just let your thoughts flow."
+            "When feeling overwhelmed, anger can sometimes mask deeper emotions like stress or exhaustion. "
+            "Taking deep breaths or doing a short meditation can help you ground yourself and regain focus. "
+            "Even small moments of calm can make a huge difference."
         )
     else:
         activity_suggestion = (