concerned sterak emotion

app.py

@@ -180,10 +180,11 @@ def get_emotion_image(emotion):
 def remove_emojis(text):
     return re.sub(r'[^\w\s,.\'\"!?]', '', text)
 
+concerned_streak = 0  
 
 @spaces.GPU
 def chat(input_text):
-    global conversation_history, current_emotion, previous_emotion
+    global conversation_history, current_emotion, previous_emotion, concerned_streak
 
     # Add user input to the conversation history
     conversation_history.append(f"User: {input_text}")
@@ -195,6 +196,15 @@ def chat(input_text):
     previous_emotion = current_emotion
     current_emotion = analyze_history(conversation_history)
 
+    if current_emotion == "concerned":
+        concerned_streak += 1
+    else:
+        concerned_streak = max(0, concerned_streak - 1)
+
+    if concerned_streak >= 3:
+        current_emotion = random.choice(["happy", "thoughtful", "playful"])
+        concerned_streak = 0  
+
     # Update the avatar image if the emotion changes
     avatar_image = get_emotion_image(current_emotion)
 
@@ -238,16 +248,7 @@ Rena:"""
         if emotional_prefix and not response.startswith(emotional_prefix):
             response = f"{emotional_prefix} {response}".strip()
 
-    if current_emotion == "concerned":
-        concerned_streak += 1
-    else:
-        concerned_streak = max(0, concerned_streak - 1)
-
-    if concerned_streak >= 3:
-        current_emotion = random.choice(["happy", "thoughtful", "playful"])  # ✅ Break the loop
-        concerned_streak = 0
-
-    # Final cleanup: Ensure no "User:" or unintended artifacts remain
+        # Final cleanup: Ensure no "User:" or unintended artifacts remain
     response = response.replace("User:", "").strip()
 
     # Handle specific inputs