Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -277,11 +277,11 @@ class WhisperBase(ABC):
             total_info = ''
             total_time = 0
             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"Language {info['lang']} detected for file '{file_name}{file_ext}'"
-                break
+                if {info["path"]}.endswith(".txt"):
+                    total_result += f'{info["subtitle"]}'
+                    total_time += info["time_for_task"]
+                    #total_info += f'{info["lang"]}'
+                    total_info += f"Language {info['lang']} detected for file '{file_name}{file_ext}'"
             
             #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 process done in {self.format_time(total_time)}"