Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -214,7 +214,7 @@ class WhisperBase(ABC):
                         translate_output: bool = False,
                         translate_model: str = "",
                         target_lang: str = "",
-                        preview_format: str = "",
+                        preview_output: bool = False,
                         progress=gr.Progress(),
                         *whisper_params,
                         ) -> list:
@@ -238,8 +238,8 @@ class WhisperBase(ABC):
             Translation model to use
         target_lang: str
             Target language to use
-        preview_format: str
-            Preview format to use
+        preview_output: bool
+            Preview option
         progress: gr.Progress
             Indicator to show progress directly in gradio.
         *whisper_params: tuple
@@ -327,13 +327,12 @@ class WhisperBase(ABC):
                 
                 ## Get preview
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
-                
-                if preview_format == "text":
-                    subtitle = get_plaintext(transcribed_segments)
-                if preview_format == "text with timestamps":
+
+                ## With or without timestamps
+                if preview_output:
                     subtitle = get_txt(transcribed_segments)
-                if preview_format == "csv":
-                    subtitle = get_csv(transcribed_segments)
+                else:
+                    subtitle = get_plaintext(transcribed_segments)
                 
                 files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext), "translation": translation_note, "transcription": transcription_note}