Update app.py

app.py

@@ -7,21 +7,22 @@ import pickle
 import gradio as gr
 import numpy as np
 import faiss
-from pytube import YouTube
+import yt_dlp
 
 embed_html1 = '<iframe width="560" height="315" src="https://www.youtube.com/embed/'
 embed_html2 = '" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen></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])
+    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