Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from transformers import pipeline | |
| from gtts import gTTS | |
| import os | |
| from youtubesearchpython import VideosSearch | |
| # Page configuration | |
| st.set_page_config(page_title="Grief Support Bot", layout="wide") | |
| st.markdown("<style>.stApp {background-color: #f0f5f9;}</style>", unsafe_allow_html=True) | |
| # Load the pre-trained chatbot model | |
| chatbot = pipeline("conversational", model="microsoft/DialoGPT-medium") | |
| # Function for voice output | |
| def speak_text(text): | |
| tts = gTTS(text, lang='en') | |
| tts.save("response.mp3") | |
| os.system("mpg321 response.mp3") # This plays the audio | |
| st.audio("response.mp3") | |
| # Function to search for YouTube videos | |
| def get_video_links(query, max_results=3): | |
| search = VideosSearch(query, max_results=max_results) | |
| results = search.result()['result'] | |
| return [(video['title'], video['link']) for video in results] | |
| # Function to detect crisis keywords | |
| crisis_keywords = ["suicide", "hopeless", "alone", "depressed"] | |
| def detect_crisis(input_text): | |
| return any(word in input_text.lower() for word in crisis_keywords) | |
| # Function to suggest hobbies | |
| def suggest_hobbies(): | |
| return ["Painting", "Gardening", "Writing", "Yoga", "Learning an instrument"] | |
| # App header | |
| st.title("Grief and Loss Support Bot") | |
| # Maintain conversation history using session state | |
| if 'history' not in st.session_state: | |
| st.session_state['history'] = [] | |
| # User input | |
| user_input = st.text_input("You:") | |
| # If there is input, process the conversation | |
| if user_input: | |
| # Crisis detection | |
| if detect_crisis(user_input): | |
| st.warning("We care about you. If you're in crisis, please reach out to a trusted person or call emergency services.") | |
| st.markdown("Hotline resources: [Link to emergency contacts]") | |
| else: | |
| # Generate chatbot response | |
| response = chatbot(user_input) | |
| generated_text = response[0]['generated_text'] | |
| # Append to history and display response | |
| st.session_state['history'].append(("You", user_input)) | |
| st.session_state['history'].append(("Bot", generated_text)) | |
| # Display conversation history | |
| for speaker, text in st.session_state['history']: | |
| st.write(f"**{speaker}:** {text}") | |
| # Voice output | |
| speak_text(generated_text) | |
| # Display hobby suggestions | |
| st.markdown("### Hobby Suggestions") | |
| st.markdown(", ".join(suggest_hobbies())) | |
| # Provide interactive YouTube video links | |
| st.markdown("### Videos to Help You Cope with Grief") | |
| video_links = get_video_links("coping with grief and loss", max_results=3) | |
| for title, link in video_links: | |
| st.markdown(f"- [{title}]({link})") | |