Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -248,8 +248,9 @@ class WhisperBase(ABC):
                     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,"input": file_name+file_ext}
+                )
+                
+                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "input": (file_name+file_ext)}
 
                 ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -259,8 +260,8 @@ class WhisperBase(ABC):
                     add_timestamp=add_timestamp,
                     file_format="txt",
                     output_dir=self.output_dir
-                )				
-                files_to_download[file_name+"_txt"] = {"path": file_path}
+                )
+                files_to_download[file_name+"_txt"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
                 
                 ## Add output file as srt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -270,8 +271,8 @@ class WhisperBase(ABC):
                     add_timestamp=add_timestamp,
                     file_format="srt",
                     output_dir=self.output_dir
-                )				
-                files_to_download[file_name+"_srt"] = {"path": file_path}
+                )
+                files_to_download[file_name+"_srt"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
             
             total_result = ''
             total_info = ''