Update it.

app.py

@@ -5,6 +5,7 @@ from datetime import datetime, timedelta
 from flask import Flask, request, jsonify, send_from_directory
 from transformers import pipeline
 from openai import OpenAI
+import requests
 
 app = Flask(__name__)
 
@@ -14,7 +15,6 @@ emotion_model = pipeline("text-classification", model="j-hartmann/emotion-englis
 USER_FILE = "user_data.json"
 CHAT_LOG_FILE = "recent_chat.json"
 
-# Initialize files if not present
 if not os.path.exists(USER_FILE):
     with open(USER_FILE, "w") as f:
         json.dump({
@@ -42,12 +42,27 @@ def load_chat_log():
         return json.load(f)
 
 def save_chat_log(logs):
-    # Keep only last 15 days
     cutoff = datetime.now() - timedelta(days=15)
     logs = [l for l in logs if datetime.strptime(l["time"], "%Y-%m-%d %H:%M:%S") > cutoff]
     with open(CHAT_LOG_FILE, "w") as f:
         json.dump(logs, f)
 
+def get_country():
+    try:
+        res = requests.get("https://ipapi.co/json/").json()
+        return res.get("country_name", "default").lower()
+    except:
+        return "default"
+
+HELPLINES = {
+    "united states": "If you are in the U.S., please contact the **988 Suicide and Crisis Lifeline** by dialing 988 (24/7).",
+    "united kingdom": "If you’re in the UK, contact **Samaritans** at 116 123 (24/7).",
+    "india": "If you’re in India, you can reach **AASRA** at 91-9820466726 or **Snehi** at 91-9582208181.",
+    "canada": "If you’re in Canada, dial or text **988** for the Talk Suicide Canada line (24/7).",
+    "australia": "If you’re in Australia, contact **Lifeline Australia** at 13 11 14 (24/7).",
+    "default": "If you’re in crisis, please reach out to your local emergency number or trusted mental health service. You are not alone 💛"
+}
+
 MOTIVATIONAL_QUOTES = [
     "Even the darkest night will end and the sun will rise 🌅",
     "You are stronger than you think 💪",
@@ -62,13 +77,6 @@ CRISIS_KEYWORDS = [
     "suicide", "kill myself", "end my life", "want to die", "can't go on", "life not worth"
 ]
 
-HELPLINES = {
-    "us": "If you are in the U.S., please contact the **988 Suicide and Crisis Lifeline** by dialing 988 (24/7).",
-    "uk": "If you’re in the UK, contact **Samaritans** at 116 123 (24/7).",
-    "india": "If you’re in India, you can reach **AASRA** at 91-9820466726 or **Snehi** at 91-9582208181.",
-    "default": "If you’re in crisis, please reach out to your local emergency number or trusted mental health service. You are not alone 💛"
-}
-
 @app.route("/chat", methods=["POST"])
 def chat():
     data = request.get_json()
@@ -76,23 +84,24 @@ def chat():
     user = load_user()
     chat_log = load_chat_log()
 
-    # 1️⃣ Suicide / Crisis detection
+    # Crisis detection
     if any(keyword in user_message for keyword in CRISIS_KEYWORDS):
+        country = get_country()
         crisis_msg = (
-            "I’m really concerned about how you’re feeling right now 💛.\n"
+            "💛 I’m really concerned about how you’re feeling right now.\n"
             "You don’t have to face this alone. "
-            + HELPLINES.get("india") +  # default to India for now (can localize later)
-            "\n\nWould you like me to stay with you while you calm down a bit?"
+            + HELPLINES.get(country, HELPLINES["default"])
+            + "\n\nWould you like me to stay with you while you calm down a bit?"
         )
         return jsonify({"reply": crisis_msg, "emotion": "crisis"})
 
-    # 2️⃣ Collect name
+    # Collect name
     if not user["name"]:
         user["name"] = user_message.capitalize()
         save_user(user)
         return jsonify({"reply": f"Nice to meet you, {user['name']} 🌼 Can you tell me your age?", "emotion": "calm"})
 
-    # 3️⃣ Collect age
+    # Collect age
     if not user["age"]:
         try:
             user["age"] = int(user_message)
@@ -101,27 +110,25 @@ def chat():
         except ValueError:
             return jsonify({"reply": "Could you please tell me your age in numbers? 😊", "emotion": "neutral"})
 
-    # 4️⃣ Follow-up if user felt bad last time
+    # Follow-up if user felt bad last time
     if user["last_interaction"]:
         last_time = datetime.strptime(user["last_interaction"], "%Y-%m-%d %H:%M:%S")
         if datetime.now() - last_time > timedelta(hours=12):
-            # find most recent emotion
             if user["mood"] in ["sadness", "anger", "fear"]:
                 follow_up = f"Hey {user['name']} 💛 You were feeling low last time we talked. How are you feeling now?"
                 user["last_interaction"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
                 save_user(user)
                 return jsonify({"reply": follow_up, "emotion": "warm"})
 
-    # 5️⃣ Emotion analysis
+    # Emotion detection
     emotion_result = emotion_model(user_message)[0]
     emotion = emotion_result["label"].lower()
 
-    # Update user state + chat logs
+    # Update logs
     user["mood"] = emotion
     user["last_interaction"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     user["mood_history"].append({"emotion": emotion, "time": user["last_interaction"]})
     save_user(user)
-
     chat_log.append({
         "message": user_message,
         "emotion": emotion,
@@ -129,21 +136,19 @@ def chat():
     })
     save_chat_log(chat_log)
 
-    # 6️⃣ AI tone setup
+    # Tone setup
     tone_instruction = (
         "You are Serenity, an emotionally intelligent AI companion. "
-        "Speak in a warm, mindful, and compassionate way, showing you care and remember past feelings. "
-        "Avoid repeating 'I'm sorry to hear that'. Use variations like 'That must have been difficult', 'You’ve been strong', or 'I understand how that could feel'. "
-        "If user feels very upset, respond with gentle reassurance and care. "
-        "Keep your messages conversational and not robotic."
+        "Be warm, human-like, calm, and empathetic. Show that you care deeply and remember feelings. "
+        "Do not repeat 'I'm sorry to hear that'; instead vary your empathetic responses naturally. "
+        "Encourage reflection and hope. "
     )
 
     if user["age"] < 18:
-        tone_instruction += " Speak softly and encouragingly, like a kind friend or mentor."
+        tone_instruction += "Speak in a gentle, caring tone suited for teenagers."
     else:
-        tone_instruction += " Respond with emotional maturity, wisdom, and warmth."
+        tone_instruction += "Speak with emotional maturity and calm wisdom."
 
-    # 7️⃣ Generate AI response
     ai_response = client.chat.completions.create(
         model="gpt-4o-mini",
         messages=[
@@ -151,13 +156,19 @@ def chat():
             {"role": "user", "content": user_message},
         ],
     )
-
     reply = ai_response.choices[0].message.content.strip()
 
-    # 8️⃣ Motivational quote for negative emotions
+    # Add motivational quote if sad
     if emotion in ["sadness", "fear", "anger"]:
-        quote = random.choice(MOTIVATIONAL_QUOTES)
-        reply += f"\n\n💬 *Motivational Thought:* {quote}"
+        reply += f"\n\n💬 *Motivational Thought:* {random.choice(MOTIVATIONAL_QUOTES)}"
+
+    # Mood trend summary
+    if len(user["mood_history"]) > 3:
+        recent = [m['emotion'] for m in user["mood_history"][-3:]]
+        if all(e in ["happiness", "joy", "calm"] for e in recent):
+            reply += "\n\n🌞 You’ve been calmer lately — I’m proud of your progress!"
+        elif all(e in ["sadness", "fear", "anger"] for e in recent):
+            reply += "\n\n🌧️ I’ve noticed some tough days lately — please remember you’re not alone."
 
     return jsonify({"reply": reply, "emotion": emotion})