Update app.py

app.py

@@ -21,38 +21,64 @@ from utils import *
 import json
 import re
 from pytube import YouTube
+from yt_dlp import YoutubeDL
+from yt_dlp import YoutubeDL
+import os
+
 def download_video(url):
+    """
+    Downloads a video from a URL using yt-dlp.
+
+    Args:
+        url: The URL of the video to download.
+
+    Returns:
+        The path to the downloaded file, or None if an error occurred.
+    """
     print("Downloading...")
-    local_file = (
-        YouTube(url)
-        .streams.filter(progressive=True, file_extension="mp4")
-        .first()
-        .download()
-    )
-    print("Downloaded")
-    return local_file
+    try:
+        ydl_opts = {
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',  # Prioritize mp4
+            'merge_output_format': 'mp4', # Ensure final output is mp4
+            'outtmpl': '%(title)s.%(ext)s',  # Use title for filename
+        }
+        with YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=True)
+            filename = ydl.prepare_filename(info_dict) # Get the actual filename.
+            print("Downloaded")
+            return filename
+    except Exception as e:
+        print(f"Error during download: {e}")
+        return None
 
 def validate_youtube(url):
     """
     Validates a YouTube URL, checks if the video exists, and returns whether its length exceeds 10 minutes.
-    
+    Uses yt-dlp for more robust URL handling.
+
     :param url: str - YouTube video URL
     :return: bool - True if the URL is invalid or video is longer than 10 minutes, otherwise False
     """
     try:
-        yt = YouTube(url)
-        video_length = yt.length  # Video length in seconds
+        with YoutubeDL({'quiet': True, 'no_warnings': True}) as ydl:
+            info = ydl.extract_info(url, download=False)
+            video_length = info.get('duration')  # Video length in seconds
+
+            if video_length is None: # Handle cases where duration isn't available.
+                print("Could not determine video length.")
+                return True # Treat as invalid for now.  Consider returning None if you want to handle differently.
+
+            if video_length > 600:
+                print("Your video is longer than 10 minutes.")
+                return True
+            else:
+                print("Your video is 10 minutes or shorter.")
+                return False
+
     except Exception as e:
         print(f"Error: The provided URL is invalid or not accessible. ({e})")
         return True  # Return True since the URL is invalid
-
-    if video_length > 600:
-        print("Your video is longer than 10 minutes.")
-        return True
-    else:
-        print("Your video is 10 minutes or shorter.")
-        return False
-
+        
 def validate_url(url):
     import validators
     if not validators.url(url):