Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -248,16 +248,21 @@ 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, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
+
+                ## Get preview 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,
-                    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, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
-    
+                subtitle = get_txt(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)}
+                
                 ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle, file_path = self.generate_and_write_file(