add url download

app.py

@@ -12,13 +12,18 @@ with gr.Blocks() as demo:
     with gr.Column():
         with gr.Row():
             with gr.Column():
+                with gr.Row():
+                    input_url.render()
+                    url_download.render()
+                    url_download.click(get_video_from_url,
+                                       input_url, input_video)
                 input_video.render()
                 input_video.change(get_duration, input_video, [
                     start_time, end_time])
+            with gr.Column():
                 with gr.Row():
                     start_time.render()
                     end_time.render()
-            with gr.Column():
                 lang_radio.render()
                 model_dropdown.render()
                 summary_checkbox.render()

file_name.py

@@ -3,6 +3,7 @@ import os
 
 current_working_directory = os.getcwd()
 
+download_file = f"{current_working_directory}/download_video.mp4"
 audio_file = f"{current_working_directory}/input.wav"
 transcribe_file = f"{current_working_directory}/transcribe.txt"
 subtitle_file = f"{current_working_directory}/subtitle.srt"

lang_ui.py

@@ -18,6 +18,8 @@ english_ui = {
     "lang_radio_warning": "Please choose a language for the video",
     "model_dropdown_warning": "Please choose a model size for the video",
 
+    "input_url_label": "Insert video url",
+    "download_button_value": "Download video",
     "input_video_label": "Insert video",
     "start_time_label": "Start time",
     "end_time_label": "End time",
@@ -67,6 +69,8 @@ indonesia_ui = {
     "lang_radio_warning": "Mohon pilih bahasa yang digunakan dalam video",
     "model_dropdown_warning": "Mohon pilih ukuran model yang digunakan untuk video",
 
+    "input_url_label": "Masukkan link video",
+    "download_button_value": "Unduh video",
     "input_video_label": "Masukkan video",
     "start_time_label": "Waktu mulai",
     "end_time_label": "Waktu selesai",

ui.py

@@ -1,6 +1,8 @@
 from lang_ui import get_ui_lang
 import gradio as gr
+import file_name
 import ffmpeg
+import urllib.request
 
 # Display available langauges and set default UI language
 ui_lang_index = 1
@@ -16,6 +18,10 @@ ui_lang_radio = gr.Radio(
 top_markdown = gr.Markdown(
     ui_lang["top_markdown"])
 # TODO Add video by link
+input_url = gr.Textbox(
+    max_lines=1, label=ui_lang["input_url_label"], interactive=True)
+url_download = gr.Button(
+    ui_lang["download_button_value"], interactive=True)
 input_video = gr.Video(
     label=ui_lang["input_video_label"], interactive=True)
 start_time = gr.Textbox(
@@ -61,6 +67,17 @@ def time_str(t):
                                             round(t % 60))
 
 
+def get_video_from_url(url):
+    try:
+        urllib.request.urlretrieve(url, file_name.download_file)
+        print("Download finished")
+        return file_name.download_file
+    except:
+        print("Download failed")
+        gr.Warning(ui_lang["input_video_warning"])
+        return None
+
+
 def get_duration(input_file):
     print("Checking file")
     if input_file is None: