Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -240,6 +240,18 @@ class WhisperBase(ABC):
                     *whisper_params,
                 )
                 
+#                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=file_format,
+#                    output_dir=self.output_dir
+#                )				
+#                
+#                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+
+                ## Add 1st file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle, file_path = self.generate_and_write_file(
                     file_name=file_name,
@@ -250,7 +262,19 @@ class WhisperBase(ABC):
                 )				
                 
                 files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
-			
+                
+                ## Add 2nd file as srt
+                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="srt",
+                    output_dir=self.output_dir
+                )				
+                
+                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+            
             total_result = ''
             total_info = ''
             total_time = 0