Update app.py

app.py

@@ -27,7 +27,7 @@ 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 in ["whisper"]:
             whisper_inf = WhisperInference()
             whisper_inf.model_dir = self.args.whisper_model_dir
         else:
@@ -50,6 +50,15 @@ class App:
         else:
             return gr.Checkbox(visible=True, value=False, label="Translate to English?", interactive=True)
 
+    def transcribe_file_wrapper(self, file, file_format, timestamp, *whisper_params):
+        try:
+            result, output_file = self.whisper_inf.transcribe_file(file, file_format, timestamp, *whisper_params)
+            if not os.path.exists(output_file):
+                raise FileNotFoundError(f"Output file {output_file} does not exist.")
+            return result, output_file
+        except Exception as e:
+            return str(e), None
+
     def launch(self):
         with self.app:
             with gr.Row():
@@ -117,7 +126,7 @@ class App:
                                                              window_size_sample=nb_window_size_sample,
                                                              speech_pad_ms=nb_speech_pad_ms)
 
-                    btn_run.click(fn=self.whisper_inf.transcribe_file,
+                    btn_run.click(fn=self.transcribe_file_wrapper,
                                   inputs=params + whisper_params.to_list(),
                                   outputs=[tb_indicator, files_subtitles])
                     btn_openfolder.click(fn=lambda: self.open_folder("outputs"), inputs=None, outputs=None)