Update app.py

app.py

@@ -1,64 +1,68 @@
 import streamlit as st
 from transformers import pipeline
-import pyttsx3
+from gtts import gTTS
 from youtubesearchpython import VideosSearch
+import os
 
-# Set up the Streamlit app
-st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🕊️", layout="centered")
+# Initialize conversational models
+empathy_bot = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-empathy")
+general_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
 
-# Customizing the app style for a soothing look
+# Set up Streamlit page
+st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
 st.markdown("""
     <style>
-    .main {
-        background-color: #F3F7F6;
-    }
-    .css-1d391kg {
-        font-family: 'Arial', sans-serif;
-    }
+    .css-1d391kg { background-color: #F3F7F6; }
+    .css-ffhzg2 { font-size: 1.5em; font-weight: 500; color: #4C6D7D; }
+    .stTextInput>div>div>input { background-color: #D8E3E2; }
+    .stButton>button { background-color: #A9D0B6; color: white; border-radius: 5px; }
+    .stButton>button:hover { background-color: #8FB79A; }
+    .stTextInput>div>label { color: #4C6D7D; }
     </style>
-    """, unsafe_allow_html=True)
+""", unsafe_allow_html=True)
 
-st.title("Grief and Loss Support Bot 🕊️")
-st.subheader("We are here for you. 💚 Your companion in tough times")
+# Title
+st.title("Grief and Loss Support Bot 🌿")
+st.subheader("Your compassionate companion in tough times 💚")
 
-# Set up the conversational model
-try:
-    chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
-except Exception as e:
-    st.error(f"Error loading the model: {e}")
-
-# Set up text-to-speech engine
-engine = pyttsx3.init()
-engine.setProperty('rate', 150)  # Speed of speech
-engine.setProperty('volume', 0.9)  # Volume level
-voices = engine.getProperty('voices')
-engine.setProperty('voice', voices[0].id)  # Choose the desired voice
-
-# User input for conversation
+# Get user input
 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:
-    # Generate a response using the model
-    response = chatbot(user_input, max_length=100)[0]['generated_text']
+    # Run sentiment analysis for crisis detection
+    sentiment = sentiment_analysis(user_input)[0]
     
-    # Display response and read it aloud
-    st.write(response)
-    engine.say(response)
-    engine.runAndWait()
-
-    # Suggest hobbies to uplift the user's mood
-    hobbies = ["Painting", "Reading books", "Gardening", "Learning a new language", "Playing an instrument"]
-    st.subheader("Suggested Hobbies")
-    st.write("Here are some hobbies that might help you feel better:")
-    for hobby in hobbies:
-        st.write(f"- {hobby}")
+    # Use empathy model for more specific support if the user is distressed
+    if sentiment['label'] == 'NEGATIVE' and sentiment['score'] > 0.8:
+        response = empathy_bot(user_input, max_length=150)[0]['generated_text']
+    else:
+        response = general_bot(user_input, max_length=150)[0]['generated_text']
+    
+    # Display response
+    st.text_area("Bot's Response:", response, height=150)
 
-    # Search for helpful YouTube videos
-    st.subheader("Recommended Videos")
-    video_search = VideosSearch('motivational activities for coping with grief', limit=3)
-    results = video_search.result()['result']
+    # Text-to-speech output
+    tts = gTTS(response, lang='en')
+    audio_file = "response.mp3"
+    tts.save(audio_file)
+    st.audio(audio_file, format="audio/mp3", use_container_width=True)
 
-    for video in results:
-        st.write(f"[{video['title']}]({video['link']})")
+    # 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)
+        
+        # Search YouTube for videos related to the selected activity
+        videosSearch = VideosSearch(activity, limit=2)
+        video_links = [result['link'] for result in videosSearch.result()['result']]
+        for link in video_links:
+            st.write(link)
 
-st.write("Remember, it's okay to seek support and take small steps towards healing. 💚")
+    # 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)")