Update app.py

app.py

@@ -61,39 +61,57 @@ class AITherapistAssistant:
         return any(keyword in message_lower for keyword in SUICIDE_KEYWORDS)
     
     def generate_response(self, message: str) -> str:
-        """Generate a supportive AI response from the conversation model."""
-        if not self.conversation_model:
-            return (
-                "I'm here to listen, but I'm currently having trouble loading my AI model. "
-                "Please try again later."
-            )
-        
-        # Prompt to steer the model toward empathy and support
-        prompt = (
-            "You are a compassionate AI therapist. Respond supportively to this message:\n"
-            f"{message}\n\n"
-            "Be empathetic, validate feelings, and avoid giving direct medical advice."
+    """Generate a supportive AI response from the conversation model."""
+    if not self.conversation_model:
+        return (
+            "I'm having some technical difficulties right now. "
+            "I'm truly sorry I can't provide support at the moment. "
+            "Would you consider reaching out to a human counselor or support helpline?"
+        )
+    
+    # Create a more structured and empathetic prompt
+    prompt = (
+        "You are a compassionate, empathetic AI therapist trained to provide supportive, "
+        "non-judgmental responses. Your goal is to validate feelings, offer gentle support, "
+        "and help the user feel heard and understood.\n\n"
+        "User's message: {}\n\n"
+        "Your compassionate response:".format(message)
+    )
+    
+    try:
+        outputs = self.conversation_model(
+            prompt, 
+            max_length=300,  # Increased length for more nuanced responses
+            num_return_sequences=1,
+            do_sample=True,
+            top_p=0.9,
+            temperature=0.7
         )
         
-        try:
-            outputs = self.conversation_model(
-                prompt, 
-                max_length=250,
-                num_return_sequences=1,
-                do_sample=True,
-                top_p=0.9,
-                temperature=0.7
+        response_text = outputs[0]["generated_text"]
+        
+        # More robust prompt stripping
+        if response_text.startswith(prompt):
+            response_text = response_text[len(prompt):].strip()
+        
+        # Additional processing to ensure response quality
+        response_text = response_text.split('\n')[0].strip()  # Take first coherent line
+        
+        # Fallback if response is too short or nonsensical
+        if len(response_text) < 20:
+            response_text = (
+                "I hear you. Your feelings are valid, and it takes courage to share what you're experiencing. "
+                "Would you like to tell me a bit more about what's on your mind?"
             )
-            response_text = outputs[0]["generated_text"]
-            
-            # If the model echoes the prompt, strip it out:
-            if response_text.startswith(prompt):
-                response_text = response_text[len(prompt):].strip()
-            
-            return response_text.strip()
-        except Exception as e:
-            st.error(f"Error generating response: {e}")
-            return "I'm sorry, but I'm having trouble responding right now."
+        
+        return response_text
+    except Exception as e:
+        st.error(f"Error generating response: {e}")
+        return (
+            "I'm experiencing some difficulties right now. "
+            "Your feelings are important, and I want to be fully present for you. "
+            "Would you be willing to try sharing again?"
+        )
     
     def generate_summary(self, conversation_text: str) -> str:
         """Generate a short summary of the entire conversation."""