Update src/responses.py

src/responses.py

@@ -1,66 +1,39 @@
 import random
+from textblob import TextBlob
 
-# Copied directly from your script’s dictionaries
+# Your response dictionaries
 responses = {
-    "sadness": [
-        "It’s okay to feel down sometimes. I’m here to support you.",
-        "I'm really sorry you're going through this. Want to talk more about it?",
-        "You're not alone — I’m here for you."
-    ],
-    "anger": [
-        "That must have been frustrating. Want to vent about it?",
-        "It's okay to feel this way. I'm listening.",
-        "Would it help to talk through it?"
-    ],
-    "love": [
-        "That’s beautiful to hear! What made you feel that way?",
-        "It’s amazing to experience moments like that.",
-        "Sounds like something truly meaningful."
-    ],
-    "happiness": [
-        "That's awesome! What’s bringing you joy today?",
-        "I love hearing good news. 😊",
-        "Yay! Want to share more about it?"
-    ],
-    "neutral": [
-        "Got it. I’m here if you want to dive deeper.",
-        "Thanks for sharing that. Tell me more if you’d like.",
-        "I’m listening. How else can I support you?"
-    ]
+    "sadness": [...],
+    "anger":   [...],
+    "love":    [...],
+    "happiness": [...],
+    "neutral": [...]
 }
-
-# Tip & goodbye logic copied from your script
-relaxation_resources = {
-    "exercise": "Try this 5-4-3-2-1 grounding method: ...",
-    "video": "Short calming video: https://youtu.be/O-6f5wQXSu8"
+relax_resources = {
+    "exercise": "Try the 5-4-3-2-1 grounding method: ...",
+    "video":      "https://youtu.be/O-6f5wQXSu8"
 }
-help_keywords    = ["suggest","help","calm","exercise","relax","any tips","can you"]
-negative_inputs  = ["not good","feel bad","feel sad","anxious","depressed","upset","stress","worried"]
-thank_you_inputs = ["thank","thanks","thank you"]
-bye_inputs       = ["bye","goodbye","exit","quit"]
+help_kw    = [...]
+neg_inputs = [...]
+thx_inputs = [...]
+bye_inputs = [...]
+awaiting   = False
 
-awaiting_tip_type = False
+def correct_spelling(text):
+    return str(TextBlob(text).correct())
 
 def get_response(emotion, user_input):
-    global awaiting_tip_type
+    global awaiting
     ui = user_input.lower()
-    # 1) Exit
-    if any(b in ui for b in bye_inputs):
-        return "Take care! I’m here whenever you want to talk. 🌿", True
-    # 2) Thanks
-    if any(t in ui for t in thank_you_inputs):
-        return "You're most welcome! 💙", False
-    # 3) Tip request
-    if awaiting_tip_type:
-        if "video" in ui:
-            awaiting_tip_type = False
-            return relaxation_resources["video"], False
-        if "exercise" in ui:
-            awaiting_tip_type = False
-            return relaxation_resources["exercise"], False
-        return "Would you prefer a short video or a simple exercise?", False
-    if any(k in ui for k in help_keywords):
-        awaiting_tip_type = True
-        return "Prefer a brief video or an exercise?", False
-    # 4) Emotion-based reply
-    return random.choice(responses.get(emotion, responses["neutral"])), False
+    # 1) exit
+    if any(b in ui for b in bye_inputs): return "Take care!", True
+    # 2) thanks
+    if any(t in ui for t in thx_inputs): return "You're welcome!", False
+    # 3) tip flow
+    if awaiting:
+        # ...same logic...
+    if any(k in ui for k in help_kw):
+        awaiting = True
+        return "Video or exercise?", False
+    # 4) emotion reply
+    return random.choice(responses.get(emotion, responses['neutral'])), False