Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -279,6 +279,17 @@ class WhisperBase(ABC):
                     output_dir=self.output_dir
                 )
                 files_to_download[file_name+"_srt"] = {"path": file_path}
+
+                ## Add output file as csv
+                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="csv",
+                    output_dir=self.output_dir
+                )
+                files_to_download[file_name+"_csv"] = {"path": file_path}
             
             total_result = ''
             total_info = ''