Deploy Gradio app with fixed issues

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import time
 import logging
@@ -7,6 +8,8 @@ from transformers import pipeline, AutoModelForSpeechSeq2Seq, AutoProcessor
 from transformers.utils import is_flash_attn_2_available
 from languages import get_language_names
 from subtitle_manager import Subtitle
+from pytube import YouTube
+import os
 
 logging.basicConfig(level=logging.INFO)
 last_model = None
@@ -47,6 +50,12 @@ def create_pipe(model, flash):
     )
     return pipe
 
+def download_youtube_audio(url):
+    yt = YouTube(url)
+    audio_stream = yt.streams.filter(only_audio=True).first()
+    output_file = audio_stream.download(filename='temp_audio')
+    return output_file
+
 def transcribe_webui_simple_progress(modelName, languageName, urlData, multipleFiles, microphoneData, task, flash,
                                     chunk_length_s, batch_size, progress=gr.Progress()):
     global last_model
@@ -82,7 +91,13 @@ def transcribe_webui_simple_progress(modelName, languageName, urlData, multipleF
     if multipleFiles:
         files += multipleFiles
     if urlData:
-        files.append(urlData)
+        try:
+            progress(0.2, desc="Downloading YouTube Audio..")
+            url_audio_file = download_youtube_audio(urlData)
+            files.append(url_audio_file)
+        except Exception as e:
+            logging.error(f"Failed to download YouTube audio: {e}")
+            return ["Error"], "Error", "Error"
     if microphoneData:
         files.append(microphoneData)
     logging.info(files)

requirements.txt

@@ -2,4 +2,7 @@
 torch>=2.1.1
 gradio==4.16.0
 transformers
-accelerate
+accelerate
+pytube
+optimum
+safetensors