Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -254,26 +254,26 @@ class WhisperBase(ABC):
                 ## 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+"_txt",
+                    file_name=file_name,
                     transcribed_segments=transcribed_segments,
                     add_timestamp=add_timestamp,
                     file_format="txt",
                     output_dir=self.output_dir
                 )				
                 
-                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+                files_info["txt_file"] = {"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+"_srt",
+                    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}
+                files_info["srt_file"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
             
             total_result = ''
             total_info = ''