Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -410,10 +410,8 @@ class WhisperBase(ABC):
                 file_count += 1
                 
                 # Add filename & info as first line
-                title_line = ""
+                title_line = "#TAB#"*2
                 tabbed_space = "\t"*3
-                if add_timestamp_preview: title_line += "#TAB#"
-                if diarize_speakers: title_line += "#TAB#"
                 title_line += f'{tabbed_space}Media file:\t{info["input_source_file"]}'
                 title_line += f'\n{tabbed_space}Language:\t{info["lang"]} (probability {info["lang_prob"]}%)'
                 if params.is_translate: title_line += f'\n{tabbed_space}Translation:\t{info["transcription"]} (Handled by OpenAI Whisper)'
@@ -428,7 +426,7 @@ class WhisperBase(ABC):
             result_file_path = [info['path'] for info in files_to_download.values()]
             #total_info += f"Total processing time:\t{self.format_time((time_end-time_start).total_seconds())}"
 
-            return [(gr.update(value={"data": self.transform_text_to_list(total_result)})),result_file_path,total_info]
+            return [gr.update(value={"data": self.transform_text_to_list(total_result)}),result_file_path,total_info]
             #return [result_str,result_file_path,total_info]
         
         except Exception as e:
@@ -628,6 +626,12 @@ class WhisperBase(ABC):
         for temp_line in temp_inputdata:
             temp_line_list = []	
             temp_line_items = temp_line.split("#TAB#")
+
+            # Add tabs to match Dataframe columns
+            count_items = len(temp_line_items)
+            while len(temp_line_items)<3:
+                temp_line_items.insert(0,"")
+            
             for temp_line_item in temp_line_items:
                 temp_line_list.append(temp_line_item)	
             outputdata.append(temp_line_list)