Update main.py

main.py

@@ -1,5 +1,30 @@
 import whisper
+import yt_dlp
+import gradio as gr
 
+def yt_download(link):
+    if not link.strip():
+        gr.Info("You need to provide a download link.")
+        print("You need to provide a download link")
+        return None
+    ydl_opts = {
+        'format': 'bestaudio',
+        'outtmpl': '%(title)s',
+        'nocheckcertificate': True,
+        'ignoreerrors': True,
+        'no_warnings': True,
+        'quiet': True,
+        'extractaudio': True,
+        'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'wav'}],
+        'postprocessor_args': [
+            '-acodec', 'pcm_f32le'
+        ],
+    }
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        result = ydl.extract_info(link, download=True)
+        download_path = ydl.prepare_filename(result, outtmpl='%(title)s.wav')
+        
+    return download_path
 
 
 def whisper_(input_audio):