Update app.py

app.py

@@ -1,77 +1,64 @@
 import streamlit as st
-import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers import pipeline
 import pyttsx3
+from youtubesearchpython import VideosSearch
 
-# Set up the page configuration for a welcoming appearance
+# Set up the Streamlit app
 st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🕊️", layout="centered")
 
-# Customizing the app style for a soothing and modern look
+# Customizing the app style for a soothing look
 st.markdown("""
     <style>
-    .css-1d391kg {
+    .main {
         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;
-        border: none;
-    }
-    .stButton>button:hover {
-        background-color: #8FB79A;
-    }
-    .stTextInput>div>label {
-        color: #4C6D7D;
+    .css-1d391kg {
+        font-family: 'Arial', sans-serif;
     }
     </style>
     """, unsafe_allow_html=True)
 
-# Title and introduction
 st.title("Grief and Loss Support Bot 🕊️")
 st.subheader("We are here for you. 💚 Your companion in tough times")
 
-# Load the model and tokenizer for text generation
-model_name = "microsoft/DialoGPT-medium"
+# Set up the conversational model
 try:
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    text_gen_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0 if torch.cuda.is_available() else -1)
+    chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
 except Exception as e:
-    st.error(f"Error loading the conversational model: {e}")
+    st.error(f"Error loading the model: {e}")
 
-# Initialize the TTS engine
-try:
-    tts_engine = pyttsx3.init()
-    tts_engine.setProperty('rate', 150)  # Adjust the speed of speech
-    tts_engine.setProperty('voice', tts_engine.getProperty('voices')[0].id)  # Choose the first voice option
-except Exception as e:
-    st.error(f"Error initializing the TTS engine: {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
 user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
 
 if user_input:
-    # Generate a conversational response
-    try:
-        response = text_gen_pipeline(user_input, max_length=100, num_return_sequences=1)
-        response_text = response[0]['generated_text']
-        
-        st.write("Bot's Response:")
-        st.write(response_text)
+    # Generate a response using the model
+    response = chatbot(user_input, max_length=100)[0]['generated_text']
+    
+    # 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}")
+
+    # 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']
 
-        # Convert the response text to speech
-        if st.button("Play Response Audio"):
-            tts_engine.say(response_text)
-            tts_engine.runAndWait()
+    for video in results:
+        st.write(f"[{video['title']}]({video['link']})")
 
-    except Exception as e:
-        st.error(f"Error generating response: {e}")
+st.write("Remember, it's okay to seek support and take small steps towards healing. 💚")