Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -388,14 +388,19 @@ class WhisperBase(ABC):
             total_time = 0
             file_count = 0
             for file_name, info in files_info.items():
-                total_result += f'{info["subtitle"]}'
-                
-                total_time += info["time_for_task"]
+                                
+                file_count += 1
                 
-                if file_count += 1
                 if file_count > 1:
-                    total_info += f'\n'    
-                
+                    total_info += f'\n' 
+
+                if file_count_total > 1:
+                    if file_count > 1:
+                        total_result += f'\n' 
+                    total_result += f'Transcription of media file\t{info["input_source_file"]}:\n'
+
+                total_time += info["time_for_task"]
+                total_result += f'{info["subtitle"]}'
                 total_info += f'Media file:\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
                 
                 if params.is_translate: