Update app.py

app.py

@@ -1,27 +1,22 @@
 import gradio as gr
-from pytube import YouTube
+import yt_dlp as youtube_dl
 import os
 
 def download_youtube_video(url):
     try:
-        # Criar um objeto YouTube
-        yt = YouTube(url)
+        # Configurações do yt-dlp
+        ydl_opts = {
+            'format': 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best[ext=mp4]/best',  # Forçar formato MP4
+            'outtmpl': 'downloads/%(title)s.%(ext)s',  # Salvar na pasta "downloads" com o título do vídeo
+            'quiet': True,  # Evitar logs desnecessários
+            'no_warnings': True,  # Ignorar avisos
+        }
 
-        # Selecionar a stream de maior resolução em formato MP4
-        stream = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
-
-        # Verificar se a stream foi encontrada
-        if not stream:
-            return "Erro: Nenhuma stream MP4 disponível."
+        # Criar a pasta de downloads, se não existir
+        os.makedirs("downloads", exist_ok=True)
 
         # Fazer o download do vídeo
-        output_path = stream.download(output_path="downloads")
-
-        # Retornar o caminho do arquivo baixado
-        return output_path
-
-    except Exception as e:
-        return f"Erro ao baixar o vídeo: {str(e)}"
+        with youtube_dl.YoutubeDL(
 
 # Interface Gradio
 iface = gr.Interface(