Update app.py

app.py

@@ -61,58 +61,58 @@ 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 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
-        )
-        
-        response_text = outputs[0]["generated_text"]
-        
-        # More robust prompt stripping
-        if response_text.startswith(prompt):
-            response_text = response_text[len(prompt):].strip()
+        """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?"
+            )
         
-        # Additional processing to ensure response quality
-        response_text = response_text.split('\n')[0].strip()  # Take first coherent line
+        # 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)
+        )
         
-        # 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?"
+        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
+            )
+            
+            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?"
+                )
+            
+            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?"
             )
         
-        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."""
         if not self.summary_model: