Update app.py

app.py

@@ -10,6 +10,7 @@ from modules.youtube_manager import get_ytmetas
 from modules.deepl_api import DeepLAPI
 from modules.whisper_parameter import *
 
+
 # Ensure the outputs directory exists
 def ensure_output_directory():
     output_directories = ["outputs", "outputs/translations"]
@@ -17,6 +18,7 @@ def ensure_output_directory():
         if not os.path.exists(directory):
             os.makedirs(directory)
 
+
 class App:
     def __init__(self, args):
         ensure_output_directory()  # Making sure the output directories exist
@@ -34,10 +36,11 @@ class App:
         if whisper_type in ["faster_whisper", "faster-whisper"]:
             whisper_inf = FasterWhisperInference()
             whisper_inf.model_dir = self.args.faster_whisper_model_dir
-        if whisper_type in ["whisper"]:
+        elif whisper_type == "whisper":
             whisper_inf = WhisperInference()
             whisper_inf.model_dir = self.args.whisper_model_dir
         else:
+            # Default to FasterWhisperInference
             whisper_inf = FasterWhisperInference()
             whisper_inf.model_dir = self.args.faster_whisper_model_dir
         return whisper_inf
@@ -140,16 +143,14 @@ class App:
                             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=self.whisper_inf.available_models, value="large-v2",
-                                               label="Model")
+                        dd_model = gr.Dropdown(choices=self.whisper_inf.available_models, value="large-v2", label="Model")
                         dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,
                                               value="Automatic Detection", label="Language")
                         dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
                     with gr.Row():
                         cb_translate = gr.Checkbox(value=False, label="Translate to English?", interactive=True)
                     with gr.Row():
-                        cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename",
-                                                   interactive=True)
+                        cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename", interactive=True)
                     with gr.Accordion("VAD Options", open=False, visible=isinstance(self.whisper_inf, FasterWhisperInference)):
                         cb_vad_filter = gr.Checkbox(label="Enable Silero VAD Filter", value=False, interactive=True)
                         sd_threshold = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label="Speech Threshold", value=0.5)
@@ -197,6 +198,7 @@ class App:
                                                              min_silence_duration_ms=nb_min_silence_duration_ms,
                                                              window_size_sample=nb_window_size_sample,
                                                              speech_pad_ms=nb_speech_pad_ms)
+
                     btn_run.click(fn=self.whisper_inf.transcribe_youtube,
                                   inputs=params + whisper_params.to_list(),
                                   outputs=[tb_indicator, files_subtitles])
@@ -262,6 +264,7 @@ class App:
                                                              min_silence_duration_ms=nb_min_silence_duration_ms,
                                                              window_size_sample=nb_window_size_sample,
                                                              speech_pad_ms=nb_speech_pad_ms)
+
                     btn_run.click(fn=self.whisper_inf.transcribe_mic,
                                   inputs=params + whisper_params.to_list(),
                                   outputs=[tb_indicator, files_subtitles])
@@ -275,15 +278,12 @@ class App:
 
                     with gr.TabItem("DeepL API"):  # sub tab1
                         with gr.Row():
-                            tb_authkey = gr.Textbox(label="Your Auth Key (API KEY)",
-                                                    value="")
+                            tb_authkey = gr.Textbox(label="Your Auth Key (API KEY)", value="")
                         with gr.Row():
                             dd_deepl_sourcelang = gr.Dropdown(label="Source Language", value="Automatic Detection",
-                                                              choices=list(
-                                                                  self.deepl_api.available_source_langs.keys()))
+                                                              choices=list(self.deepl_api.available_source_langs.keys()))
                             dd_deepl_targetlang = gr.Dropdown(label="Target Language", value="English",
-                                                              choices=list(
-                                                                  self.deepl_api.available_target_langs.keys()))
+                                                              choices=list(self.deepl_api.available_target_langs.keys()))
                         with gr.Row():
                             cb_deepl_ispro = gr.Checkbox(label="Pro User?", value=False)
                         with gr.Row():
@@ -294,13 +294,11 @@ class App:
                             btn_openfolder = gr.Button('📂', scale=1)
 
                     btn_run.click(fn=self.deepl_api.translate_deepl,
-                                  inputs=[tb_authkey, file_subs, dd_deepl_sourcelang, dd_deepl_targetlang,
-                                          cb_deepl_ispro],
+                                  inputs=[tb_authkey, file_subs, dd_deepl_sourcelang, dd_deepl_targetlang, cb_deepl_ispro],
                                   outputs=[tb_indicator, files_subtitles])
 
                     btn_openfolder.click(fn=lambda: self.open_folder(os.path.join("outputs", "translations")),
-                                         inputs=None,
-                                         outputs=None)
+                                         inputs=None, outputs=None)
 
                     with gr.TabItem("NLLB"):  # sub tab2
                         with gr.Row():
@@ -308,11 +306,9 @@ class App:
                                                         choices=self.nllb_inf.available_models)
                             dd_nllb_sourcelang = gr.Dropdown(label="Source Language",
                                                              choices=self.nllb_inf.available_source_langs)
-                            dd_nllb_targetlang = gr.Dropdown(label="Target Language",
-                                                             choices=self.nllb_inf.available_target_langs)
-                       with gr.Row():
-                            cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename",
-                                                       interactive=True)
+                            dd_nllb_targetlang = gr.Dropdown(label="Target Language", choices=self.nllb_inf.available_target_langs)
+                        with gr.Row():
+                            cb_timestamp = gr.Checkbox(value=True, label="Add a timestamp to the end of the filename", interactive=True)
                         with gr.Row():
                             btn_run = gr.Button("TRANSLATE SUBTITLE FILE", variant="primary")
                         with gr.Row():
@@ -327,8 +323,7 @@ class App:
                                   outputs=[tb_indicator, files_subtitles])
 
                     btn_openfolder.click(fn=lambda: self.open_folder(os.path.join("outputs", "translations")),
-                                         inputs=None,
-                                         outputs=None)
+                                         inputs=None, outputs=None)
 
         # Launch the app with optional gradio settings
         launch_args = {}