Update app.py

app.py

@@ -1,11 +1,18 @@
-from pytube import YouTube
+import yt_dlp
 import gradio as gr
 
 def download_video(video_url):
     try:
-        yt = YouTube(video_url)
-        video_title = yt.title
-        yt.streams.get_highest_resolution().download()
+        # Set download options
+        ydl_opts = {
+            'outtmpl': '%(title)s.%(ext)s',  # Save file with video title as the name
+            'format': 'best',  # Download the best available format
+        }
+
+        # Download the video
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(video_url, download=True)
+            video_title = info_dict.get('title', 'Video')
         return f"✅ Downloaded: {video_title}"
     except Exception as e:
         return f"❌ Error: {str(e)}"