Update app.py

app.py

@@ -1,13 +1,16 @@
 import streamlit as st
-from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import pipeline
 from gtts import gTTS
-import random
+from pytube import Search
+import os
 
-# Load GPT-2 model and tokenizer from Hugging Face
-tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
-model = GPT2LMHeadModel.from_pretrained("gpt2")
+# Initialize conversational model for empathetic dialogue
+conversational_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 
-# Set up Streamlit page configuration
+# Initialize sentiment analysis
+sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
+
+# Set up Streamlit page
 st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
 st.markdown("""
     <style>
@@ -20,70 +23,43 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Title and introduction to the bot
+# Title
 st.title("Grief and Loss Support Bot 🌿")
 st.subheader("Your compassionate companion in tough times 💚")
 
-# User input
+# Get user input
 user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
 
-# Store previous responses to check for repetition
-if 'previous_responses' not in st.session_state:
-    st.session_state.previous_responses = []
-
-# Function to generate a more empathetic and focused response
-def generate_response(user_input):
-    # Empathy responses categorized by emotion
-    emotion_responses = {
-        "sadness": [
-            "I'm truly sorry you're going through this difficult time. Grief and sadness can weigh heavy, but you're not alone in your feelings. It's okay to take things one step at a time, and together we can explore ways to help you cope and find peace.",
-            "I can only imagine how overwhelming things must feel right now. When we’re grieving or under stress, it can be hard to see the light at the end of the tunnel, but you don’t have to carry the burden alone. I’m here for you.",
-            "I'm really sorry you're feeling this way. It’s completely understandable to feel sad, and it's okay to let those emotions flow. Grief can be exhausting, and it’s normal to need time to process everything."
-        ],
-        "anger": [
-            "It sounds like you're feeling really frustrated, and that’s totally valid. Sometimes, our anger is a way of protecting us from pain or stress. If you’d like, I’m here to help you work through these feelings and find ways to release some of that tension.",
-            "I understand that things may feel frustrating and intense right now. Anger can be a difficult emotion to manage, but it’s completely natural. Sometimes, taking a step back and focusing on deep breathing or a calming activity can help.",
-            "It’s okay to feel angry – sometimes it’s our body’s way of telling us that something needs to change or that we need to release pent-up stress. You’re not alone in this, and there are healthy ways we can work through it together."
-        ]
-    }
-
-    # Tailored coping suggestions based on the user's input
-    activity_suggestions = {
-        "journaling": "Journaling can be a helpful way to process your thoughts and emotions. Writing down everything you’re feeling can allow you to see things more clearly and release some of the weight you’re carrying.",
-        "deep breathing": "Deep breathing can help calm your mind and ease tension. Try taking a few slow, deep breaths, focusing on each inhale and exhale. It can help you feel grounded and in control of your emotions.",
-        "exercise": "Physical activity is a wonderful way to release built-up tension and boost your mood. Taking a walk in nature or doing some light stretching can help ground you in the present moment and relieve stress.",
-        "talking to someone": "Sometimes, sharing how you’re feeling with someone you trust can relieve some of the burden. Talking through your emotions with a friend or loved one can provide comfort and help you feel supported.",
-        "art": "Art can be a therapeutic outlet for your emotions. Creating something with your hands, like drawing or painting, allows you to express what might be too difficult to put into words."
-    }
-
-    # Determine emotion based on keywords in user input
-    if "angry" in user_input.lower() or "frustrated" in user_input.lower():
-        emotion = "anger"
-        activity = "deep breathing"
-        response = random.choice(emotion_responses[emotion])
-        activity_suggestion = activity_suggestions[activity]
-    else:
-        emotion = "sadness"
-        activity = random.choice(list(activity_suggestions.keys()))
-        response = random.choice(emotion_responses[emotion])
-        activity_suggestion = activity_suggestions[activity]
-
-    # Add the tailored coping activity suggestion to the response
-    response += f"\n\nHere's something you could try to help cope with how you're feeling:\n{activity_suggestion}"
-    
-    return response
-
-# Check if the user has typed something
+# Check if user has entered text
 if user_input:
-    # Generate the empathetic response
-    response = generate_response(user_input)
-    
-    # Store and show the new response
-    st.session_state.previous_responses.append(response)
-    st.text_area("Bot's Response:", response, height=250)
+    # Run sentiment analysis to check for distress
+    sentiment = sentiment_analysis(user_input)[0]
+    # Generate empathetic response
+    response = conversational_bot(user_input, max_length=250, temperature=0.95, top_k=60)[0]['generated_text']
+
+    # Display response
+    st.text_area("Bot's Response:", response, height=200)
 
-    # Text-to-speech output (optional)
+    # Text-to-speech output
     tts = gTTS(response, lang='en')
     audio_file = "response.mp3"
     tts.save(audio_file)
     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", "overwhelmed"]):
+        st.info("Here's a suggestion to help you cope:")
+        activities = ["journaling", "yoga", "painting"]
+        selected_activity = st.selectbox("Choose an activity you'd like to try:", activities)
+        
+        # Search YouTube for videos related to the selected activity
+        search = Search(selected_activity)
+        search_results = search.results[:3]  # limit results to 3 videos 
+        for video in search_results:
+            st.write(f"[{video.title}]({video.watch_url})")
+
+    # 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.")
+        st.write("[Find emergency resources here](https://www.helpguide.org/find-help.htm)")