Update app.py

app.py

@@ -18,10 +18,12 @@ st.set_page_config(
 )
 
 #@st.cache_resource
-#@st.cache_data
 def download_audio(link):
-    with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': 'video.mp3'}) as video:
-        video.download(link)
+    with yt_dlp.YoutubeDL({'extract_audio': True, 'format': 'bestaudio', 'outtmpl': '%(id)s.mp3'}) as video:
+        info_dict = video.extract_info(link, download = True)
+        id = info_dict['id']
+        video.download(link)    
+        return id
 
 #Load Whisper pipeline via HuggingFace
 @st.cache_resource
@@ -54,7 +56,7 @@ def load_nltk():
 
 #Make the ASR task
 @st.cache_data
-def audio_speech_recognition(_model_pipeline, video="video.mp3"):
+def audio_speech_recognition(_model_pipeline, video_id):
     return _model_pipeline(video, batch_size=64)["text"].strip()
 
 #Make the Summarization task
@@ -120,7 +122,7 @@ def main():
             with st.spinner("Carregando modelos..."):
 
                 if language == "Inglês (en)":
-                    download_audio(link)
+                    id = download_audio(link)
                     load_nltk()
                     whisper = load_whisper()
                     extractive = load_extractive()
@@ -133,7 +135,7 @@ def main():
                     st.header("Erro na seleção de linguagem.")
 
             with st.spinner("Transcrevendo texto..."):
-                transcript_text = audio_speech_recognition(whisper)
+                transcript_text = audio_speech_recognition(whisper, id)
 
             with model_1:
                 st.header("Texto Sumarizado:")