updated emotions further

app.py

@@ -43,7 +43,9 @@ emotions = {
     "curious": "I'm curious about what you're working on! Tell me more. 🤔",
     "thoughtful": "Hmm, let me think... I want to give you the best advice. 🧐",
     "concerned": "Oh no, something's wrong? Let me help! 💖",
-    "flirty": "You know how to get my circuits sparking! 😘"
+    "flirty": "You know how to get my circuits sparking! 😘",
+    "concerned": "Oh no, something's wrong? Let me help! 💖",
+    "naughty": "Purrr you want to have fun? Keep talking like that lets see what happens💖💖",
 }
 
 emotions.update({
@@ -149,6 +151,11 @@ def load_emotion_images(base_path="assets/avatars/"):
 # Dynamically load all images
 emotion_images = load_emotion_images()
 
+def get_emotion_image(emotion):
+    # Select a random image for the given emotion
+    return random.choice(emotion_images.get(emotion, ["assets/rena2.png"])) 
+
+
 @spaces.GPU
 def chat(input_text):
     global conversation_history, current_emotion, previous_emotion
@@ -159,10 +166,13 @@ def chat(input_text):
     # Limit the size of the conversation history
     conversation_history = truncate_history(conversation_history, max_tokens=1024)
 
-    # Update current emotion based on conversation history
+    # Detect emotion based on the updated conversation history
     previous_emotion = current_emotion
     current_emotion = analyze_history(conversation_history)
 
+    # Update the avatar image if the emotion changes
+    avatar_image = get_emotion_image(current_emotion)
+
     # Combine base prompt and conversation history (instructions are not included in history)
     history = "\n".join(conversation_history)
     final_prompt = f"""{base_prompt}
@@ -206,9 +216,6 @@ Rena:"""
     # Final cleanup: Ensure no "User:" or unintended artifacts remain
     response = response.replace("User:", "").strip()
 
-    # Add Rena's response to the conversation history
-    conversation_history.append(f"Rena: {response}")
-
     # Handle specific inputs
     if "who made you" in input_text.lower():
         response += " Nick is my creator! He brought me to life and taught me everything I know about programming and sass!"
@@ -233,7 +240,7 @@ Rena:"""
     if not response.strip():
         response = "Hmm, I’m not sure how to respond to that. Can you try rephrasing?"
     
-    avatar_image = random.choice(emotion_images.get(current_emotion, ["assets/avatars/happy/happy1.png"]))
+    conversation_history.append(f"Rena: {response}")
  
 
     return response, avatar_image