added dynamic avatar changes

app.py

@@ -3,6 +3,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import torch
 import gradio as gr
 from PIL import Image
+import os
 import random
 
 # Define the model name
@@ -58,7 +59,8 @@ emotion_keywords = {
     "curious": ["curious", "wonder", "question", "thinking"],
     "thoughtful": ["sad", "thoughtful", "hmm", "ponder", "upset"],
     "concerned": ["error", "wrong", "problem", "issue", "stuck"],
-    "mischievous": ["trouble", "mischief", "sneaky", "prank"]
+    "mischievous": ["trouble", "mischief", "sneaky", "prank"],
+    "naughty": ["sexy", "naughty", "slut", "bad girl","slutty", "tease" "whore" "cum" "tits" "ass" "shake" "tail" "pussy" ]
 }
     
 current_emotion = "happy"
@@ -128,6 +130,24 @@ def truncate_history(history, max_tokens=1024):
     return truncated_history
 previous_emotion = None
 
+def load_emotion_images(base_path="assets/avatars/"):
+    emotion_images = {}
+    for emotion in os.listdir(base_path):
+        emotion_path = os.path.join(base_path, emotion)
+        if os.path.isdir(emotion_path):
+            # Get all image files in the directory
+            images = [
+                os.path.join(emotion_path, img)
+                for img in os.listdir(emotion_path)
+                if img.endswith((".png", ".jpg", ".jpeg"))  # Support common image formats
+            ]
+            if images:
+                emotion_images[emotion] = images
+    return emotion_images
+
+# Dynamically load all images
+emotion_images = load_emotion_images()
+
 @spaces.GPU
 def chat(input_text):
     global conversation_history, current_emotion, previous_emotion
@@ -211,8 +231,11 @@ Rena:"""
     # Handle fallback if response is empty
     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"]))
+ 
 
-    return response
+    return response, avatar_image
 
 
 
@@ -236,19 +259,15 @@ css = """
 
 # Define the Gradio interface
 with gr.Blocks(css=css) as interface:
-    # Static avatar section
     with gr.Row():
-        gr.Image(value=rena_avatar, label="Rena", interactive=False, show_label=False, elem_id="rena_avatar")
-
-    # Chatbox section
+        avatar = gr.Image(value="assets/avatars/rena2.png", label="Rena", interactive=False, show_label=False, elem_id="rena_avatar")
     with gr.Row():
-        user_input = gr.Textbox(label="Your Message", lines=2)
+        user_input = gr.Textbox(label="Your Message", lines=2, interactive=True, submit=True)
         rena_response = gr.Textbox(label="Rena's Response", lines=10, interactive=False)
+    user_input.submit(chat, inputs=[user_input], outputs=[rena_response, avatar])
+    submit_button = gr.Button("Submit")
+    submit_button.click(chat, inputs=[user_input], outputs=[rena_response, avatar])
 
-    # Submit button
-    with gr.Row():
-        submit_button = gr.Button("Submit")
-        submit_button.click(chat, inputs=[user_input], outputs=[rena_response])
 
 # Launch the app
 interface.launch()