Update app.py

app.py

@@ -4,7 +4,9 @@ import json
 import datetime
 import gradio as gr
 
-# User data
+# ----------------------------
+# User data & existing lists
+# ----------------------------
 user_info = {"name": "", "age": "", "gender": "", "language": "english" , "Guardian_info": ""}
 chat_history = []
 journal_entries = []
@@ -31,54 +33,14 @@ lang_codes = {
     "french": "fr", "german": "de"
 }
 
-# List of harmful or dangerous keywords
+# (Keep your harmful_keywords and emotions as-is — shortened here for clarity)
 harmful_keywords = [
-   "abomination", "annihilate", "arson", "ass", "assault", "atrocity",
-"backstab", "barbaric", "barbarian", "beast", "belittle", "berserk",
-"betray", "betrayal", "bigot", "blacklist", "blood", "bloodbath",
-"bloody", "bomb", "bonehead", "brainless", "brutal", "brute", "burn",
-"butcher", "cage", "callous", "cancer", "chaos", "cheater", "cold-blooded",
-"conman", "corrupt", "cowardly", "crash", "crime", "criminality", "crook",
-"cruel", "cruelty", "curseword", "damage", "danger", "dangerous", "deadly",
-"deceive", "defame", "defeat", "degenerate", "demon", "despise", "destroy",
-"destruction", "destructive", "devastate", "devil", "diabolical", "dictator",
-"dirty", "disaster", "dishonor", "disrespect", "disturb", "divisive", "drunk",
-"dumbass", "dungeon", "enemies", "enemy", "enrage", "evilness", "exile",
-"exploit", "exterminate", "fake", "fanatic", "fascist", "fatal", "fire",
-"forgery", "fraud", "fraudster", "freakish", "gang", "gangster", "ghastly",
-"ghoul", "gory", "grotesque", "gruesome", "hag", "halfwit", "hardhearted",
-"harm", "hateable", "hateful", "havoc", "hell", "hellish", "heretic",
-"hideous", "horrid", "horrific", "hostility", "howl", "hypocrite", "illegal",
-"illicit", "immoral", "imposter", "injure", "insensitive", "insidious",
-"intolerant", "irate", "jerky", "joke", "joker", "kidnap", "killjoy", "liar",
-"loath", "loathsome", "lowlife", "madness", "malicious", "malign",
-"manipulate", "menace", "monster", "monstrous", "mug", "murderous",
-"mutant", "mutiny", "nasty", "nefarious", "neglect", "negative", "offend",
-"offender", "ominous", "oppress", "outrage", "outrageous", "parasite",
-"penalty", "perish", "pest", "plague", "poison", "poisonous", "pollute",
-"psychopath", "punch", "punish", "punishment", "quarrel", "rage", "rat",
-"rebellion", "reckless", "repel", "repulsive", "revolt", "revolting",
-"rioter", "rob", "robber", "rotten", "ruin", "savage", "scare", "scary",
-"schemer", "scold", "scorn", "scoundrel", "scream", "screwed", "serpent",
-"shady", "sham", "shameful", "shatter", "shocking", "sin", "sinister",
-"slavey", "slimy", "smash", "smite", "snob", "spite", "spiteful", "stab",
-"stench", "stink", "strike", "subvert", "suspicious", "target", "tease",
-"terrible", "terrify", "thief", "threat", "threaten", "thug", "torment",
-"torture", "trap", "trick", "trigger", "trouble", "uncivilized", "undermine",
-"unfair", "unjust", "unkind", "usurper", "vandal", "vandalize", "venom",
-"venomous", "villain", "villainous", "vindictive", "violent", "viper",
-"vulgarity", "war", "warmonger", "waste", "wickedness", "wild", "wound",
-"wrath", "wreck", "wrong", "yell", "zombie","suicide", "kill myself", "end my life", "harm myself", "cut myself",
-"want to die", "die", "jump off", "self-harm"
-"suicide", "kill myself", "end my life", "harm myself", "cut myself",
-"want to die", "die", "jump off", "self-harm", "give up", "not worth it",
-"ending it all", "self-hate", "suicidal", "tragic", "trauma", "pain",
-"painful", "sorrow", 
-"unworthy", "hurt", "grief", "agony", "doom", "doomed"
-
+    # ... (keep the full list from your original file) ...
+    "suicide", "kill myself", "end my life", "harm myself", "cut myself",
+    "want to die", "die", "jump off", "self-harm", "self-hate", "suicidal",
+    # etc.
 ]
 
