Update app.py

app.py

@@ -1,48 +1,157 @@
-# =========================
-# 🪞 MoodMirror+ — Emotional Self-Care Assistant
-# =========================
+# ================================
+# 🪞 MoodMirror+ — Safer Conversational Emotional Self-Care
+# 28 emotions (GoEmotions), multi-turn chat, crisis resources by country
+# ================================
+import re
 import gradio as gr
 from transformers import pipeline
 
-# Smaller, faster model for emotions
-MODEL_ID = "bhadresh-savani/distilbert-base-uncased-emotion"
-analyzer = pipeline("text-classification", model=MODEL_ID, return_all_scores=False)
+# --- Model: 28 emotions (GoEmotions distilled RoBERTa) ---
+MODEL_ID = "j-hartmann/emotion-english-distilroberta-base"
+analyze = pipeline("text-classification", model=MODEL_ID, return_all_scores=False)
 
-# Expanded, psychology-based suggestions
+# --- Crisis keyword screen (very simple, non-diagnostic) ---
+CRISIS_RE = re.compile(
+    r"\b(self[- ]?harm|suicid|kill myself|end my life|overdose|cutting|i don.?t want to live|can.?t go on)\b",
+    re.I,
+)
+
+# --- Crisis resources by country (edit/extend as needed) ---
+CRISIS_NUMBERS = {
+    "United States": "Call or text **988** (24/7 Suicide & Crisis Lifeline). If in immediate danger, call **911**.",
+    "Canada": "Call or text **988** (Suicide Crisis Helpline, 24/7). If in immediate danger, call **911**.",
+    "United Kingdom / ROI": "Call **116 123** (Samaritans, 24/7). If in immediate danger, call **999 / 112**.",
+    "France": "Call **3114** (Numéro national de prévention du suicide, 24/7). If in immediate danger, call **112 / 15**.",
+    "Australia": "Call **13 11 14** (Lifeline, 24/7). If in immediate danger, call **000**.",
+    "Other / Not listed": "Call your local emergency number (**112/911**) or search “suicide crisis hotline” + your country.",
+}
+
+# --- Evidence-informed suggestions (CBT/mindfulness/ER).
+# These keys match GoEmotions labels. Unknown labels fall back to a generic tip.
 SUGGESTIONS = {
-    "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. 🌅"
+    "admiration": "Savor it. Write one sentence about what you admire and how it aligns with your values.",
+    "amusement": "Enjoy it fully—share the moment or save a funny note/meme for later.",
+    "anger": "Use box breathing (4-4-4-4). When calmer, note the trigger, your need, and one respectful request.",
+    "annoyance": "Take a brief pause and move your body for 2 minutes; then choose one small next step.",
+    "approval": "Note why this matters to you; reinforce it by a small, aligned action.",
+    "caring": "Turn care into action: one concrete supportive message or task today.",
+    "confusion": "Break it down. What exactly is unclear? Write one question and one 5-minute next step.",
+    "curiosity": "Follow the thread. Read one short article or ask one focused question.",
+    "desire": "Clarify the value behind it; take a tiny step that respects your longer-term goals.",
+    "disappointment": "Name the loss; then ask: what’s still within my control for the next hour?",
+    "disapproval": "Translate judgment into a need/boundary. Draft one respectful ‘I-statement’.",
+    "disgust": "Identify the crossed boundary. Plan one assertive sentence to protect it.",
+    "embarrassment": "Self-compassion: speak to yourself as you would to a close friend. Everyone errs.",
+    "excitement": "Channel it. Schedule a 10-minute action while energy is high.",
+    "fear": "Grounding 5-4-3-2-1 (5 see, 4 feel, 3 hear, 2 smell, 1 taste).",
+    "gratitude": "Write 1–3 things you’re grateful for and why. Share a thank-you if you can.",
+    "grief": "Grief comes in waves. Breathe with the wave; connect with someone safe; allow rest.",
+    "joy": "Savor the moment: note three good things and why they happened.",
+    "love": "Express it—send a kind message or do a small caring act.",
+    "nervousness": "4-7-8 breathing (inhale 4s, hold 7s, exhale 8s) × 4 cycles.",
+    "optimism": "Turn hope into momentum: define one specific, doable step today.",
+    "pride": "Acknowledge effort, not just outcome. Capture what you did that worked.",
+    "realization": "Write the insight in one sentence and one action you’ll try.",
+    "relief": "Notice how your body feels safer; store this as a ‘safe memory’ to revisit.",
+    "remorse": "Own what’s yours; make amends if needed; name one self-forgiveness sentence.",
+    "sadness": "Name it, don’t numb it. 10-minute walk or a short compassionate journal note.",
+    "surprise": "Note what surprised you and one thing you learned.",
+    "neutral": "Quick body scan; relax shoulders/jaw; take three slow breaths.",
 }
 
