Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import shutil
 import whisper
-import os
+import os, time
 model = whisper.load_model("turbo")
 
 destination_language = "fa" # @param ["en","fa","es","it","fr","du"]
@@ -18,6 +18,11 @@ def format_time(seconds_float):
     sign = '-' if seconds_float < 0 else ''
     return f"{sign}{hours:02d}:{minutes:02d}:{seconds:02d},{milliseconds:03d}"
 
+def ytdown(text):
+    os.system(f'yt-dlp -f 18 -o video.mp4 {text} ')
+    time.sleep(5)
+    return ['video.mp4']
+    
 def subtitle(path):
     shutil.copyfile(path.name, 'video.mp4')
     result = model.transcribe('video.mp4')
@@ -42,6 +47,13 @@ def subtitle2(text):
 gr.set_static_paths(paths=["/home/user/app"])
 
 with gr.Blocks() as demo:
+    with gr.Tab("Download Youtube"):
+        gr.Markdown("Enter your link")
+        with gr.Row():
+            inp = gr.Textbox()
+            out = gr.File()
+        btn = gr.Button("Download it")
+        btn.click(fn=ytdown, inputs=inp, outputs=out)
     with gr.Tab("From Your PC"):
         gr.Markdown("Upload Video File for Auto Subtitle")
         with gr.Row():