-# Harmful word detection response
 harmful_response = (
     "⚠️ It sounds like you're going through a really tough time.\n\n"
     "Please reach out to a professional:\n\n"
@@ -93,284 +55,260 @@ harmful_response = (
     "You are not alone. There are people who care and want to help 💚"
 )
 
-# Simplified emotions dictionary
+# Simplified emotions dictionary (keep your full mapping; trimmed in this view)
 emotions = {
-     "sad": "I hear you 💙. I’m really sorry you’re feeling sad right now.Do you want to tell me what’s making you feel this way, or would you rather I just try to gently cheer you up?Here are a few small things that sometimes help when sadness feels heavy:🌬️ Take 3 deep, slow breaths — it really calms the body.✍️ Write down what you’re feeling in a notebook (or even here with me). Sometimes just letting it out helps.🎧 Put on a calming or favorite song — music can shift your mood more than we realize.🌳 Step outside for 5 minutes and notice something simple in nature, like the sky, trees, or birds.💡 You don’t have to fight sadness alone — it’s okay to feel it and also okay to lean on others when it’s too much.",
-    "anxiety": "It’s okay — anxiety can feel overwhelming, like your mind is running a race it didn’t sign up for.Try slowing down your breathing: inhale for 4 seconds, hold for 4, exhale for 6.Your thoughts are not facts — they’re just passing clouds, and you are the sky.",
-    "angry": "It’s normal to feel angry — it’s your mind’s way of telling you something feels unfair or hurtful.Before reacting, give yourself space: take a deep breath, unclench your jaw, and let your shoulders drop.Once you feel calmer, you can decide how to express your feelings in a way that helps, not hurts.",
-    "lonely": "You are not alone. Talking helps — I’m here to listen.",
-    "rejected": "Rejection hurts but it doesn’t define your worth.",
-    "exams": "Plan short sessions with breaks. You can do this!",
-    "parents": "Family pressure is real. Want to talk more about it?",
-    "money": "Finances are tough, but your value isn’t tied to them.",
-    "too much": "Pause. Let’s break it into small steps.",
-    "friend": "Friendship can be tricky. Want to share what happened?",
-    "love": "Love is powerful — and sometimes painful. I’m here to help you reflect.",
-    "forced": "You don’t have to do what doesn’t feel right.",
-    "compare": "You’re unique. Comparison steals joy.",
-    "numb": "It’s okay to feel nothing. Let’s gently bring you back.",
-    "instagram": "Let’s take a small break from social media together.",
-    "pretend": "You don’t need to pretend here. Be real, be you.",
-    "bullied": "You don’t deserve this. You are strong and valuable.",
-    "misunderstood": "That can feel frustrating. I understand you.",
-    "rainy": "Rainy days can be gloomy. A warm drink and chat may help.",
-    "unwell": "Rest, hydrate, and be kind to yourself today.",
-    "overthink": "Let’s try to focus on what’s in your control.",
-    "who am i": "Great question! Let’s explore that together.",
-    "climate": "You care — and that’s powerful. Even small actions help.",
-    "pointless": "Your life matters, even when it doesn’t feel like it.",
-    "ignored": "You are seen and heard here. Let’s talk.",
-    "can’t sleep": "Try breathing deeply. Would soft music help?",
-    "stuck": "Let’s look at one thing we *can* do today.",
-    "hopeless": "Even small steps forward are progress. Hope grows.",
-    "depressed": "You’re not alone in this. Let’s talk through it.",
-    "scared": "Let’s face this together, one moment at a time.",
-    "afraid": "Fear shows you care. Let’s understand it better.",
-    "confused": "It’s okay not to have all the answers yet.",
-    "guilty": "Guilt teaches us. Let’s grow from it, not drown in it.",
-    "ashamed": "You are more than your mistakes.",
-    "failure": "Failure means you tried. That’s strength.",
-    "panic": "Breathe with me. Ground yourself. You are okay.",
-    "stressed": "Let’s identify your stress and sort it gently.",
-    "bored": "Want to do something fun or creative together?",
-    "insecure": "You have strengths worth celebrating.",
-    "uncertain": "Uncertainty is part of growth. Let’s navigate it.",
-    "worried": "What are you worried about? Let’s untangle it.",
-    "homesick": "It’s okay to miss home. What comforts you?",
-    "crying": "Crying is healing. Let it out if needed.",
-    "jealous": "Let’s explore what you truly desire.",
-    "envy": "You can build your own journey. You’re enough.",
-    "hate": "Let’s turn that strong emotion into understanding.",
-    "bitterness": "Bitterness is heavy. Want to release some of it?",
-    "regret": "We all have regrets. What can we learn from yours?",
-    "betrayed": "That hurts deeply. Let’s process it together.",
-    "heartbroken": "Healing takes time. Let’s take the first step.",
-    "alone": "You’re not alone here. I’m right with you.",
-    "trapped": "There’s always a way out. Let’s look for options.",
-    "disappointed": "Disappointment is valid. Let’s reflect and reset.",
-    "pressured": "You don’t have to carry it all. Let’s prioritize.",
-    "unloved": "You are lovable. Truly. Start with self-kindness.",
-    "exhausted": "Rest. That’s productive too.",
-    "withdrawn": "It’s okay to pull back. But don’t shut out support.",
-    "disrespected": "Respect matters. You deserve it.",
-    "avoided": "You matter. Let’s talk about what happened.",
-    "insomnia": "Let’s try a gentle nighttime routine.",
-    "neglected": "You deserve care and attention.",
-    "nervous": "New things are scary. But you’re capable.",
-    "frustrated": "Let’s release that pressure gently.",
-    "humiliated": "That experience doesn’t define you.",
-    "unimportant": "You matter. Just by being you.",
-    "abandoned": "That’s painful. You’re not alone anymore.",
-    "defeated": "You’re still here. That’s strength.",
-    "shy": "Quiet doesn’t mean weak. Your voice matters.",
-    "paranoid": "Let’s ground our thoughts in facts and truth.",
-    "restless": "Let’s find a healthy outlet together.",
-    "gloomy": "Let’s look for a little light together.",
-    "anxious": "Let’s name the worry, then tame it.",
-    "worthless": "You are worthy — of love, peace, and joy.",
-    "invisible": "I see you. You matter.",
-    "lethargic": "Small movement helps. Try stretching.",
-    "annoyed": "Want to vent? I’m listening.",
-    "fomo": "It’s okay not to do everything. Your pace is right.",
-    "peer pressure": "Your choices are your own. Stand strong.",
-    "embarrassed": "Everyone messes up sometimes. Let it go.",
-    "unappreciated": "I appreciate you. Let’s celebrate your efforts.",
-    "hollow": "You may feel empty now — but you’re not.",
-    "resentful": "Let’s process it before it poisons your peace.",
-    "in pain": "Pain hurts, but talking eases it. Let’s talk.",
-    "drained": "Rest is fuel. It’s okay to pause.",
-    "moody": "Moods pass. Let’s ride this one out together.",
-    "isolated": "Let’s reconnect — even one small step helps.",
-    "fake": "You can be real here. No masks.",
-    "mentally tired": "Mental fatigue is real. Let’s rest and reset.",
-    "broken": "Cracks let the light in. You’re still whole inside.",
-    "crushed": "That sounds painful. Let’s unpack it together.",
-    "troubled": "Name the trouble, then tame it. I’m here.",
-    "underestimated": "You are more than they think. Show them. Gently.",
-    "burned out": "Burnout needs healing. Step back to move forward.",
-    "stress" : "Take 3 deep breaths and relax your shoulders.",
-    "stressful" : "Break tasks into smaller steps to feel less overwhelmed.",
-     "anxiety" : "Focus on slow breathing for 5 minutes to calm your mind.",
-    "panic" : "Ground yourself by naming 5 things you see around you.",
-"fear" : "Write down your fear and list why it may not be true.",
-"anger" : "Walk away for a moment and count to 10.",
-"rage" : "Listen to calming music to release intense anger.",
-"sadness" : "Call a friend or talk to someone who cares.",
-"grief" : "Allow yourself to cry and express your emotions.",
-"trauma" : "Practice grounding exercises like touching nearby objects.",
-"abuse" : "Remember, you deserve respect—consider reaching out for help.",
-"depression" : "Take a short walk outside and notice nature.",
-"loneliness" : "Text someone you trust to connect.",
-"hopeless" : "Think of one small goal you can achieve today.",
-"exhausted" : "Drink water and rest for at least 10 minutes.",
-"overwhelmed" : "Make a priority list and focus on just 1 thing.",
-"nervous" : "Practice positive affirmations like 'I can handle this.'",
-"tired" : "Take a quick power nap if possible.",
-"fatigue" : "Stretch your body to release tension.",
-"burnout" : "Take a complete 10-minute break without screens.",
-"frustrated" : "Pause and do 5 deep breaths to regain focus.",
-"helpless" : "Ask for support—it's okay to seek help.",
-"confused" : "Write down your thoughts to organize them.",
-"worried" : "Challenge your thoughts by asking 'Is this likely to happen?'",
-"lost" : "Take one small step towards clarity, like making a to-do list.",
-"crying" : "Drink water and sit in a comfortable position to calm.",
-"heartbreak" : "Allow yourself to heal and talk to a trusted friend.",
-"guilt" : "Forgive yourself and remind yourself you are human.",
-"shame" : "Remind yourself that mistakes do not define you.",
-"fearful" : "Focus on your breathing for a few minutes.",
-"meltdown" : "Step away and try a 5-minute meditation.",
-"pressured" : "Break tasks into smaller, manageable chunks.",
-"uneasy" : "Listen to relaxing music or nature sounds.",
-"unmotivated" : "Start with one tiny task to build momentum.",
-"restless" : "Try a short physical activity like stretching.",
-"freaking" : "Count to 10 slowly to calm down.",
-"cry" : "Let your tears out—it helps to release emotions.",
-"breakdown" : "Sit somewhere quiet and breathe deeply.",
-"fail" : "Remind yourself failure is part of learning.",
-"failure" : "Focus on what you can improve, not what you lost.",
-"pressure" : "Take breaks and drink water to refresh your mind.",
-"worry" : "Write down your worries and challenge them logically.",
-"trouble" : "Focus on one thing you can solve right now.",
-"numb" : "Take a walk outside to feel present.",
-"tense" : "Relax your shoulders and stretch for a minute.",
-"hurting" : "Talk to someone you trust about your pain.",
-"hurt" : "Be kind to yourself and allow healing time.",
-"broken" : "Remember that healing is slow but possible.",
-"tension" : "Roll your shoulders and take deep breaths.",
-"angry" : "Punch a pillow or squeeze a stress ball instead of yelling.",
-"pessimistic" : "List 3 things that went well today.",
-"scared" : "Remind yourself of times you overcame fear.",
-"worn-out" : "Take a 5-minute break and hydrate.",
-"mental pain" : "Talk about your feelings with a trusted person.",
-"emotional" : "Allow yourself to feel without judging your emotions.",
-"upset" : "Pause and do 10 slow breaths.",
-"low" : "Do something small that brings joy.",
-"bitter" : "Write down 3 positive things about your day.",
-"uneasiness" : "Practice slow and mindful breathing.",
-"tense" : "Try progressive muscle relaxation.",
-"tight" : "Do a gentle body stretch.",
-"stressed-out" : "Focus on what you can control right now.",
-"paranoid" : "Talk to someone to verify your fears.",
-"devastated" : "Give yourself time to heal.",
-"crybaby" : "Your tears are valid—let them flow.",
-"unhappy" : "List 1 thing that can make you smile.",
-"negative" : "Counter each negative thought with a positive one.",
-"hopelessness" : "Set one small achievable goal today.",
-"unsure" : "Write your pros and cons to gain clarity.",
-"self-doubt" : "Remind yourself of your past achievements.",
-"hurtful" : "Distance yourself from negativity for a while.",
-"unwell" : "Drink water and rest to recover.",
-"emotionless" : "Do something creative to spark feelings.",
-"lostness" : "Journal your thoughts for clarity.",
-"distress" : "Breathe in for 4 seconds, hold 4, out for 4.",
-"emergency" : "Call someone you trust immediately.",
-"unsettled" : "Meditate for 5 minutes on your breath.",
-"chaotic" : "Focus on organizing one small area around you.",
-"break" : "Take 10 minutes to step outside.",
-"unstable" : "Seek support from a friend or counselor.",
-"hopelessly" : "Remind yourself nothing lasts forever.",
-"pain" : "Try deep breathing or journaling about your pain.",
-"painful" : "Distract yourself with something positive.",
-"angsty" : "Express feelings through writing or art.",
-"uneasiness" : "Light a candle and focus on the flame for a minute.",
-"irritated" : "Do 5 deep breaths to calm irritation.",
-"frustration" : "Talk about your frustration instead of bottling it.",
-"exhaustion" : "Sleep for at least 15 minutes if possible.",
-"mental stress" : "Pause and take a few deep breaths.",
-"low mood" : "Listen to uplifting music.",
-"discouraged" : "Remind yourself of past successes.",
-"confusion" : "Pause, take a breath, and write down thoughts.",
-"overthinking" : "Distract yourself with a small physical activity.",
-"hurtful words" : "Remind yourself words don’t define your worth.",
-"traumatized" : "Seek safe space and talk about your feelings.",
-"frantic" : "Slow your breathing and count backwards from 10.",
-"chaos" : "Clean or organize something small around you.",
-"lost mind" : "Ground yourself by feeling the floor beneath you.",
-"hectic" : "Prioritize and focus on one task at a time.",
-"unbalanced" : "Sit and take 5 deep breaths.",
-"mentally tired" : "Close your eyes for 2 minutes and relax.",
-"self-blame" : "Remind yourself that mistakes happen to everyone.",
-"failure feeling" : "Focus on one thing you’ve done right today.",
-"feeling bad" : "Say one kind thing to yourself.",
-"worthless" : "Remember your value is not defined by others.",
-"helplessness" : "Ask for support—it’s a sign of strength.",
-"emotional pain" : "Talk to someone who cares about you.",
-"lack of energy" : "Stretch your arms and drink water.",
-"lack of sleep" : "Take a power nap of 10-20 minutes.",
-"mood swings" : "Track your mood to find triggers.",
-"frightened" : "Hold a comforting object and breathe slowly.",
-"self-hate" : "Replace self-criticism with self-kindness.",
-"negativity" : "Say one positive affirmation out loud.",
-"anxious" : "Practice slow inhaling and exhaling.",
-"feeling down" : "Do something you enjoy, even for 5 minutes.",
-"unstable mind" : "Pause, breathe, and write what you feel.",
-"uneasy feeling" : "Ground yourself by noticing your 5 senses.",
-"confusing" : "Write down questions to get clarity.",
-"worn out" : "Take a rest break with calm music.",
-"pressured" : "Let go of what’s not urgent.",
-"nervousness" : "Take 10 deep breaths slowly.",
-"mental exhaustion" : "Do a simple 5-min body stretch.",
-"crisis" : "Focus on what is in your control right now.",
-"melancholy" : "Write or draw to express feelings.",
-"mental breakdown" : "Sit down, close your eyes, and breathe.",
-"feeling weak" : "Remind yourself you’ve overcome challenges before.",
-"shaking" : "Sit still and breathe deeply for 2 minutes.",
-"mental fatigue" : "Take a short screen-free break.",
-"scared feelings" : "Tell yourself, 'This feeling will pass.'",
-"restless mind" : "Do 5 minutes of meditation or yoga.",
-"mind chaos" : "Write down your thoughts to clear your mind.",
-"uncertain" : "List your options and think calmly.",
-"self-harming thoughts" : "Reach out to someone you trust immediately.",
-"dark thoughts" : "Remember, you are not your thoughts.",
-"mental pressure" : "Pause and breathe for 5 deep breaths.",
-"losing hope" : "Focus on one good thing in your day.",
-"bad mood" : "Do something fun or silly for 5 minutes.",
-"pain inside" : "Talk to someone, don’t hold it in.",
-"numb feeling" : "Do a gentle self-massage on your hands.",
-"lack of motivation" : "Start with just 1 easy task.",
-"mind clutter" : "Write your thoughts on paper.",
-"feeling low" : "Listen to uplifting songs or affirmations.",
-"irritation" : "Pause and breathe deeply 3 times.",
-"disturbed" : "Sit somewhere quiet for 5 minutes.",
-"heavy heart" : "Talk about how you feel with someone.",
-"mental burden" : "Break it into small, manageable steps.",
-"scared mind" : "Repeat to yourself: 'I am safe now.'",
-"tension headache" : "Massage your temples gently.",
-"mind storm" : "Sit quietly and focus on one thought.",
-"mental struggle" : "Write a small gratitude note.",
-"mind fatigue" : "Do light stretches or walk for 5 minutes."
+    "sad": "I hear you 💙. I’m really sorry you’re feeling sad right now. Do you want to tell me what’s making you feel this way, or would you rather I just try to gently cheer you up?",
+    "anxiety": "It’s okay — anxiety can feel overwhelming, like your mind is running a race it didn’t sign up for. Try slowing down your breathing: inhale for 4 seconds, hold for 4, exhale for 6.",
+    "angry": "It’s normal to feel angry — it’s your mind’s way of telling you something feels unfair or hurtful.",
+    # ... keep the rest of your emotion replies ...
 }
 
-# Functions
-def set_personal_info(name, age, gender, language, Guardian_info):
-    user_info.update({"name": name, "age": age, "gender": gender, "language": language, "Guardian_info": Guardian_info,})
-    return gr.update(visible=True), f"✅ Welcome {name}! Preferences saved."
+# ----------------------------
+# Conversation state for Hybrid (C) memory model
+# ----------------------------
+conversation_state = {
+    "emotion": None,       # e.g., "sad"
+    "context": None,       # e.g., "school", "parents", "friend", "exam"
+    "followup_stage": 0,   # how many follow-ups we've done for this emotion
+    "turns_since_set": 0   # used to expire after a small idle period if needed
+}
 
-def show_personal_data():
-    today = datetime.date.today().strftime("%Y-%m-%d (%A)")
-    return f"📅 {today}\n👤 Name: {user_info['name']}\n🎂 Age: {user_info['age']}\n♀ Gender: {user_info['gender']}\n🌐 Language: {user_info['language']}\n Guardian_info: {user_info['Guardian_info']}"
+# Keywords to detect context quickly
+context_markers = {
+    "school": ["school", "exam", "test", "teacher", "class"],
+    "parents": ["parent", "mom", "dad", "parents", "home"],
+    "friend": ["friend", "friends", "bully", "shouted", "argument", "fight"],
+    "health": ["sick", "unwell", "hospital", "doctor"],
+    "love": ["crush", "boyfriend", "girlfriend", "heartbroken", "love"],
+    # add more markers as needed
+}
 
-def generate_reply(input_text):
-    text = input_text.lower()
-    for word in harmful_keywords:
-        if word in text:
-            return harmful_response
-    for word, reply in emotions.items():
-        if word in text:
-            return reply
-    return "Tell me more about your day."
+# Follow-up templates (3 steps each, then action suggestions)
+followups = {
+    "sad": [
+        "I’m really sorry you’re feeling sad. Can you tell me if this happened recently or is it been building up?",
+        "Thank you for sharing. How did that make you feel — more hurt, embarrassed, or something else?",
+        "That sounds heavy. Would you like a calming exercise, to write about it in the journal, or to talk more?"
+    ],
+    "anxiety": [
+        "Anxiety can be overwhelming. Is it about something specific like exams or relationships?",
+        "When your anxiety spikes, do you notice physical signs like a racing heart or trouble breathing?",
+        "Would you like to try a short breathing exercise now, or do you want to talk about what’s triggering it?"
+    ],
+    "angry": [
+        "Anger is a valid response. Do you want to share what caused this anger?",
+        "When you feel angry, what usually helps — space, talking, or doing something active?",
+        "Would you like ideas to safely express the anger, or would you prefer calming techniques first?"
+    ],
+    # fallback for other emotions:
+    "default": [
+        "Tell me more about that — when did you start feeling this way?",
+        "How strong does this feeling feel on a scale of 1 to 10?",
+        "Would you like a calming activity, a journal prompt, or to continue talking?"
+    ]
+}
+
+# Reset phrases (user indicates they are fine or want to change topic)
+reset_phrases = [
+    "i'm fine", "i am fine", "i'm okay", "i am okay", "i feel better",
+    "thank you", "thanks", "that's all", "let's talk about something else",
+    "stop", "no", "I'm good", "im good"
+]
 
+# ----------------------------
+# Helper functions
+# ----------------------------
+def _contains_any(text, keywords):
+    text = text.lower()
+    for kw in keywords:
+        if kw in text:
+            return True
+    return False
+
+def detect_harmful(text):
+    t = text.lower()
+    for kw in harmful_keywords:
+        if kw in t:
+            return True
+    return False
+
+def detect_emotion_in_text(text):
+    t = text.lower()
+    # Prefer longer keys first (to avoid 'sad' matching inside other words) — optional improvement
+    sorted_keys = sorted(emotions.keys(), key=lambda x: -len(x))
+    for key in sorted_keys:
+        if key in t:
+            return key
+    return None
+
+def detect_context_in_text(text):
+    t = text.lower()
+    for ctx, markers in context_markers.items():
+        for m in markers:
+            if m in t:
+                return ctx
+    return None
+
+def is_reset_phrase(text):
+    t = text.lower()
+    for phrase in reset_phrases:
+        if phrase in t:
+            return True
+    return False
+
+def build_empathy_reply(emotion):
+    # Base empathy from emotions dict, plus a gentle follow-up question from followups
+    base = emotions.get(emotion, "")
+    # Pick a gentle opener if no followups defined
+    return base
+
+# ----------------------------
+# Upgraded reply generator with memory, context, multi-turn flow
+# ----------------------------
+def generate_reply(input_text):
+    text = input_text.strip()
+    if not text:
+        return "Please say or type something when you are ready."
+
+    lower = text.lower()
+
+    # 1) Immediate harmful check (escalate without delay)
+    if detect_harmful(lower):
+        # Reset conversation state because we escalate
+        conversation_state["emotion"] = None
+        conversation_state["context"] = None
+        conversation_state["followup_stage"] = 0
+        conversation_state["turns_since_set"] = 0
+        return harmful_response
+
+    # 2) User reset intent — end current flow politely
+    if is_reset_phrase(lower):
+        conversation_state["emotion"] = None
+        conversation_state["context"] = None
+        conversation_state["followup_stage"] = 0
+        conversation_state["turns_since_set"] = 0
+        return "That's ok — we can pause here. If you'd like to talk again later, I'm right here."
+
+    # 3) Detect if user explicitly states a new emotion -> override current state
+    new_emotion = detect_emotion_in_text(lower)
+    if new_emotion and new_emotion != conversation_state["emotion"]:
+        # start new flow
+        conversation_state["emotion"] = new_emotion
+        conversation_state["context"] = detect_context_in_text(lower)
+        conversation_state["followup_stage"] = 0
+        conversation_state["turns_since_set"] = 0
+        # Compose reply: empathy + first follow-up
+        empathy = build_empathy_reply(new_emotion)
+        # get followup prompt
+        prompts = followups.get(new_emotion, followups["default"])
+        follow = prompts[0] if prompts else ""
+        # Combine neatly
+        combined = f"{empathy}\n\n{follow}"
+        return combined
+
+    # 4) If an emotion is already active, continue that flow
+    if conversation_state["emotion"]:
+        emotion = conversation_state["emotion"]
+        # Update context if we find context in this message
+        ctx = detect_context_in_text(lower)
+        if ctx:
+            conversation_state["context"] = ctx
+
+        # If the user provided details (short heuristic: message length > 20 or contains certain markers),
+        # treat this as progress and advance the follow-up stage
+        provided_detail = (len(lower) > 20) or any(w in lower for w in ["because", "so", "then", "after", "today", "yesterday", "last", "friend", "parents", "teacher", "exam", "test"])
+
+        stage = conversation_state["followup_stage"]
+        prompts = followups.get(emotion, followups["default"])
+
+        # If user asked a question (contains '?'), attempt to respond empathetically referencing emotion/context
+        if "?" in lower:
+            # Respond referencing emotion and context
+            ctx_text = f" about {conversation_state['context']}" if conversation_state["context"] else ""
+            response = f"I hear your question{ctx_text}. Let's look at that together. {emotions.get(emotion,'')}"
+            # do not advance stage on simple question
+            conversation_state["turns_since_set"] += 1
+            return response
+
+        if provided_detail:
+            # Provide a reply that references the emotion and context and then advance
+            ctx_text = ""
+            if conversation_state["context"]:
+                ctx_text = f" because of {conversation_state['context']}"
+            # Construct reflective reply
+            if stage < len(prompts):
+                response = f"I understand — it makes sense you'd feel {emotion}{ctx_text}. {prompts[stage]}"
+                conversation_state["followup_stage"] += 1
+                conversation_state["turns_since_set"] = 0
+                return response
+            else:
+                # Follow-ups exhausted; offer actionable choices
+                conversation_state["turns_since_set"] += 1
+                return ("You have shared a lot — would you like to: \n"
+                        "1) Try a calming exercise now\n"
+                        "2) Write this in your journal\n"
+                        "3) Talk more about it\n\n"
+                        "Type '1', '2', or '3' to choose, or tell me if you want a helpline.")
+        else:
+            # User gave a short reply — ask a clarifying / gentle follow-up without advancing too far
+            if stage < len(prompts):
+                response = prompts[stage]
+                # only advance if we already showed that prompt previously
+                conversation_state["turns_since_set"] += 1
+                # If we had asked the same prompt previously and user still gives short replies, advance stage
+                if conversation_state["turns_since_set"] > 1:
+                    conversation_state["followup_stage"] = min(len(prompts), stage + 1)
+                    conversation_state["turns_since_set"] = 0
+                return response
+            else:
+                # if stages done but no detail, ask for choice
+                return ("I’m listening. If it's hard to describe, choose: 1) calming exercise 2) journal 3) talk more. "
+                        "Or say 'I'm fine' to stop.")
+    # 5) No active emotion and no new emotion detected -> try to detect context or ask to elaborate
+    ctx = detect_context_in_text(lower)
+    if ctx:
+        # Gentle probing
+        return f"You mentioned {ctx}. Is this making you feel upset or worried? How does it make you feel?"
+    # Default fallback
+    return "Tell me more about your day — what happened, or how are you feeling right now?"
+
+# ----------------------------
+# Chat function (uses TTS + conversation_state)
+# ----------------------------
 def chat_function(audio_input, text_input):
-    user_text = text_input.strip()
+    # We ignore audio_input here for text-only flow; you can add speech-to-text to use it.
+    user_text = (text_input or "").strip()
     if not user_text:
         return "Please type something.", None
+
     reply = generate_reply(user_text)
-    chat_history.append({"user": user_text, "bot": reply})
+
+    # Save conversation to history
+    chat_history.append({"user": user_text, "bot": reply, "emotion": conversation_state.get("emotion"), "context": conversation_state.get("context")})
+
+    # Text-to-speech (language selection)
     lang_code = lang_codes.get(user_info["language"], "en")
-    tts = gTTS(reply, lang=lang_code)
-    audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
-    tts.save(audio_file.name)
-    return reply, audio_file.name
+    try:
+        tts = gTTS(reply, lang=lang_code)
+        audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+        tts.save(audio_file.name)
+        voice_path = audio_file.name
+    except Exception as e:
+        # If TTS fails, still return the reply and no voice
+        voice_path = None
+
+    return reply, voice_path
+
+# ----------------------------
+# Remaining utility functions (unchanged)
+# ----------------------------
+def set_personal_info(name, age, gender, language, Guardian_info):
+    user_info.update({"name": name, "age": age, "gender": gender, "language": language, "Guardian_info": Guardian_info,})
+    return gr.update(visible=True), f"✅ Welcome {name}! Preferences saved."
+
+def show_personal_data():
+    today = datetime.date.today().strftime("%Y-%m-%d (%A)")
+    return f"📅 {today}\n👤 Name: {user_info['name']}\n🎂 Age: {user_info['age']}\n♀ Gender: {user_info['gender']}\n🌐 Language: {user_info['language']}\n Guardian_info: {user_info['Guardian_info']}"
 
 def get_chat_history():
     if not chat_history:
@@ -396,7 +334,9 @@ def next_study_tip():
     tip_index["study"] += 1
     return tip
 
-# -------------- Gradio App --------------
+# ----------------------------
+# Gradio UI (kept same as your original file)
+# ----------------------------
 with gr.Blocks() as app:
     welcome_screen = gr.Column(visible=True)
     full_app = gr.Tabs(visible=False)
@@ -431,7 +371,7 @@ with gr.Blocks() as app:
                 }
             </style>
         """)
-        gr.Markdown("<h1 id='main-title'>StrongMind Therapist 2.0</h1>")
+        gr.Markdown("<h1 id='main-title'>StrongMind Therapist 3.0</h1>")
         gr.Markdown("<h3 id='subtitle'>Your peaceful space to talk, journal, and focus.</h3>")
         start_button = gr.Button("🌸 Get Started", elem_id="start-btn")
 
@@ -464,7 +404,6 @@ with gr.Blocks() as app:
             bot_reply = gr.Textbox(label="🧠 Therapist")
             voice = gr.Audio(label="🔊 Voice Reply")
             send.click(chat_function, [audio_input, text_input], [bot_reply, voice])
-        
 
         with gr.Tab("4️⃣ Chat History"):
             show_history = gr.Button("📜 Show Chats")