Update app.py

app.py

@@ -4,8 +4,10 @@ from gtts import gTTS
 from pytube import Search
 import os
 
-# Initialize 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")
 
 # Set up Streamlit page
@@ -26,64 +28,42 @@ st.title("Grief and Loss Support Bot 🌿")
 st.subheader("Your compassionate companion in tough times 💚")
 
 # Get user input
-user_input = st.text_input("How are you feeling today?", placeholder="Share what's on your mind...", max_chars=500)
+user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
 
 # Check if user has entered text
 if user_input:
-    # Run sentiment analysis to assess mood
-    try:
-        sentiment = sentiment_analysis(user_input)[0]
-        sentiment_label = sentiment['label'].lower()
-        st.write(f"Sentiment detected: {sentiment_label}")  # Debugging line
-    except Exception as e:
-        st.write(f"Error in sentiment analysis: {e}")  # Error handling
+    # Run sentiment analysis to check for distress
+    sentiment = sentiment_analysis(user_input)[0]
+
+    # Generate empathetic response (model generates responses with empathy)
+    response = conversational_bot(user_input, max_length=150)[0]['generated_text']
 
-    # Generate empathetic response from the conversational bot
-    try:
-        bot_response = conversational_bot(user_input, max_length=150)[0]['generated_text']
-        st.write(f"Bot Response: {bot_response}")  # Debugging line
-    except Exception as e:
-        bot_response = "I'm really sorry you're feeling this way. It must be difficult. Let's talk about it."
-        st.write(f"Error generating bot response: {e}")  # Error handling
-        st.write("Using fallback response.")
+    # 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:", bot_response, height=150)
+    st.text_area("Bot's Response:", response, height=150)
 
     # Text-to-speech output
-    tts = gTTS(bot_response, lang='en')
+    tts = gTTS(response, lang='en')
     audio_file = "response.mp3"
     tts.save(audio_file)
     st.audio(audio_file, format="audio/mp3")
 
-    # Suggest an activity
-    if sentiment_label == "negative":
-        st.info("Here's a suggestion to help lift your spirits:")
-        activities = {
-            "relaxation": ["mindful breathing", "yoga"],
-            "creative": ["painting", "writing in a journal"],
-            "exercise": ["a short walk", "stretching"],
-            "self-care": ["listening to calming music", "meditation"]
-        }
+    # Suggest a productive activity based on detected keywords
+    if any(keyword in user_input.lower() for keyword in ["lonely", "lost", "sad"]):
+        st.info("Here's a suggestion to help you cope:")
+        hobbies = ["journaling", "yoga", "painting"]
+        activity = st.selectbox("Choose an activity you'd like to try:", hobbies)
         
-        # Choose activity based on keyword in input
-        if any(keyword in user_input.lower() for keyword in ["lonely", "lost", "sad", "down"]):
-            activity_category = "self-care"
-        elif any(keyword in user_input.lower() for keyword in ["stress", "pressure", "overwhelmed"]):
-            activity_category = "relaxation"
-        else:
-            activity_category = "creative"
-        
-        suggested_activity = activities[activity_category]
-        st.write(f"Activity Suggestion: {', '.join(suggested_activity)}")
-
         # Search YouTube for videos related to the selected activity
-        search = Search(suggested_activity[0])
+        search = Search(activity)
         search_results = search.results[:2]  # limit results to 2 videos
         for video in search_results:
             st.write(f"[{video.title}]({video.watch_url})")
 
-    # Crisis resources for critical keywords
+    # Crisis resources
     crisis_keywords = ["help", "suicide", "depressed", "emergency", "hurt", "lost"]
     if any(keyword in user_input.lower() for keyword in crisis_keywords):
         st.warning("It seems like you might be in distress. Please reach out to a crisis hotline or a trusted individual.")