Update app.py

app.py

@@ -4,10 +4,11 @@ from gtts import gTTS
 from pytube import Search
 import os
 
-# Initialize conversational models
+# Initialize conversational model for empathetic dialogue
 conversational_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+
+# Initialize sentiment analysis
 sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
-motivational_response = pipeline("text-generation", model="huggingtweets/motivational")
 
 # Set up Streamlit page
 st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
@@ -31,24 +32,24 @@ user_input = st.text_input("Share what's on your mind...", placeholder="Type her
 
 # Check if user has entered text
 if user_input:
-    # Run sentiment analysis for crisis detection
+    # Run sentiment analysis to check for distress
     sentiment = sentiment_analysis(user_input)[0]
-    
-    # Use conversational bot for responses
+
+    # Generate empathetic response (model generates responses with empathy)
     response = conversational_bot(user_input, max_length=150)[0]['generated_text']
-    
+
+    # Ensure response does not repeat what the user said, and is supportive
+    if user_input.lower() in response.lower():
+        response = "I understand how you're feeling. You're not alone in this. I'm here to listen and help."
+
     # Display response
     st.text_area("Bot's Response:", response, height=150)
 
-    # Generate motivational message
-    motivation = motivational_response(user_input, max_length=150)[0]['generated_text']
-    st.text_area("Motivational Response:", motivation, height=150)
-
-    # Text-to-speech output for both responses
-    tts = gTTS(response + " " + motivation, lang='en')
+    # Text-to-speech output
+    tts = gTTS(response, lang='en')
     audio_file = "response.mp3"
     tts.save(audio_file)
-    st.audio(audio_file, format="audio/mp3")  # Removed use_container_width=True
+    st.audio(audio_file, format="audio/mp3")
 
     # Suggest a productive activity based on detected keywords
     if any(keyword in user_input.lower() for keyword in ["lonely", "lost", "sad"]):
@@ -58,7 +59,7 @@ if user_input:
         
         # Search YouTube for videos related to the selected activity
         search = Search(activity)
-        search_results = search.results[:3]  # limit results to 2 videos
+        search_results = search.results[:2]  # limit results to 2 videos
         for video in search_results:
             st.write(f"[{video.title}]({video.watch_url})")