Change to yt-dlp to download youtube videos

download.py

@@ -64,46 +64,12 @@ def download_twitch(url, type):
         os.system(f'rm {AUDIO_FOLDER}/{DOWNLOAD_AUDIO_NAME}.{args.format}')
 
 def download_youtube_video(url):
-    yt = YouTube(url)
-
-    # Get all video streams
-    video_streams = yt.streams.filter(mime_type="video/mp4")
-
-    # Get max resolution of video streams
-    max_resolution = 0
-    for stream in video_streams:
-        if stream.audio_codec is not None:
-            resolution = int(stream.resolution[:-1])
-            if resolution > max_resolution:
-                max_resolution = resolution
-    max_resolution = f"{max_resolution}p"
-
-    # Get stream with max resolution
-    video_streams_max_resolution = video_streams.filter(resolution=max_resolution)
-    for stream in video_streams_max_resolution:
-        if stream.audio_codec is not None:
-            video_stream = stream
-    
-    # Download video
-    video_stream.download(filename=f'{VIDEO_FOLDER}/{DOWNLOAD_VIDEO_NAME}.{DOWNLOAD_VIDEO_FORMAT}')
+    command = f"yt-dlp -o '{VIDEO_FOLDER}/{DOWNLOAD_VIDEO_NAME}.{DOWNLOAD_VIDEO_FORMAT}' -f 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]' '{url}'"
+    os.system(command)
 
 def download_youtube_audio(url):
-    yt = YouTube(url)
-
-    # Get all audio streams
-    audio_streams = yt.streams.filter(mime_type="audio/mp4")
-
-    # Get sorted list of audio bitrates 
-    abr_list = []
-    for stream in audio_streams:
-        abr_list.append(stream.abr)
-    abr_list = sorted(set(abr_list))
-
-    # Get stream with max audio bitrate
-    audio_stream = audio_streams.filter(abr=abr_list[0]).first()
-
-    # Download audio
-    audio_stream.download(filename=f'{AUDIO_FOLDER}/{DOWNLOAD_AUDIO_NAME}.{DOWNLOAD_AUDIO_FORMAT}')
+    command = f"yt-dlp '{url}' -o '{AUDIO_FOLDER}/{DOWNLOAD_AUDIO_NAME}.{DOWNLOAD_AUDIO_FORMAT}' --extract-audio --audio-format mp3 --audio-quality 0"
+    os.system(command)
 
 def download_youtube(url, type):
     if type == DOWNLOAD_VIDEO:

requirements.txt

@@ -2,7 +2,8 @@
 gradio
 
 # Download youtube and twitch videos
-pytube
+# pytube
+yt-dlp
 twitch-dl
 
 # # mossformer --> separate speech audios