Update app.py

app.py

@@ -4,7 +4,7 @@ from gtts import gTTS
 from io import BytesIO
 import random
 import requests
-from youtubesearchpython import VideosSearch
+import yt_dlp
 
 # Initialize the empathy bot using llama_cpp
 llm = Llama.from_pretrained(
@@ -60,25 +60,24 @@ def generate_dynamic_response(user_input):
 
     return empathetic_response, suggested_activity, video_results
 
-# Function to search for YouTube videos
-def search_youtube_videos(query):
-    # Perform the search using youtube-search-python
-    videos_search = VideosSearch(query, limit=3)
-    results = videos_search.result()
-
-    # Collect video details
-    video_links = []
-    for video in results['videos']:
-        title = video['title']
-        link = video['link']
-        video_links.append((title, link))
 
-    if not video_links:
-        video_links = [("No relevant videos found", "")]  # Gracefully handle empty result
+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 💚")