Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from gtts import gTTS
-import random
 from pytube import Search
+import random
 
 # Load DialoGPT model and tokenizer from Hugging Face
 tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
@@ -52,30 +52,20 @@ def generate_response(user_input):
 
     # Add general supportive message
     response += "\n\nYou're doing your best, and that’s all anyone can ask for. Please take care of yourself and know that it’s okay to take a step back when things feel too much. Your well-being is the most important thing."
-    
+
     # Suggest a productive activity based on detected keywords
-    if any(keyword in user_input.lower() for keyword in ["lonely", "lost", "sad", "overwhelmed", "academic", "exam"]):
+    if any(keyword in user_input.lower() for keyword in ["lonely", "lost", "sad", "overwhelmed"]):
         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)
+        # List of activities
+        hobbies = ["journaling", "yoga", "painting", "exercise", "meditation"]
+        activity = st.selectbox("Choose an activity you'd like to try:", hobbies)
         
-        # Check and debug if the search results are coming in
+        # Search YouTube for videos related to the selected activity
         try:
+            search = Search(activity)
             search_results = search.results[:3]  # limit results to 3 videos
+            
             if not search_results:
                 st.write(f"No results found for '{activity}'. Please try again.")
             else: