Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from transformers import pipeline | |
| from gtts import gTTS | |
| from youtubesearchpython import VideosSearch | |
| import os | |
| # 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") | |
| # 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) | |
| # Title | |
| st.title("Grief and Loss Support Bot 🌿") | |
| st.subheader("Your compassionate companion in tough times 💚") | |
| # 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: | |
| # Run sentiment analysis for crisis detection | |
| sentiment = sentiment_analysis(user_input)[0] | |
| # 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) | |
| # 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) | |
| # 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) | |
| # 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)") | |