Update app.py

app.py

@@ -1,18 +1,13 @@
 import streamlit as st
 from transformers import pipeline, GPT2LMHeadModel, GPT2Tokenizer
 from gtts import gTTS
-from pytube import Search
-import os
 import random
 
-# Initialize GPT-2 model and tokenizer from Hugging Face
+# Load GPT-2 model and tokenizer from Hugging Face
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 model = GPT2LMHeadModel.from_pretrained("gpt2")
 
-# Create a text generation pipeline using GPT-2
-generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
-
-# Set up Streamlit page
+# Set up Streamlit page configuration
 st.set_page_config(page_title="Grief and Loss Support Bot", page_icon="🌿", layout="centered")
 st.markdown("""
     <style>
@@ -25,83 +20,61 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Title
+# Title and introduction to the bot
 st.title("Grief and Loss Support Bot 🌿")
 st.subheader("Your compassionate companion in tough times 💚")
 
-# Get user input
+# User input
 user_input = st.text_input("Share what's on your mind...", placeholder="Type here...", max_chars=500)
 
 # Store previous responses to check for repetition
 if 'previous_responses' not in st.session_state:
     st.session_state.previous_responses = []
 
-# Check if user has entered text
-if user_input:
-    # Run the text generation model to generate a response based on user input
-    generated_responses = generator(user_input, max_length=250, num_return_sequences=3, temperature=0.7)
-
-    # Filter out any responses that are too similar to previous responses or user input
-    new_responses = [response['generated_text'] for response in generated_responses]
-    new_responses = [resp for resp in new_responses if resp.lower() not in [prev.lower() for prev in st.session_state.previous_responses] and resp.lower() != user_input.lower()]
-
-    # If there are valid new responses, pick one, otherwise fallback
-    if new_responses:
-        selected_response = random.choice(new_responses)
-    else:
-        # If no new response, fallback to a more generic empathetic message
-        fallback_responses = [
-            "I understand how you're feeling. You're not alone in this. I'm here to listen and help.",
-            "I'm really sorry you're going through this. Let's take one step at a time. I'm here for you.",
-            "It sounds really tough right now. It's okay to feel overwhelmed. You're doing your best, and that's enough."
-        ]
-        selected_response = random.choice(fallback_responses)
-
-    # Add extra empathetic phrases to the response
-    extra_empathy = [
-        "It’s completely normal to feel this way when things get tough. You're doing great by reaching out.",
-        "I know it can feel like a lot right now, but one step at a time. You're not alone in this.",
-        "Even in the toughest times, remember that there’s always support around you."
+# Function to generate a more empathetic and focused response
+def generate_response(user_input):
+    # Predefined empathetic responses for cases of sadness and overwhelming stress
+    empathy_responses = [
+        "I'm really sorry you're going through this. It’s okay to feel this way, and I’m here to help you process it.",
+        "I understand how overwhelming things can feel right now. You're not alone. It’s important to take things one step at a time.",
+        "It sounds really tough, but reaching out is a big first step. You’re doing great. Take a deep breath. You're not alone in this."
     ]
-    selected_response += " " + random.choice(extra_empathy)
 
-    # Store the new response for future checks
-    st.session_state.previous_responses.append(selected_response)
+    # Tailored coping suggestions based on the user's input
+    activity_suggestions = {
+        "journaling": "Journaling is a great way to process your emotions. Write down whatever comes to mind to help release the feelings you're carrying.",
+        "yoga": "Yoga can help you relax and find calm. Simple breathing exercises or gentle stretches might ease the tension you're feeling.",
+        "meditation": "Mindful meditation can help you center yourself and reduce stress. Even a few minutes can make a big difference.",
+        "exercise": "Physical activity can lift your mood and clear your mind. A short walk or some light exercise could help you feel better."
+    }
+
+    # Pick a relevant empathetic response
+    response = random.choice(empathy_responses)
+    
+    # Based on keywords in the input, provide a relevant activity suggestion
+    if "exam" in user_input.lower() or "study" in user_input.lower():
+        activity = "journaling"
+    elif "stress" in user_input.lower() or "overwhelmed" in user_input.lower():
+        activity = "yoga"
+    else:
+        activity = random.choice(list(activity_suggestions.keys()))
 
-    # Display response
-    st.text_area("Bot's Response:", selected_response, height=250)
+    # Add a coping activity suggestion to the response
+    response += f"\n\nHere's something you could try to help cope with how you're feeling:\n{activity_suggestions[activity]}"
+    
+    return response
 
-    # Text-to-speech output
-    tts = gTTS(selected_response, lang='en')
+# Check if the user has typed something
+if user_input:
+    # Generate the empathetic response
+    response = generate_response(user_input)
+    
+    # Store and show the new response
+    st.session_state.previous_responses.append(response)
+    st.text_area("Bot's Response:", response, height=250)
+
+    # Text-to-speech output (optional)
+    tts = gTTS(response, lang='en')
     audio_file = "response.mp3"
     tts.save(audio_file)
     st.audio(audio_file, format="audio/mp3")
-
-    # Suggest a productive activity based on detected keywords
-    if any(keyword in user_input.lower() for keyword in ["lonely", "lost", "sad", "overwhelmed", "academic", "exam"]):
-        st.info("Here's a suggestion to help you cope:")
-
-        # Providing a variety of activities based on user mood and needs
-        activities = {
-            "journaling": "Express your feelings in writing. Journaling is a great way to process emotions.",
-            "yoga": "Yoga helps you relax and refocus. Try some deep breathing exercises or light stretching.",
-            "painting": "Creative expression through painting or drawing can be soothing and help you release pent-up emotions.",
-            "meditation": "Take a moment to calm your mind. Guided meditation can help reduce stress and anxiety.",
-            "exercise": "Physical activity can lift your mood. Even a short walk in nature can make a big difference."
-        }
-
-        # Randomly select an activity category to suggest
-        activity = random.choice(list(activities.keys()))
-        st.write(f"How about {activity}? {activities[activity]}")
-
-        # Search YouTube for videos related to the selected activity
-        search = Search(activity)
-        search_results = search.results[:3]  # limit results to 3 videos
-        for video in search_results:
-            st.write(f"[{video.title}]({video.watch_url})")
-
-    # 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)")