Spaces:
Running
Running
| import gradio as gr | |
| from modules.model_Inference import WhisperInference | |
| import os | |
| from ui.htmls import CSS,MARKDOWN | |
| from modules.youtube_manager import get_ytmetas | |
| def open_output_folder(): | |
| folder_path = "outputs" | |
| if os.path.exists(folder_path): | |
| os.system(f"start {folder_path}") | |
| else: | |
| print(f"The folder {folder_path} does not exist.") | |
| whisper_inf = WhisperInference() | |
| block = gr.Blocks(css=CSS).queue(api_open=False) | |
| with block: | |
| with gr.Row(): | |
| with gr.Column(): | |
| gr.Markdown(MARKDOWN,elem_id="md_project") | |
| with gr.Tabs(): | |
| with gr.TabItem("File"): # tab1 | |
| with gr.Row(): | |
| input_file = gr.File(type="file", label="Upload File here") | |
| with gr.Row(): | |
| dd_model = gr.Dropdown(choices=whisper_inf.available_models,value="large-v2",label="Model") | |
| dd_lang = gr.Dropdown(choices=["Automatic Detection"]+whisper_inf.available_langs,value="Automatic Detection",label="Language") | |
| dd_subformat = gr.Dropdown(["SRT","WebVTT"],value="SRT",label="Subtitle Format") | |
| with gr.Row(): | |
| cb_translate = gr.Checkbox(value=False,label="Translate to English?",interactive=True) | |
| with gr.Row(): | |
| btn_run = gr.Button("GENERATE SUBTITLE FILE") | |
| with gr.Row(): | |
| tb_indicator = gr.Textbox(label="Output") | |
| btn_openfolder = gr.Button('📂').style(full_width=False) | |
| btn_run.click(fn=whisper_inf.transcribe_file,inputs=[input_file,dd_model,dd_lang,dd_subformat,cb_translate],outputs=[tb_indicator]) | |
| btn_openfolder.click(fn=open_output_folder,inputs=[],outputs=[]) | |
| with gr.TabItem("Youtube"): # tab2 | |
| with gr.Row(): | |
| tb_youtubelink = gr.Textbox(label="Youtube Link" ) | |
| with gr.Row().style(equal_height=True): | |
| with gr.Column(): | |
| img_thumbnail = gr.Image(label="Youtube Thumbnail") | |
| with gr.Column(): | |
| tb_title = gr.Label(label="Youtube Title") | |
| tb_description = gr.Textbox(label="Youtube Description",max_lines=15) | |
| with gr.Row(): | |
| dd_model = gr.Dropdown(choices=whisper_inf.available_models,value="large-v2",label="Model") | |
| dd_lang = gr.Dropdown(choices=["Automatic Detection"]+whisper_inf.available_langs,value="Automatic Detection",label="Language") | |
| dd_subformat = gr.Dropdown(choices=["SRT","WebVTT"],value="SRT",label="Subtitle Format") | |
| with gr.Row(): | |
| cb_translate = gr.Checkbox(value=False,label="Translate to English?",interactive=True) | |
| with gr.Row(): | |
| btn_run = gr.Button("GENERATE SUBTITLE FILE") | |
| with gr.Row(): | |
| tb_indicator = gr.Textbox(label="Output") | |
| btn_openfolder = gr.Button('📂').style(full_width=False) | |
| btn_run.click(fn=whisper_inf.transcribe_youtube,inputs=[tb_youtubelink,dd_model,dd_lang,dd_subformat,cb_translate],outputs=[tb_indicator]) | |
| tb_youtubelink.change(get_ytmetas,inputs=[tb_youtubelink],outputs=[img_thumbnail,tb_title,tb_description]) | |
| btn_openfolder.click(fn=open_output_folder,inputs=[],outputs=[]) | |
| with gr.TabItem("Mic"): # tab3 | |
| with gr.Row(): | |
| mic_input = gr.Microphone(label="Record with Mic",type="filepath",interactive=True) | |
| with gr.Row(): | |
| dd_model = gr.Dropdown(choices=whisper_inf.available_models,value="large-v2",label="Model") | |
| dd_lang = gr.Dropdown(choices=["Automatic Detection"]+whisper_inf.available_langs,value="Automatic Detection",label="Language") | |
| dd_subformat = gr.Dropdown(["SRT","WebVTT"],value="SRT",label="Subtitle Format") | |
| with gr.Row(): | |
| cb_translate = gr.Checkbox(value=False,label="Translate to English?",interactive=True) | |
| with gr.Row(): | |
| btn_run = gr.Button("GENERATE SUBTITLE FILE") | |
| with gr.Row(): | |
| tb_indicator = gr.Textbox(label="Output") | |
| btn_openfolder = gr.Button('📂').style(full_width=False) | |
| btn_run.click(fn=whisper_inf.transcribe_mic,inputs=[mic_input,dd_model,dd_lang,dd_subformat,cb_translate],outputs=[tb_indicator]) | |
| btn_openfolder.click(fn=open_output_folder,inputs=[],outputs=[]) | |
| block.launch() | |