Update app.py

app.py

@@ -1,17 +1,30 @@
+# =========================
+# 🪞 MoodMirror+ — Emotional Self-Care Assistant
+# =========================
 import gradio as gr
 from transformers import pipeline
 
-MODEL_ID = "j-hartmann/emotion-english-distilroberta-base"
+# Smaller, faster model for emotions
+MODEL_ID = "bhadresh-savani/distilbert-base-uncased-emotion"
 analyzer = pipeline("text-classification", model=MODEL_ID, return_all_scores=False)
 
+# Expanded, psychology-based suggestions
 SUGGESTIONS = {
-    "joy": "Keep up the good vibes! 🥳 Share your happiness with someone.",
-    "sadness": "Take a short break, drink some water, or go for a walk. 🌿",
-    "anger": "Take a deep breath and try to relax. 💪",
-    "fear": "Talk to someone you trust. 💬",
-    "love": "Send a kind message to someone you care about. 💖",
-    "surprise": "Write down what surprised you — curiosity is powerful! 🤔",
-    "neutral": "Take a moment to stretch and breathe. 🌸",
+    "joy": "Celebrate your happiness! Write down three things you’re grateful for today. Positive reflection reinforces joy. 🥳",
+    "sadness": "Allow yourself to feel it, but don’t isolate. Reach out to someone or do a small act of self-kindness, like journaling or a short walk. 🌿",
+    "anger": "Pause and breathe slowly through your nose, exhale longer than you inhale. When calm, write down what triggered you — it helps regain control. 💪",
+    "fear": "Ground yourself in the present: notice 5 things you see, 4 you feel, 3 you hear. This calms the nervous system. 💬",
+    "love": "Express it! Send a message or do something kind for someone you care about. Connection strengthens emotional well-being. 💖",
+    "surprise": "Note what surprised you — was it positive or challenging? Curiosity helps your brain adapt to change. 🤔",
+    "neutral": "Check in with your body: are you tense or relaxed? Take a short stretch or deep breath to stay balanced. 🌸",
+    "disgust": "It’s okay to feel repelled; ask yourself what boundary was crossed. Channel that energy into assertive communication. 🚫",
+    "guilt": "Reflect: what part of this can you learn from, and what can you let go? Apologize if needed, but forgive yourself too. ❤️‍🩹",
+    "shame": "Remember that everyone makes mistakes. You deserve compassion. Talk to someone safe — shame shrinks in empathy. 🤝",
+    "anxiety": "Try the 4-7-8 breathing method (inhale 4s, hold 7s, exhale 8s). Gentle movement or calming music can help. 🌬️",
+    "gratitude": "Write down one thing that went well today, no matter how small. Gratitude rewires the brain toward positivity. 🙏",
+    "boredom": "Switch tasks or move your body — even 2 minutes helps reset focus. Try learning one new fact or word. ⚡",
+    "confusion": "Break your thoughts down: what exactly feels unclear? Write your questions, and find one small next step. 🧩",
+    "hope": "Nurture it! Visualize a small goal and take one concrete step toward it today. Progress feeds optimism. 🌅"
 }
 
 def detect_emotion(text):
@@ -19,20 +32,16 @@ def detect_emotion(text):
         return {"Emotion": "—", "Confidence": "—", "Suggestion": "Please write something 🙂"}
     result = analyzer(text)[0]
     label = result["label"]
-    score = result["score"]
-    suggestion = SUGGESTIONS.get(label.lower(), SUGGESTIONS["neutral"])
-    return {
-        "Emotion": label,
-        "Confidence": f"{score*100:.1f}%",
-        "Suggestion": suggestion
-    }
+    score = float(result["score"])
+    suggestion = SUGGESTIONS.get(label.lower(), "Take a mindful pause — you’re doing great for checking in. 💫")
+    return {"Emotion": label, "Confidence": f"{score*100:.1f}%", "Suggestion": suggestion}
 
 demo = gr.Interface(
     fn=detect_emotion,
-    inputs=gr.Textbox(lines=2, placeholder="Type how you feel... (English preferred)"),
+    inputs=gr.Textbox(lines=3, placeholder="Type how you feel... (English preferred)"),
     outputs="json",
-    title="🪞 MoodMirror",
-    description="Detects the main emotion in your text and gives a self-care suggestion 💬"
+    title="🪞 MoodMirror+",
+    description="Detects your emotion and gives psychology-based self-care advice 💬"
 )
 
 if __name__ == "__main__":