Update app.py

app.py

@@ -1,55 +1,49 @@
 import gradio as gr
-import yt_dlp
+from pytube import YouTube
 import os
-import io
 
-def download_youtube_video(youtube_url):
-    """Downloads a YouTube video using yt_dlp and returns the video data and filename."""
-    try:
-        ydl_opts = {
-            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',
-            'outtmpl': 'video.%(ext)s',
-        }
-
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(youtube_url, download=False)
-            video_title = info.get('title', 'video')
-            video_ext = info.get('ext', 'mp4')
-            
-            # Use a simple filename
-            filename = f"youtube_video.{video_ext}"
-            
-            # Download to memory
-            ydl_opts['outtmpl'] = '-'
-            ydl_opts['logtostderr'] = True
-
-            with io.BytesIO() as video_buffer:
-                ydl.download([youtube_url])
-                video_data = video_buffer.getvalue()
-
-            return video_data, filename, video_title
-
-    except Exception as e:
-        print("An error occurred:", e)
-        return None, None, None
-
-def app(video_link):
-    video_data, filename, video_title = download_youtube_video(video_link)
-    if video_data and filename:
-        return (filename, video_data), f"Download successful! Video: {video_title}"
+def download_youtube_videos(urls):
+    """Downloads multiple YouTube videos and returns a list of downloaded file paths."""
+    downloaded_files = []
+    for url in urls:
+        try:
+            yt = YouTube(url)
+            video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
+            filename = f"{yt.title}.mp4"
+            video.download(filename=filename)
+            downloaded_files.append(filename)
+        except Exception as e:
+            print(f"Error downloading {url}: {str(e)}")
+    
+    return downloaded_files
+
+def app(urls):
+    # Split the input string into a list of URLs
+    url_list = [url.strip() for url in urls.split('\n') if url.strip()]
+    
+    if not url_list:
+        return None, "No valid URLs provided. Please enter at least one YouTube URL."
+    
+    downloaded_files = download_youtube_videos(url_list)
+    
+    if downloaded_files:
+        # Create a list of (filename, file path) tuples for Gradio File component
+        file_tuples = [(f, f) for f in downloaded_files if os.path.exists(f)]
+        message = f"Successfully downloaded {len(file_tuples)} videos."
+        return file_tuples, message
     else:
-        return None, "Download failed. Please check the URL and try again."
+        return None, "Download failed. Please check the URLs and try again."
 
 with gr.Blocks() as interface:
-    gr.Markdown("## YouTube Video Downloader")
-    video_link = gr.Textbox(label="Enter YouTube link 🔗 To Download Video⬇️")
-    download_button = gr.Button("Download")
-    output = gr.File(label="Downloaded Video")
+    gr.Markdown("## YouTube Batch Video Downloader")
+    urls_input = gr.Textbox(label="Enter YouTube links (one per line) 🔗", lines=5)
+    download_button = gr.Button("Download Videos")
+    output = gr.File(label="Downloaded Videos", file_count="multiple")
     message = gr.Markdown()
 
     download_button.click(
         fn=app,
-        inputs=video_link,
+        inputs=urls_input,
         outputs=[output, message],
         api_name="download"
     )