Update app.py

app.py

@@ -50,39 +50,62 @@ def add_local_files(module):
     elif module == "paper":
         return [{"title": os.path.basename(file_path), "url": None, "file_path": file_path, "type": "paper"}]
 
+import os
+import re
+from yt_dlp import YoutubeDL
+
 def download_youtube_video(video_url, output_dir, title=None):
     """Download a YouTube video using yt_dlp."""
-    sanitized_title = re.sub(r'[\\/*?:"<>|]', '_', title) if title else None
+    sanitized_title = re.sub(r'[\\/*?:"<>|]', '_', title) if title else "unknown_title"
     ydl_opts = {
         'quiet': True,
-        'outtmpl': f"{output_dir}/{sanitized_title}.%(ext)s" if sanitized_title else f"{output_dir}/%(title)s.%(ext)s",
+        'outtmpl': f"{output_dir}/{sanitized_title}.%(ext)s",
         'format': 'best',
     }
     try:
         with YoutubeDL(ydl_opts) as ydl:
-            ydl.download([video_url])
-        return os.path.join(output_dir, f"{sanitized_title}.mp4")
+            info = ydl.extract_info(video_url, download=True)
+            # Extract the final downloaded file path
+            downloaded_file = ydl.prepare_filename(info)
+        return downloaded_file
     except Exception as e:
         print(f"Failed to download video {video_url}. Error: {e}")
         return None
 
-def fetch_and_download_youtube_video(query, output_dir="./downloads"):
-    """Fetch and download the best YouTube video for a query."""
+def fetch_and_download_youtube_video(query, output_dir="./videos"):
+    """Fetch and download a YouTube video based on a query."""
+    print(f"Fetching YouTube video for query: '{query}'")
     ydl_opts = {
-    'quiet': True,
-    'outtmpl': f"{output_dir}/{sanitized_title}.%(ext)s",
-    'format': 'best',
-    'cookiesfrombrowser': ('chrome',),  # Adjust for your browser (e.g., 'firefox', 'edge')
-}
+        'quiet': True,
+        'format': 'best',
+        'outtmpl': f"{output_dir}/%(title)s.%(ext)s",  # Default template
+    }
 
     try:
         with YoutubeDL(ydl_opts) as ydl:
-            search_results = ydl.extract_info(query, download=False)
-            video = search_results['entries'][0]  # Get the first result
-            video_title = video['title']
-            video_url = video['webpage_url']
+            # Perform a search for the query on YouTube
+            search_results = ydl.extract_info(f"ytsearch:{query}", download=False)
+            
+            if 'entries' not in search_results or len(search_results['entries']) == 0:
+                print(f"No YouTube results found for query: '{query}'")
+                return []
+
+            video_info = search_results['entries'][0]  # Take the first result
+            video_title = video_info.get("title", "unknown_title")
+            video_url = video_info.get("webpage_url", None)
+            
+            if not video_url:
+                print("No URL found for the video.")
+                return []
+
+            # Download the video
             local_path = download_youtube_video(video_url, output_dir, title=video_title)
+            if not local_path:
+                return []
+
+            print(f"Successfully downloaded video: {video_title}")
             return [{"title": video_title, "url": video_url, "file_path": local_path, "type": "video"}]
+
     except Exception as e:
         print(f"Error fetching YouTube video for query '{query}': {e}")
         return []