Update app.py

app.py

@@ -1,64 +1,99 @@
 import streamlit as st
 from transformers import pipeline
 from gtts import gTTS
-from pytube import Search
-import os
+from io import BytesIO
+import random
+import requests
+from youtubesearchpython import VideosSearch
 
-# Initialize conversational models
-conversational_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
-sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
+# Initialize the empathy bot pipeline
+empathy_bot = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-empathy")
 
-# Set up Streamlit page
-st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
-st.markdown("""
-    <style>
-    .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)
+# Function to process the user's input and generate a dynamic response
+def generate_dynamic_response(user_input):
+    # Generate an empathetic response using the model
+    response = empathy_bot(user_input)
+    empathetic_response = response[0]['generated_text']
 
-# Title
-st.title("Grief and Loss Support Bot 🌿")
-st.subheader("Your compassionate companion in tough times 💚")
+    # Possible activities to suggest
+    activities = {
+        "relaxation": [
+            "Deep Breathing Exercises",
+            "Guided Meditation",
+            "Mindfulness Breathing"
+        ],
+        "creative": [
+            "Journaling your thoughts",
+            "Drawing or doodling",
+            "Writing a poem"
+        ],
+        "exercise": [
+            "Try some yoga",
+            "Go for a walk outside",
+            "Stretching exercises to release tension"
+        ],
+        "self-care": [
+            "Take a warm bath",
+            "Listen to calming music",
+            "Treat yourself with some relaxation time"
+        ]
+    }
+
+    # Logic to determine context of user input (e.g., stress, sadness, etc.)
+    if "sad" in user_input or "overwhelmed" in user_input:
+        suggested_activity = random.choice(activities["relaxation"])
+    elif "bored" in user_input or "unmotivated" in user_input:
+        suggested_activity = random.choice(activities["creative"])
+    elif "stressed" in user_input or "tired" in user_input:
+        suggested_activity = random.choice(activities["exercise"])
+    else:
+        suggested_activity = random.choice(activities["self-care"])
+
+    # Search YouTube for relevant videos based on the activity suggestion
+    video_keywords = suggested_activity
+    video_results = search_youtube_videos(video_keywords)
+
+    return empathetic_response, suggested_activity, video_results
+
+# Function to search for YouTube videos
+def search_youtube_videos(query):
+    videos_search = VideosSearch(query, limit=3)
+    results = videos_search.result()
+
+    # Collect video details
+    video_links = []
+    for video in results['videos']:
+        title = video['title']
+        link = video['link']
+        video_links.append((title, link))
 
-# Get user input
-user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
+    return video_links
+
+# Streamlit UI
+st.title("Grief and Loss Support Bot 🌿")
+st.markdown("Your compassionate companion in tough times 💚")
+user_input = st.text_input("Share what's on your mind...")
 
-# Check if user has entered text
 if user_input:
-    # Run sentiment analysis for crisis detection
-    sentiment = sentiment_analysis(user_input)[0]
-    
-    # Use conversational bot for responses
-    response = conversational_bot(user_input, max_length=150)[0]['generated_text']
-    
-    # Display response
-    st.text_area("Bot's Response:", response, height=150)
+    empathetic_response, suggested_activity, video_results = generate_dynamic_response(user_input)
 
-    # 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
+    # Display the empathetic response
+    st.write(f"**Bot's Response:**")
+    st.write(empathetic_response)
 
-    # 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
-        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})")
+    # Suggest an activity
+    st.write(f"Here's a suggestion to help you cope:")
+    st.write(f"**Activity to try:** {suggested_activity}")
 
-    # 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)")
+    # Display YouTube video suggestions
+    if video_results:
+        st.write("Here are some YouTube videos that may help:")
+        for title, link in video_results:
+            st.write(f"[{title}]({link})")
+
+    # Generate and play an audio response using gTTS
+    tts = gTTS(empathetic_response, lang='en')
+    audio_file = BytesIO()
+    tts.save(audio_file)
+    audio_file.seek(0)
+    st.audio(audio_file, format="audio/mp3", use_container_width=True)