Improves the yt download reliablity

app.py

@@ -127,7 +127,7 @@ Tips to get better results:
 </div>
 </div>
 """
-
+from download_youtube import download_youtube_video_ytdlp
 
 def download_from_youtube(
         video_id,
@@ -188,6 +188,11 @@ def configure_input():
     )
     video_input = gr.Video(label="Upload Video", height=520)
     youtube_link = gr.Textbox(label="YouTube Link", value=None)
+    gr.Markdown(
+        "Note: Often, YouTube download can fail because the video may not be public or YouTube asks for Sign in."\
+        "We recommend downloading the video in other ways on your machine and uploading it here."\
+        " Alternatively, you can clone the repository and run the demo locally which can allow for Sign-in.",
+    )
     return [video_input, youtube_link]
 
 
@@ -267,8 +272,9 @@ def process_input(video, youtube_link):
             "YouTube Link cannot be empty if no video is provided."
         
         video_id = get_video_id_from_url(youtube_link)
-        video_path = download_from_youtube(
-            video_id, save_dir="/tmp/", convert_to_mp4=False,
+        print("Video ID:", video_id)
+        video_path = download_youtube_video_ytdlp(
+            video_id, save_dir="/tmp/", 
         )
 
         # Get predictions

download_youtube.py

@@ -0,0 +1,53 @@
+"""Util functions to download videos from youtube."""
+from pytube import YouTube
+import yt_dlp
+from glob import glob
+import os
+
+
+def download_youtube_video_pytube(video_id, save_path="."):
+    url = f"https://www.youtube.com/watch?v={video_id}" 
+    try:
+        # Create a YouTube object
+        yt = YouTube(url)
+        
+        # Select the highest resolution stream
+        stream = yt.streams.get_highest_resolution()
+        
+        # Download the video
+        stream.download(output_path=save_path)
+        print(f"Downloaded: {yt.title}")
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
+
+def download_youtube_video_ytdlp(video_id, save_dir="/tmp/"):
+    try:
+        # Construct the URL using the video ID
+        url = f"https://www.youtube.com/watch?v={video_id}"
+        
+        # yt-dlp options
+        ydl_opts = {
+            'format': 'best',  # Download the best video quality
+            'outtmpl': f'{save_dir}/{video_id}.%(ext)s',  # Save using video ID as the filename
+            'noprogress': True,  # Suppress progress bar
+        }
+        
+        # Download the video
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+        
+        print(f"Downloaded video with ID: {video_id} to {save_dir}")
+        saved_path = glob(f"{save_dir}/{video_id}.*")[0]
+        assert os.path.exists(saved_path), f"File not found: {saved_path}"
+        return saved_path
+    except Exception as e:
+        print(f"An error occurred for video ID {video_id}: {e}")
+
+
+if __name__ == "__main__":
+    # Example usage
+    # video_id = "FwZfVEuCnI4"
+    video_id = "PCHIWN-Wi3M"
+    # download_youtube_video(video_id)
+    download_youtube_video_ytdlp(video_id)