Fixed model behaviour

app.py

@@ -103,20 +103,18 @@ def respond(message, history):
         # Tokenize
         input_ids = tokenizer.encode(conversation, return_tensors="pt")
         
-        # Generate response with conservative settings for DialoGPT
+        # Generate response with improved settings for more creative, less repetitive responses
         with torch.no_grad():
             chat_history_ids = model.generate(
                 input_ids,
-                max_length=input_ids.shape[-1] + 50,  # Shorter responses
-                num_beams=5,
-                no_repeat_ngram_size=3,
-                do_sample=True,
-                early_stopping=True,
-                pad_token_id=tokenizer.eos_token_id,
-                eos_token_id=tokenizer.eos_token_id,
+                max_length=1250,
                 temperature=0.9,
+                top_p=0.9,
+                repetition_penalty=1.2,
+                do_sample=True,
                 top_k=50,
-                top_p=0.9
+                pad_token_id=tokenizer.eos_token_id,
+                no_repeat_ngram_size=3
             )
         
         # Decode only the new response
@@ -138,42 +136,143 @@ def respond(message, history):
         return "I hear you, and I want you to know that I'm here for you. Sometimes I need a moment to find the right words."
 
 def add_empathy_to_response(response, user_message):
-    """Add Aura's empathetic touch to the raw response"""
-    # Detect emotional keywords to tailor empathy
-    emotional_words = ['sad', 'angry', 'frustrated', 'worried', 'scared', 'lonely', 'hurt', 'upset', 'stressed']
+    """Add Aura's empathetic touch to the raw response with high variety"""
+    import random
+    
+    # Clean up the response first
+    cleaned_response = response.replace("Human:", "").replace("Aura:", "").strip()
+    
+    # Detect context and emotions to provide varied, appropriate responses
     user_lower = user_message.lower()
     
