Update Download_Video.py

Download_Video.py

@@ -1,8 +1,20 @@
-import pytube
+from yt_dlp import YoutubeDL
+import os
 
-def download_video(url):
- yt=pytube.YouTube(url)
- video_files=yt.streams.first()
- video_path=video_files.download()
- return video_path
+def download_video(url, download_dir=None):
+    try:
+        # Set download options
+        ydl_opts = {
+            'format': 'best',  # Download the best quality available
+            'outtmpl': os.path.join(download_dir, '%(title)s.%(ext)s') if download_dir else '%(title)s.%(ext)s',  # Save file with title as name
+            'quiet': True,  # Suppress output (set to False for debugging)
+        }
+
+        with YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=True)  # Extract video info and download
+            video_path = ydl.prepare_filename(info_dict)  # Get the path of the downloaded file
+            return video_path
+
+    except Exception as e:
+        return f"An error occurred: {e}"