Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -416,7 +416,7 @@ class WhisperBase(ABC):
                 if params.is_translate: title_line += f'\n{tabbed_space}Translation:\t{info["transcription"]} (Handled by OpenAI Whisper)'
                 if translate_output: title_line += f'\n{tabbed_space}Translation:\t{info["translation"]} (Handled by Facebook NLLB)'                    
 
-                total_result += "#TAB##TAB#"+title_line+"#NEWLINE#"+(((info["subtitle"].rstrip("\n")).replace("\t","#TAB#")).replace("\n","#NEWLINE#"))
+                total_result += title_line+"#NEWLINE#"+(((info["subtitle"].rstrip("\n")).replace("\t","#TAB#")).replace("\n","#NEWLINE#"))
                 total_time += info["time_for_task"]
             
             time_end = datetime.now()
@@ -424,7 +424,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,add_timestamp_preview,diarize_speakers)}),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:
@@ -618,23 +618,29 @@ class WhisperBase(ABC):
         gc.collect()
 
     @staticmethod
-    def transform_text_to_list(inputdata: str,timestamps: bool, diarize: bool) -> list:
-        outputdata = []
-        temp_inputdata = inputdata.split("#NEWLINE#")
-        for temp_line in temp_inputdata:
-            temp_line_list = []
-            temp_line_items = temp_line.split("#TAB#")
-
-            # Add columns to match Dataframe
-            if len(temp_line_items)<3:
-                if timestamps==False and diarize==True: temp_line_items.insert(0,"")
-                if timestamps==True and diarize==False: temp_line_items.insert(1,"")
-            
-            for temp_line_item in temp_line_items:
-                temp_line_list.append(temp_line_item)	
-            outputdata.append(temp_line_list)
+def transform_text_to_list(inputdata: str) -> list:
+    outputdata = []
+    temp_inputdata = inputdata.split("#NEWLINE#")
+    for temp_line in temp_inputdata:
+        temp_line_items = temp_line.split("#TAB#")
+        col_time,col_speaker,col_text = "","",""
+        
+        if len(temp_line_items)==1:
+            col_text = temp_line_items[0]
+        if len(temp_line_items)==2:
+            if (temp_line_items[0])[0].isdigit():
+                col_time = temp_line_items[0]
+            else:
+                col_speaker = temp_line_items[0]
+            col_text = temp_line_items[1]
+        if len(temp_line_items)==3:
+            col_time = temp_line_items[0]
+            col_speaker = temp_line_items[1]
+            col_text = temp_line_items[2]
         
-        return outputdata 
+        outputdata.append([col_time,col_speaker,col_text])
+
+    return outputdata
     
     @staticmethod
     def format_time(elapsed_time: float) -> str: