Spaces:
Runtime error
Runtime error
File size: 5,635 Bytes
e698260 96f6a7b e698260 96f6a7b e698260 96f6a7b e698260 4198c04 e698260 4198c04 e698260 96f6a7b e698260 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 |
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
available_ui_lang = ["English", "Bahasa Indonesia"]
ui_lang = get_ui_lang(ui_lang_index)
lang_radio_choice = 1
model_dropdown_choice = 2
# Transcribe components
ui_lang_radio = gr.Radio(
available_ui_lang, type="index", value=available_ui_lang[ui_lang_index], interactive=True, show_label=False)
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(
max_lines=1, placeholder="00:00:00", label=ui_lang["start_time_label"], interactive=False)
end_time = gr.Textbox(
max_lines=1, placeholder="01:00:00", label=ui_lang["end_time_label"], interactive=False)
# TODO Use custom language
lang_radio = gr.Radio(
ui_lang["lang_radio_choices"], label=ui_lang["lang_radio_label"], info=ui_lang["lang_radio_info"], type='index', interactive=True)
model_dropdown = gr.Dropdown(
ui_lang["model_dropdown_choices"], label=ui_lang["model_dropdown_label"], info=ui_lang["model_dropdown_info"], type='index', interactive=True)
summary_checkbox = gr.Checkbox(
label=ui_lang["summary_checkbox_label"], info=ui_lang["summary_checkbox_info"], interactive=False)
start_button = gr.Button(
ui_lang["start_button_value"], variant="primary", interactive=True)
# Adjust components
middle_markdown = gr.Markdown(
ui_lang["middle_markdown"])
adjust_audio = gr.Audio(interactive=False)
adjust_speaker = gr.Textbox(
label=ui_lang["adjust_speaker_value"], interactive=False)
prev_button = gr.Button(ui_lang["prev_button_value"], interactive=False)
next_button = gr.Button(ui_lang["next_button_value"], interactive=False)
adjust_button = gr.Button(
ui_lang["adjust_button_value"], variant="primary", interactive=False)
# Result components
bottom_markdown = gr.Markdown(
ui_lang["bottom_markdown"]
)
output_video = gr.Video(label=ui_lang["output_video_label"], interactive=False)
output_file = gr.File(file_count="multiple", interactive=False)
output_transcribe = gr.Textbox(
label=ui_lang["output_transcribe_label"], interactive=False, show_copy_button=True)
output_summary = gr.Textbox(
interactive=False, show_copy_button=True)
def time_str(t):
return '{0:02d}:{1:02d}:{2:02d}'.format(round(t // 3600),
round(t % 3600 // 60),
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:
gr.Warning(ui_lang["input_video_warning"])
return [
start_time.update(None, interactive=False),
end_time.update(None, interactive=False)
]
print("Getting duration")
info_json = ffmpeg.probe(input_file)
print("Probing finished")
duration_seconds = float(info_json['format']['duration'])
duration_formatted = time_str(duration_seconds)
return [
start_time.update("00:00:00", interactive=True),
end_time.update(duration_formatted, interactive=True)
]
# Change language function
def change_lang(input):
global ui_lang
ui_lang = get_ui_lang(input)
print(f"Change language to {available_ui_lang[input]}")
return [
# Top
top_markdown.update(
ui_lang["top_markdown"]),
input_video.update(
label=ui_lang["input_video_label"]),
start_time.update(
label=ui_lang["start_time_label"]),
end_time.update(
label=ui_lang["end_time_label"]),
lang_radio.update(
choices=ui_lang["lang_radio_choices"], value=None, label=ui_lang["lang_radio_label"], info=ui_lang["lang_radio_info"],),
model_dropdown.update(
choices=ui_lang["model_dropdown_choices"], value=None, label=ui_lang["model_dropdown_label"], info=ui_lang["model_dropdown_info"]),
summary_checkbox.update(
label=ui_lang["summary_checkbox_label"], info=ui_lang["summary_checkbox_info"]),
start_button.update(
ui_lang["start_button_value"]),
# Middle
middle_markdown.update(
ui_lang["middle_markdown"]),
adjust_speaker.update(label=ui_lang["adjust_speaker_value"]),
prev_button.update(
ui_lang["prev_button_value"]),
next_button.update(
ui_lang["next_button_value"]),
adjust_button.update(
ui_lang["adjust_button_value"]),
# Bottom
bottom_markdown.update(
ui_lang["bottom_markdown"]),
output_video.update(label=ui_lang["output_video_label"]),
output_transcribe.update(label=ui_lang["output_transcribe_label"]),
]
# comp_to_update and change_lang return must always be in equal number
comp_to_update = [
top_markdown, input_video, start_time, end_time, lang_radio, model_dropdown, summary_checkbox, start_button, middle_markdown, adjust_speaker, prev_button, next_button, adjust_button, bottom_markdown, output_video, output_transcribe]
|