Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -249,7 +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, "input": (file_name+file_ext)}
+                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "input_source_file": (file_name+file_ext)}
 
                 ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -279,13 +279,12 @@ class WhisperBase(ABC):
             for file_name, info in files_info.items():
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
-                #total_info += f'{info["lang"]}'
-                total_info += f"Input file: {info['input']}\nLanguage prediction: {info['lang']}\n"
+                total_info += f'Input file: {info["input_source_file"]}\nLanguage prediction: {info["lang"]}\n'
             
             #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)}"
             result_str = total_result
-            result_file_path = [info['path'] for info in files_info.values()]
+            result_file_path = [info['path'] for info in files_to_download.values()]
 
             return [result_str, result_file_path, total_info]