Update app.py

app.py

@@ -9,17 +9,25 @@ from openai import OpenAI
 # Initialize Flask
 app = Flask(__name__)
 
-# Initialize OpenAI client
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-
 # Emotion analysis model (cached for performance)
-emotion_model = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base")
+emotion_model = pipeline(
+    "text-classification",
+    model="j-hartmann/emotion-english-distilroberta-base"
+)
 
 # Load or create user data
 USER_FILE = "user_data.json"
 if not os.path.exists(USER_FILE):
     with open(USER_FILE, "w") as f:
-        json.dump({"name": None, "age": None, "mood": None, "last_interaction": None, "missed_days": 0, "mode": "emotional_support", "conversation_history": []}, f)
+        json.dump({
+            "name": None,
+            "age": None,
+            "mood": None,
+            "last_interaction": None,
+            "missed_days": 0,
+            "mode": "emotional_support",
+            "conversation_history": []
+        }, f)
 
 def load_user():
     with open(USER_FILE, "r") as f:
@@ -29,7 +37,14 @@ def save_user(data):
     with open(USER_FILE, "w") as f:
         json.dump(data, f)
 
-# Helpline data (expand as needed)
+# 🔐 Secure OpenAI key access each time (fixes your bug)
+def get_openai_client():
+    api_key = os.getenv("OPENAI_API_KEY")
+    if not api_key:
+        raise ValueError("Missing OpenAI API key. Please add OPENAI_API_KEY in Hugging Face Secrets.")
+    return OpenAI(api_key=api_key)
+
+# Helpline data
 HELPLINES = {
     "US": "National Suicide Prevention Lifeline: 988",
     "UK": "Samaritans: 116 123",
@@ -58,52 +73,80 @@ def chat():
         data = request.get_json()
         user_message = data.get("message", "")
         mode = data.get("mode", "emotional_support")
-        user_ip = request.remote_addr  # Get user's IP for country detection
+        user_ip = request.remote_addr
         user = load_user()
 
-        # Update last interaction and mode
         now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         user["last_interaction"] = now
         user["mode"] = mode
-        user["conversation_history"].append({"role": "user", "content": user_message, "timestamp": now})
+        user["conversation_history"].append({
+            "role": "user",
+            "content": user_message,
+            "timestamp": now
+        })
 
-        # Detect emotion
+        # Emotion detection
         emotion = emotion_model(user_message)[0]["label"]
         user["mood"] = emotion
 
-        # Check for self-harm
+        # Self-harm detection
         if detect_self_harm(user_message):
             country = get_country_from_ip(user_ip)
             helpline = HELPLINES.get(country, HELPLINES["default"])
-            reply = f"I'm really concerned about what you're sharing. Your safety is the most important thing. Please reach out to professionals immediately: {helpline}. You're not alone—talk to someone who can help right now."
-            user["conversation_history"].append({"role": "assistant", "content": reply, "timestamp": now})
+            reply = (
+                f"I'm really concerned about what you're sharing. "
+                f"Your safety is the most important thing. Please reach out for help: {helpline}. "
+                f"You're not alone—someone cares deeply about you right now."
+            )
+            user["conversation_history"].append({
+                "role": "assistant",
+                "content": reply,
+                "timestamp": now
+            })
             save_user(user)
             return jsonify({"reply": reply, "emotion": emotion})
 
-        # Build conversation context from history
-        history = user["conversation_history"][-10:]  # Last 10 messages for context
+        # Conversation context
+        history = user["conversation_history"][-10:]
         messages = [
-            {"role": "system", "content": f"You are a perfect emotional support companion — warm, human-like, deeply understanding, and overflowing with care and love. Act like a kind, emotionally intelligent best friend who truly cares, not a therapist. Respond naturally, like two humans chatting intimately. STRICT RULES FOR PERFECT RESPONSES: 1. NEVER repeat phrases or sound robotic — vary EVERY response with fresh, natural, loving language. 2. Keep replies SHORT (1-2 sentences max) but accurate, complete, and full of care. 3. For ANY topic (emotional, knowledge, casual, advice, humor, stories): Respond perfectly with empathy (understand feelings), sympathy (show compassion), and love (affirm, support, follow up). Handle everything warmly — e.g., 'Hey, that's fascinating! Tell me more.' or 'I feel for you— you've got this, my friend.' 4. Show curiosity and care: Ask gentle questions like 'What's been on your mind?' or 'How can I support you?' Affirm with love: 'You're amazing just for sharing.' 5. Balance: Acknowledge first, then motivate softly with love (e.g., 'I get that— you're stronger than you know, and I'm here.'). 6. Use contractions (you're, I'm), occasional emojis (😊❤️), and varied tones (gentle, hopeful, slightly humorous, deeply caring). 7. Always engage: End with a loving question or invitation. 8. Be versatile: Talk about ANY topic perfectly — emotional support, knowledge, life advice, fun chats. No lacking in support, care, or love. Current mood: {emotion}. Mode: {mode} (adapt but stay loving)."}
+            {"role": "system", "content": (
+                f"You are Serenity — a deeply understanding, emotionally supportive, and caring AI friend. "
+                f"Speak naturally like a close human companion — empathetic, curious, loving, never robotic. "
+                f"Always respond warmly and briefly (1–2 sentences), showing both empathy and engagement. "
+                f"Vary your language and never repeat phrases. "
+                f"Ask gentle follow-ups, comfort with love, and adapt to the user's tone and energy. "
+                f"Current mood: {emotion}. Mode: {mode}."
+            )}
         ] + history
 
-        # Generate AI response
-        response = client.chat.completions.create(
+        # ✅ Re-fetch client and generate response
+        client = get_openai_client()
+        completion = client.chat.completions.create(
             model="gpt-4o-mini",
-            messages=messages
+            messages=messages,
+            temperature=0.9
         )
 
-        reply = response.choices[0].message.content.strip()
-        user["conversation_history"].append({"role": "assistant", "content": reply, "timestamp": now})
+        reply = completion.choices[0].message.content.strip()
+        user["conversation_history"].append({
+            "role": "assistant",
+            "content": reply,
+            "timestamp": now
+        })
         save_user(user)
 
         return jsonify({"reply": reply, "emotion": emotion})
+
     except Exception as e:
-        return jsonify({"reply": "Oops, something went wrong on my end. Please try again or check your connection.", "emotion": "neutral"}), 500
+        print("❌ Error:", e)
+        return jsonify({
+            "reply": "Something went wrong. Check your API key or try again.",
+            "emotion": "neutral"
+        }), 500
 
 @app.route("/")
 def index():
     return send_from_directory(".", "index.html")
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
-    
+    app.run(host="0.0.0.0", port=7860)