Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import pipeline
 from gtts import gTTS
 from pytube import Search
 import os
+import random
 
 # Initialize conversational model for empathetic dialogue
 conversational_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
@@ -30,33 +31,42 @@ st.subheader("Your compassionate companion in tough times 💚")
 # Get user input
 user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
 
-# Store previous response to check for repetition
-if 'previous_response' not in st.session_state:
-    st.session_state.previous_response = ""
+# Store previous responses to check for repetition
+if 'previous_responses' not in st.session_state:
+    st.session_state.previous_responses = []
 
 # Check if user has entered text
 if user_input:
     # Run sentiment analysis to check for distress
     sentiment = sentiment_analysis(user_input)[0]
 
-    # Generate empathetic response with increased length and more context
-    response = conversational_bot(user_input, max_length=300, temperature=0.9, top_k=50, num_return_sequences=1)
+    # Generate multiple responses with increased randomness
+    responses = conversational_bot(user_input, max_length=300, temperature=1, top_k=50, num_return_sequences=3)
 
-    # Choose the response that does not repeat what the user said
-    best_response = response[0]['generated_text']
+    # Filter out any responses that are too similar to previous responses
+    new_responses = [response['generated_text'] for response in responses]
+    new_responses = [resp for resp in new_responses if resp.lower() not in [prev.lower() for prev in st.session_state.previous_responses]]
 
-    # Ensure the response is supportive and does not repeat the user's input
-    if user_input.lower() in best_response.lower():
-        best_response = "I understand how you're feeling. You're not alone in this. I'm here to listen and help."
-    
-    # Store the response for future comparison
-    st.session_state.previous_response = best_response
+    # If there are valid new responses, pick one, otherwise fallback
+    if new_responses:
+        selected_response = random.choice(new_responses)
+    else:
+        # If no new response, fallback to a more generic empathetic message
+        fallback_responses = [
+            "I understand how you're feeling. You're not alone in this. I'm here to listen and help.",
+            "I'm really sorry you're going through this. Let's take one step at a time. I'm here for you.",
+            "It sounds really tough right now. It's okay to feel overwhelmed. You're doing your best, and that's enough."
+        ]
+        selected_response = random.choice(fallback_responses)
+
+    # Store the new response for future checks
+    st.session_state.previous_responses.append(selected_response)
 
     # Display response
-    st.text_area("Bot's Response:", best_response, height=250)
+    st.text_area("Bot's Response:", selected_response, height=250)
 
     # Text-to-speech output
-    tts = gTTS(best_response, lang='en')
+    tts = gTTS(selected_response, lang='en')
     audio_file = "response.mp3"
     tts.save(audio_file)
     st.audio(audio_file, format="audio/mp3")