Download from youtube with pytube

download.py

@@ -66,14 +66,30 @@ def download_twitch(url, type):
         os.system(f'rm {AUDIO_FOLDER}/{DOWNLOAD_AUDIO_NAME}.{args.format}')
 
 def download_youtube_video(url):
-    user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-    command = f"yt-dlp -o '{VIDEO_FOLDER}/{DOWNLOAD_VIDEO_NAME}.{DOWNLOAD_VIDEO_FORMAT}' -f 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]' --user-agent '{user_agent}' --no-check-certificates --force-ipv4 --geo-bypass '{url}'"
-    os.system(command)
+    try:
+        yt = YouTube(url)
+        # Get stream with best resolution
+        video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
+        if video:
+            # Download video
+            video.download(output_path=VIDEO_FOLDER, filename=f'{DOWNLOAD_VIDEO_NAME}.{DOWNLOAD_VIDEO_FORMAT}')
+        else:
+            print('No se encontró un stream de video compatible')
+    except Exception as e:
+        print(f'Error descargando el video: {str(e)}')
 
 def download_youtube_audio(url):
-    user_agent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
-    command = f"yt-dlp '{url}' -o '{AUDIO_FOLDER}/{DOWNLOAD_AUDIO_NAME}.{DOWNLOAD_AUDIO_FORMAT}' --user-agent '{user_agent}' --no-check-certificates --force-ipv4 --geo-bypass --extract-audio --audio-format mp3 --audio-quality 0"
-    os.system(command)
+    try:
+        yt = YouTube(url)
+        # Get stream with best quality
+        audio = yt.streams.filter(only_audio=True).order_by('abr').desc().first()
+        if audio:
+            # Download audio
+            audio.download(output_path=AUDIO_FOLDER, filename=f'{DOWNLOAD_AUDIO_NAME}.{DOWNLOAD_AUDIO_FORMAT}')
+        else:
+            print('No se encontró un stream de audio compatible')
+    except Exception as e:
+        print(f'Error descargando el audio: {str(e)}')
 
 def download_youtube(url, type):
     if type == DOWNLOAD_VIDEO: