Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -249,8 +249,7 @@ class WhisperBase(ABC):
                     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}
+                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))
@@ -261,7 +260,7 @@ class WhisperBase(ABC):
                     file_format="txt",
                     output_dir=self.output_dir
                 )				
-                files_to_download[file_name+"_txt"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+                files_to_download[file_name+"_txt"] = {"path": file_path}
                 
                 ## Add output file as srt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -272,7 +271,7 @@ class WhisperBase(ABC):
                     file_format="srt",
                     output_dir=self.output_dir
                 )				
-                files_to_download[file_name+"_srt"] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language}
+                files_to_download[file_name+"_srt"] = {"path": file_path}
             
             total_result = ''
             total_info = ''
@@ -281,7 +280,7 @@ class WhisperBase(ABC):
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
                 #total_info += f'{info["lang"]}'
-                total_info += f"Processed file: {file}\nLanguage prediction: {info['lang']}"
+                total_info += f"Input file: {info['input']}\nLanguage prediction: {info['lang']}"
             
             #result_str = f"Processing of file '{file_name}{file_ext}' done in {self.format_time(total_time)}:\n\n{total_result}"
             total_info += f"\nTranscription duration: {self.format_time(total_time)}"