Update app.py

app.py

@@ -3,7 +3,7 @@ import time
 import json
 import os
 import gradio as gr
-from pytube import YouTube
+import yt_dlp
 
 YT_embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/'
 YT_embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></iframe>'
@@ -14,17 +14,18 @@ SIMBALS_MAIN_DATABASE = 2
 embed_html1 = '<iframe src="https://open.spotify.com/embed/track/'
 embed_html2 = '" width="300" height="380" frameborder="0" allowtransparency="true" allow="encrypted-media"></iframe>'
 
+def download_audio_(link):
+  with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': '%(title)s.mp3'}) as video:
+    info_dict = video.extract_info(link, download = True)
+    video_title = info_dict['title']
+    video.download(link)    
+    return video_title
 
 def download_audio(id_video):
-    video = YouTube(id_video,use_oauth=True, allow_oauth_cache=True)
     id = id_video.split("?v=")[-1][:11]
-    print("ID youtube ", id)
-    audio_streams = video.streams.filter(only_audio=True)
-    audio_stream = audio_streams[0]
-    audio_file = audio_stream.download()
-    audio_parts = audio_file.split('/')[-1]
-    audio_title = '.'.join(audio_parts.split('.')[:-1])
-    embed_html_all = YT_embed_html1 + id + YT_embed_html2
+    audio_file = download_audio_(id_video)
+    audio_file = audio_file+'.mp3'
+    embed_html_all = embed_html1 + id +embed_html2
     return audio_file, audio_file, embed_html_all
 
 def process_url(input_path):