Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -260,7 +260,7 @@ class WhisperBase(ABC):
             List of files to transcribe from gr.Video()
         files_multi: list
             List of files to transcribe from gr.Files_multi()
-        input_multi: bool
+        input_multi: str
             Process single or multiple files
         input_folder_path: str
             Input folder path to transcribe from gr.Textbox(). If this is provided, `files` will be ignored and
@@ -294,6 +294,7 @@ class WhisperBase(ABC):
 
         try:
             file_count_total = 0
+            process_multiple = False
             files = ""
 
             if input_multi == "Audio":
@@ -303,6 +304,7 @@ class WhisperBase(ABC):
             else:
                 files = files_multi
                 file_count_total = len(files)
+                process_multiple = True
             
             if input_folder_path:
                 files = get_media_files(input_folder_path)
@@ -406,17 +408,23 @@ class WhisperBase(ABC):
             for file_name, info in files_info.items():
                                 
                 file_count += 1
-                
-                if file_count > 1:
-                    total_info += f'\n' 
-
-                if file_count_total > 1:
-                    if file_count > 1:
-                        total_result += f'\n'
-                    total_result += f'« Transcription of media file \'{info["input_source_file"]}\': »\n\n'
 
+                # Add filename as column if multiple file processing
+                if process_multiple:
+                    temp_lines = {info["subtitle"]}
+                    line_count = 0
+                    result_fixed = []
+                    for temp_line in temp_lines:
+                        line_count += 1
+                        if line_count==1:
+                            result_fixed.append(info["input_source_file"]+"\t"+temp_line)
+                        else:
+                            result_fixed.append("\t"+temp_line)
+                    total_result += result_fixed
+                else:
+                    total_result += f'{info["subtitle"]}'
+                    
                 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: