Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -440,7 +440,7 @@ class WhisperBase(ABC):
         add_timestamp: bool
             Determines whether to add a timestamp to the end of the filename.
         file_format: str
-            File format to write. Supported formats: [SRT, WebVTT, txt]
+            File format to write. Supported formats: [SRT, WebVTT, txt, csv]
         output_dir: str
             Directory path of the output
 
@@ -471,6 +471,10 @@ class WhisperBase(ABC):
             content = get_txt(transcribed_segments)
             output_path += '.txt'
 
+        elif file_format == "csv":
+            content = get_txt(transcribed_segments)
+            output_path += '.csv'
+
         write_file(content, output_path)
         return content, output_path