Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -256,7 +256,7 @@ class WhisperBase(ABC):
             Input folder path to transcribe from gr.Textbox(). If this is provided, `files` will be ignored and
             this will be used instead.
         file_format: str
-            Subtitle File format to write from gr.Dropdown(). Supported format: [SRT, WebVTT, txt]
+            Subtitle File format to write from gr.Dropdown(). Supported format: [CSV, SRT, TXT]
         add_timestamp: bool
             Boolean value from gr.Checkbox() that determines whether to add a timestamp at the end of the subtitle filename.
         translate_output: bool
@@ -360,16 +360,10 @@ class WhisperBase(ABC):
                 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}
                 
                 ## Add output file as txt, srt and/or csv
-                for format in file_format:
+                for output_format in file_format:
                     file_name, file_ext = os.path.splitext(os.path.basename(file))
-                    subtitle, file_path = self.generate_and_write_file(
-                        file_name=file_name,
-                        transcribed_segments=transcribed_segments,
-                        add_timestamp=add_timestamp,
-                        file_format=format.lower(),
-                        output_dir=self.output_dir
-                    )
-                    files_to_download[file_name+"_"+format.lower()] = {"path": file_path}
+                    subtitle, file_path = self.generate_and_write_file(file_name,transcribed_segments,add_timestamp,output_format.lower(),self.output_dir)
+                    files_to_download[file_name+"_"+output_format.lower()] = {"path": file_path}
                 
             total_result = ""
             total_info = ""