Update app.py

app.py

@@ -1,108 +1,69 @@
 import streamlit as st
-from llama_cpp import Llama
+from transformers import pipeline
 from gtts import gTTS
-from io import BytesIO
+from pytube import Search
 import random
-import requests
-import yt_dlp
-
-# Initialize the empathy bot using llama_cpp
-llm = Llama.from_pretrained(
-    repo_id="featherless-ai-quants/HyunCello-KULLM3-empathy-v1.0-GGUF",
-    filename="HyunCello-KULLM3-empathy-v1.0-IQ4_XS.gguf",
-)
-
-# Function to process the user's input and generate a dynamic response
-def generate_dynamic_response(user_input):
-    # Generate an empathetic response using the Llama model
-    response = llm.create_chat_completion(
-        messages=[{"role": "user", "content": user_input}]
-    )
-    empathetic_response = response['choices'][0]['message']['content']
-
-    # 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
-
-
-def search_youtube_videos(query):
-    ydl_opts = {
-        'quiet': True,
-        'extract_flat': True,
-        'noplaylist': True,
-        'max_results': 3,
-    }
-
-    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-        result = ydl.extract_info(f"ytsearch:{query}", download=False)
-        videos = result.get("entries", [])
-        video_links = [(video['title'], video['url']) for video in videos]
-    
-    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...")
-
+import os
+
+# Initialize pretrained Hugging Face models
+empathetic_response = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-empathy")
+sentiment_analysis = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
+
+# Configure Streamlit page
+st.set_page_config(page_title="Grief & Loss Support Bot", page_icon="🌱", layout="centered")
+st.markdown("""
+    <style>
+    .main { background-color: #F4F9F9; }
+    h1, h2, h3, h4 { color: #4C6D7D; }
+    .stTextInput>div>div>input { background-color: #E8F1F2; border-radius: 5px; }
+    .stButton>button { background-color: #A4D4AE; color: white; }
+    .stButton>button:hover { background-color: #92C8A6; }
+    </style>
+""", unsafe_allow_html=True)
+
+# Title and description
+st.title("Grief & 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)
+
+# Detect user sentiment and respond
 if user_input:
-    empathetic_response, suggested_activity, video_results = generate_dynamic_response(user_input)
-
-    # Display the empathetic response
-    st.write(f"**Bot's Response:**")
-    st.write(empathetic_response)
-
-    # Suggest an activity
-    st.write(f"Here's a suggestion to help you cope:")
-    st.write(f"**Activity to try:** {suggested_activity}")
-
-    # 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()
+    sentiment = sentiment_analysis(user_input)[0]
+    empathy_response = empathetic_response(user_input, max_length=100)[0]['generated_text']
+    
+    # Display empathetic response
+    st.text_area("Bot's Response:", empathy_response, height=150)
+    
+    # Convert text to speech for added comfort
+    tts = gTTS(empathy_response, lang='en')
+    audio_file = "response.mp3"
     tts.save(audio_file)
-    audio_file.seek(0)
-    st.audio(audio_file, format="audio/mp3", use_container_width=True)
+    st.audio(audio_file, format="audio/mp3")
+
+    # Activity Suggestion based on emotions detected
+    mood = sentiment['label'].lower()
+    activity_suggestions = {
+        "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"]
+    }
+    activities = activity_suggestions.get(mood, ["try relaxation exercises", "reach out to someone you trust"])
+    
+    # Show dynamic activity suggestion
+    st.info("Here’s something you could try to lift your spirits:")
+    selected_activity = random.choice(activities)
+    st.write(f"**Activity Suggestion:** {selected_activity}")
+
+    # YouTube search for video resources
+    search = Search(selected_activity)
+    st.write("Recommended Videos:")
+    for video in search.results[:2]:  # Display top 2 videos
+        st.write(f"[{video.title}]({video.watch_url})")
+    
+    # Crisis response if critical keywords are detected
+    crisis_keywords = ["help", "suicide", "depressed", "emergency", "hurt", "lost"]
+    if any(keyword in user_input.lower() for keyword in crisis_keywords):
+        st.error("It sounds like you're going through a very tough time. Please don't hesitate to reach out for help.")
+        st.write("[Click here for emergency resources](https://www.helpguide.org/find-help.htm)")