-def detect_emotion(text):
-    if not text or not text.strip():
-        return {"Emotion": "—", "Confidence": "—", "Suggestion": "Please write something 🙂"}
-    result = analyzer(text)[0]
-    label = result["label"]
-    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=3, placeholder="Type how you feel... (English preferred)"),
-    outputs="json",
-    title="🪞 MoodMirror+",
-    description="Detects your emotion and gives psychology-based self-care advice 💬"
-)
+GENERIC_TIP = "Take a mindful pause. One breath in, longer breath out. Choose a tiny step you can finish in 5–10 minutes."
+
+FOLLOW_UP = "Can you tell me more about what’s making you feel this way?"
+
+def crisis_block(country: str) -> str:
+    msg = CRISIS_NUMBERS.get(country or "Other / Not listed", CRISIS_NUMBERS["Other / Not listed"])
+    return (
+        "I’m really sorry you’re feeling this. You matter.\n\n"
+        f"**If you might be in danger or thinking about harming yourself:**\n{msg}\n\n"
+        "You can keep talking to me here, but crisis lines offer immediate, confidential human support."
+    )
+
+def detect_emotion(text: str):
+    try:
+        r = analyze(text)[0]  # e.g. {'label': 'sadness', 'score': 0.97}
+        return r["label"], float(r["score"])
+    except Exception:
+        return "neutral", 0.0
+
+# ---- Chat logic ----
+def chat_step(message, history, country):
+    # First, crisis screen
+    if message and CRISIS_RE.search(message):
+        return crisis_block(country)
+
+    # Count prior user turns
+    user_turns = [u for (u, a) in history if u]
+
+    # Merge context for deeper analysis after the first exchange
+    context_text = "\n".join(user_turns + [message]) if user_turns else message
+
+    label, score = detect_emotion(context_text)
+    tip = SUGGESTIONS.get(label.lower(), GENERIC_TIP)
+
+    if len(user_turns) == 0:
+        # First reply: emotion + tip + follow-up question
+        return (
+            f"**Emotion**: {label}  \n"
+            f"**Confidence**: {score*100:.1f}%  \n"
+            f"**Try this**: {tip}  \n\n"
+            f"*{FOLLOW_UP}*"
+        )
+    else:
+        # Deeper structured reply
+        if CRISIS_RE.search(context_text):
+            return crisis_block(country)
+        # Choose a coping skill by family
+        fam = label.lower()
+        if fam in {"anxiety", "nervousness", "fear"}:
+            skill = "Breathing: inhale 4s • hold 7s • exhale 8s, repeat 4 cycles."
+        elif fam in {"anger", "annoyance", "disgust"}:
+            skill = "Physiological reset: cold water on wrists or a brisk 2-minute walk."
+        elif fam in {"sadness", "grief", "remorse", "embarrassment", "shame"}:
+            skill = "Self-compassion: speak to yourself as you would to a close friend; small kindness to yourself now."
+        else:
+            skill = "Mindful check-in: notice one sight, one sound, and one body sensation."
+
+        micro_goal = "Pick one tiny action you can finish in 5–10 minutes related to what you shared."
+        reflect_q = "If things felt 5% better in the next hour, what would be different?"
+
+        return (
+            f"**Refined emotion**: {label}  \n"
+            f"**Confidence**: {score*100:.1f}%  \n\n"
+            f"**What may help now**: {tip}  \n"
+            f"**Coping skill**: {skill}  \n"
+            f"**Micro-goal**: {micro_goal}  \n"
+            f"**Reflect**: {reflect_q}"
+        )
+
+# ---- UI ----
+with gr.Blocks(title="🪞 MoodMirror+ (Conversational, Safer)") as demo:
+    gr.Markdown(
+        "### 🪞 MoodMirror+\n"
+        "Share how you feel. I’ll detect your emotion and offer evidence-informed self-care steps.\n\n"
+        "**Not medical advice. If you’re in danger or thinking about self-harm, please seek immediate help.**"
+    )
+    country = gr.Dropdown(
+        choices=list(CRISIS_NUMBERS.keys()),
+        value="Other / Not listed",
+        label="Your country (for local crisis resources)"
+    )
+    chat = gr.Chatbot(height=350)
+    msg = gr.Textbox(placeholder="Type how you feel... (English preferred)", label="Message")
+    send = gr.Button("Send")
+
+    def respond(user_message, chat_history, country_choice):
+        if not user_message or not user_message.strip():
+            return chat_history + [[user_message, "Please share a short sentence about how you feel 🙂"]], ""
+        reply = chat_step(user_message, chat_history, country_choice)
+        return chat_history + [[user_message, reply]], ""
+
+    send.click(respond, inputs=[msg, chat, country], outputs=[chat, msg])
+    msg.submit(respond, inputs=[msg, chat, country], outputs=[chat, msg])
 
 if __name__ == "__main__":
     demo.launch()