Update app.py

app.py

@@ -1,70 +1,74 @@
 import streamlit as st
-from llama_cpp import Llama
+from transformers import pipeline
 from gtts import gTTS
 from pytube import Search
 import random
 import os
 
-# Initialize the Llama model by specifying the model path
-llm = Llama(model_path="HyunCello-KULLM3-empathy-v1.0-IQ4_XS.gguf")
+# 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")
 
-# Configure Streamlit page
-st.set_page_config(page_title="Grief & Loss Support Bot", page_icon="🌱", layout="centered")
+# Streamlit page configuration
+st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
 st.markdown("""
     <style>
-    .main { background-color: #F4F9F9; }
-    h1, h2, h3, h4 { color: #4C6D7D; }
+    .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: #A4D4AE; color: white; }
-    .stButton>button:hover { background-color: #92C8A6; }
+    .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 & Loss Support Bot 🌱")
+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 empathetic response using Llama model
+# Generate response
 if user_input:
-    response = llm(
-        messages=[
-            {"role": "user", "content": user_input}
-        ]
-    )['choices'][0]['message']['content']
-    
-    # Display empathetic response
+    # 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)
 
-    # Convert text to speech for added comfort
+    # Text-to-speech conversion
     tts = gTTS(response, lang='en')
     audio_file = "response.mp3"
     tts.save(audio_file)
     st.audio(audio_file, format="audio/mp3")
 
-    # Suggested activities based on user input
-    activity_suggestions = {
+    # 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"]
     }
-    activities = activity_suggestions["neutral"]  # Default to "neutral" for simplicity
 
-    # 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}")
+    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(selected_activity)
+    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 response if critical keywords are detected
+    # Crisis intervention resources
     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)")
+        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)")