-    empathetic_starters = [
-        "I hear you, and ",
-        "That sounds really difficult. ",
-        "I can imagine that feels heavy. ",
-        "Thank you for sharing that with me. "
+    # Context-specific empathetic starters with high variety
+    if "not good enough" in user_lower or "failure" in user_lower or "worthless" in user_lower:
+        starters = [
+            "Those feelings are so valid, and I want you to know that ",
+            "It takes real courage to share those feelings with me. ",
+            "I can hear how much pain you're carrying right now. ",
+            "That inner voice can be so harsh sometimes. ",
+            "What you're feeling makes complete sense, and "
+        ]
+    elif "tough day" in user_lower or "bad day" in user_lower or "difficult" in user_lower:
+        starters = [
+            "Some days really are harder than others, aren't they? ",
+            "It sounds like today has been especially challenging. ",
+            "Days like this can feel so heavy. ",
+            "I'm sorry you're going through such a rough time. ",
+            "Thank you for trusting me with how you're feeling today. "
+        ]
+    elif "overwhelmed" in user_lower or "stressed" in user_lower:
+        starters = [
+            "That feeling of overwhelm can be so intense. ",
+            "It sounds like there's a lot weighing on you right now. ",
+            "When everything feels like too much, it's really hard. ",
+            "I can imagine how exhausting that must feel. ",
+            "Sometimes life piles up in ways that feel impossible to manage. "
+        ]
+    elif "listen" in user_lower or "someone" in user_lower:
+        starters = [
+            "I'm right here with you, and I'm listening. ",
+            "You have my full attention, and I want you to know ",
+            "Thank you for reaching out. I'm here, and ",
+            "Sometimes we all need someone to just be present with us. ",
+            "I'm grateful you felt safe sharing this with me. "
+        ]
+    elif any(word in user_lower for word in ['hurt', 'pain', 'sad', 'crying']):
+        starters = [
+            "I can hear how much you're hurting right now. ",
+            "Pain like this is so real and so valid. ",
+            "It takes strength to sit with feelings like these. ",
+            "Your pain matters, and I want you to know ",
+            "I'm holding space for everything you're feeling. "
+        ]
+    elif "fractured" in user_lower or "injury" in user_lower or "hurt" in user_lower:
+        starters = [
+            "Oh no, that sounds incredibly painful! ",
+            "I'm so sorry that happened to you. ",
+            "That must be really scary and painful. ",
+            "Injuries like that are no small thing. ",
+            "I can only imagine how much that must hurt. "
+        ]
+    else:
+        # General varied empathetic responses
+        starters = [
+            "Thank you for sharing that with me. ",
+            "I can hear that this is important to you. ",
+            "It sounds like you're going through something difficult. ",
+            "I'm here with you in this moment. ",
+            "What you're experiencing sounds really challenging. ",
+            "I want you to know that your feelings are completely valid. ",
+            "It takes courage to open up like this. ",
+            "I'm grateful you felt safe sharing this with me. "
+        ]
+    
+    # Choose a random starter for variety
+    starter = random.choice(starters)
+    
+    # Add thoughtful questions sometimes to show deeper engagement
+    questions = [
+        " What's been the hardest part for you?",
+        " How long have you been carrying this?",
+        " Is there anything that's helped, even a little?",
+        " What do you need most right now?",
+        " How are you taking care of yourself through this?",
+        ""
     ]
     
-    # Choose appropriate empathetic starter
-    if any(word in user_lower for word in emotional_words):
-        starter = "I can imagine that feels really difficult. "
-    elif "not good enough" in user_lower or "failure" in user_lower:
-        starter = "I hear you, and those feelings are so valid. "
-    else:
-        starter = "I hear you. "
+    # Add question 30% of the time for engagement variety
+    question = random.choice(questions) if random.random() < 0.3 else ""
     
-    # Clean up the response and add empathy
-    cleaned_response = response.replace("Human:", "").replace("Aura:", "").strip()
-    return f"{starter}{cleaned_response}"
+    # Combine everything
+    if cleaned_response:
+        return f"{starter}{cleaned_response}{question}"
+    else:
+        return f"{starter}I'm here to listen to whatever you need to share{question}"
 
 def get_fallback_aura_response(user_message):
-    """Provide appropriate fallback responses based on user input"""
+    """Provide appropriate fallback responses based on user input with variety"""
+    import random
     user_lower = user_message.lower()
     
     if "not good enough" in user_lower:
-        return "I hear you, and I want you to know that those feelings of not being good enough are so valid. It takes courage to share that."
+        responses = [
+            "Those feelings of not being good enough are so painful, and I want you to know they're completely valid. It takes real courage to share something so vulnerable.",
+            "I can hear how much you're struggling with those feelings about yourself. That inner critic can be so harsh, can't it?",
+            "What you're feeling makes complete sense, and I'm grateful you trusted me with something so personal. You're not alone in feeling this way.",
+            "That's such a heavy feeling to carry. I want you to know that sharing this with me shows real strength, even when you don't feel strong."
+        ]
+        return random.choice(responses)
     elif "what" in user_lower and "do" in user_lower:
-        return "I can hear that you're looking for some direction. Rather than telling you what to do, I'm wondering - what feels most important to you right now?"
+        responses = [
+            "I can hear that you're looking for some direction. Rather than telling you what to do, I'm wondering - what feels most important to you right now?",
+            "It sounds like you're feeling a bit lost about next steps. Sometimes when we're overwhelmed, it helps to start with what you're feeling rather than what you should do. What's going on for you?",
+            "That feeling of not knowing what to do can be so frustrating. Instead of focusing on actions right now, what would feel most supportive for you in this moment?"
+        ]
+        return random.choice(responses)
+    elif "tough day" in user_lower or "bad day" in user_lower:
+        responses = [
+            "Some days are just harder than others, aren't they? I'm here with you through this difficult time.",
+            "I'm sorry today has been so rough for you. Would it help to talk about what's been making it especially difficult?",
+            "Days like this can feel so heavy. Thank you for reaching out - you don't have to go through this alone."
+        ]
+        return random.choice(responses)
     elif len(user_message.strip()) <= 2:  # Short messages like "??"
-        return "I'm here with you. Sometimes it's hard to find words, and that's okay. Take your time."
+        responses = [
+            "I'm here with you. Sometimes it's hard to find words, and that's completely okay. Take your time.",
+            "You don't need to have the perfect words. I'm here to listen to whatever you're feeling, however you want to express it.",
+            "Sometimes feelings are too big for words, and that's alright. I'm here with you in this moment."
+        ]
+        return random.choice(responses)
     else:
-        return "I hear you, and I want you to know that I'm here to listen. Your feelings matter to me."
+        # General varied empathetic fallback responses
+        responses = [
+            "Thank you for sharing that with me. I can hear that something important is going on for you, and I want you to know I'm here to listen.",
+            "I'm grateful you felt safe reaching out. Whatever you're going through, you don't have to face it alone.",
+            "It sounds like you're going through something difficult right now. I'm here with you, and your feelings matter to me.",
+            "I can sense that this is weighing on you. Thank you for trusting me with what you're experiencing.",
+            "Whatever brought you here today, I want you to know that I'm here to listen without judgment. You're not alone."
+        ]
+        return random.choice(responses)
 
 # Create Gradio interface
 with gr.Blocks(title="Aura - Your Supportive Friend") as demo: