Update app.py

app.py

@@ -98,14 +98,15 @@ class App:
         with gr.Row():
             with gr.Column(scale=1):
                 with gr.Row():
-                    input_multi = gr.Radio(["Single", "Multiple"], label="Process one or multiple files", value="Single")
+                    input_multi = gr.Radio(["Audio", "Video", "Multiple"], label="Process one or multiple files", value="Audio")
                 with gr.Row():
                     dd_file_format = gr.Dropdown(choices=["CSV","SRT","TXT"], value="CSV", label="Output format", multiselect=True, interactive=True, visible=True)
                 with gr.Row():                    
                     cb_timestamp_preview = gr.Checkbox(value=whisper_params["add_timestamp_preview"],label="Show preview with timestamps", interactive=True)
                     cb_timestamp_file = gr.Checkbox(value=whisper_params["add_timestamp_file"], label="Add timestamp to filenames", interactive=True)
             with gr.Column(scale=4):
-                input_file = gr.Video(type='filepath', elem_id="audio_input", show_download_button=True, visible=True, interactive=True)
+                input_file_audio = gr.Audio(type='filepath', elem_id="audio_input", show_download_button=True, visible=True, interactive=True)
+                input_file_video = gr.Video(elem_id="audio_input", show_download_button=True, visible=False, interactive=True)
                 input_file_multi = gr.Files(label="Upload one or more audio/video files here", elem_id="audio_input", type='filepath', file_count="multiple", allow_reordering=True, file_types=["audio","video"], visible=False, interactive=True)                        
                 
         with gr.Row():
@@ -267,7 +268,8 @@ class App:
                 uvr_save_file=cb_uvr_save_file, uvr_enable_offload=cb_uvr_enable_offload
             ),
             input_multi,
-            input_file,
+            input_file_audio,
+            input_file_video,
             input_file_multi,
             dd_file_format,
             cb_timestamp_file,
@@ -316,7 +318,7 @@ class App:
                                                  visible=self.args.colab,
                                                  value="")
 
-                    whisper_params, input_multi, input_file, input_file_multi, dd_file_format, cb_timestamp_file, cb_translate_output, dd_translate_model, dd_target_lang, cb_timestamp_preview = self.create_whisper_parameters()
+                    whisper_params, input_multi, input_file_audio, input_file_video, input_file_multi, dd_file_format, cb_timestamp_file, cb_translate_output, dd_translate_model, dd_target_lang, cb_timestamp_preview = self.create_whisper_parameters()
 
                     with gr.Row():
                         btn_run = gr.Button("Transcribe", variant="primary")
@@ -330,14 +332,14 @@ class App:
                             files_subtitles = gr.Files(label="Output data", interactive=False, file_count="multiple")
                             # btn_openfolder = gr.Button('📂', scale=1)                        
 
-                    params = [input_file, input_file_multi, input_multi, tb_input_folder, dd_file_format, cb_timestamp_file, cb_translate_output, dd_translate_model, dd_target_lang, cb_timestamp_preview]
+                    params = [input_file_audio, input_file_multi, input_multi, tb_input_folder, dd_file_format, cb_timestamp_file, cb_translate_output, dd_translate_model, dd_target_lang, cb_timestamp_preview]
  
                     btn_run.click(fn=self.whisper_inf.transcribe_file,
                                   inputs=params + whisper_params.as_list(),
                                   outputs=[tb_indicator, files_subtitles, tb_info])
                     # btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)
 
-                    input_multi.change(fn=self.update_visibility,inputs=input_multi,outputs=[input_file,input_file_multi])
+                    input_multi.change(fn=self.update_visibility,inputs=input_multi,outputs=[input_file_audio,input_file_video,input_file_multi])
                 
                 with gr.TabItem("Device info"):  # tab2
                     with gr.Column():
@@ -376,10 +378,12 @@ class App:
 
     @staticmethod
     def update_visibility(radio_text):
-        if radio_text == "Single":
-            return gr.update(visible=True), gr.update(visible=False)
+        if radio_text == "Audio":
+            return gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)
+        elif radio_text == "video":
+            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
         else:
-            return gr.update(visible=False), gr.update(visible=True)
+            return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
 
 # Create the parser for command-line arguments
 parser = argparse.ArgumentParser()