import streamlit as st
from transformers import pipeline
from gtts import gTTS
from pytube import Search
import random
import os

# Initialize conversational and sentiment analysis models
conversational_bot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")

# Streamlit page configuration
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: #E8F1F2; border-radius: 5px; }
    .stButton>button { background-color: #A9D0B6; color: white; }
    .stButton>button:hover { background-color: #8FB79A; }
    .stTextInput>div>label { color: #4C6D7D; }
    </style>
""", unsafe_allow_html=True)

# Title and description
st.title("Grief and Loss Support Bot 🌿")
st.subheader("Your compassionate companion in tough times 💚")

# User input
user_input = st.text_input("How are you feeling today?", placeholder="Share your thoughts here...", max_chars=500)

# Generate response
if user_input:
    # Analyze sentiment
    sentiment = sentiment_analysis(user_input)[0]

    # Use conversational bot for response
    response = conversational_bot(user_input, max_length=150)[0]['generated_text']
    st.text_area("Bot's Response:", response, height=150)

    # Text-to-speech conversion
    tts = gTTS(response, lang='en')
    audio_file = "response.mp3"
    tts.save(audio_file)
    st.audio(audio_file, format="audio/mp3")

    # Suggest activity based on mood keywords
    mood_keywords = {
        "positive": ["try journaling your feelings", "practice yoga", "learn a new recipe"],
        "neutral": ["take a short walk", "listen to calming music", "try some mindful breathing"],
        "negative": ["explore creative activities like painting", "watch a motivational video", "write down small goals to feel organized"]
    }

    mood_type = "neutral"  # Default category
    if any(word in user_input.lower() for word in ["lonely", "lost", "sad", "stressed"]):
        mood_type = "negative"
    elif sentiment["label"] == "POSITIVE":
        mood_type = "positive"

    suggested_activity = random.choice(mood_keywords[mood_type])
    st.info("Here's a suggestion to help lift your spirits:")
    st.write(f"**Activity Suggestion:** {suggested_activity}")

    # YouTube search for video resources
    search = Search(suggested_activity)
    st.write("Recommended Videos:")
    for video in search.results[:2]:  # Display top 2 videos
        st.write(f"[{video.title}]({video.watch_url})")

    # Crisis intervention resources
    crisis_keywords = ["help", "suicide", "depressed", "emergency", "hurt", "lost"]
    if any(keyword in user_input.lower() for keyword in crisis_keywords):
        st.warning("It sounds like you're going through a very tough time. Please reach out to someone you trust or a professional for support.")
        st.write("[Find emergency resources here](https://www.helpguide.org/find-help.htm)")