Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -428,33 +428,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())}"
 
-            # Format Dataframe data    
-            new_headers = []
-            new_column_widths = []
-            new_datatype = []
-
-            if add_timestamp_preview==True and diarize_speakers==True:
-                new_headers = ["Time","Speaker","Text"]
-                new_column_widths = ["10%","15%","70%"]
-                new_datatype = ["str","str","markdown"]
-            elif add_timestamp_preview==True and diarize_speakers==False:
-                new_headers = ["Time","Text"]
-                new_column_widths = ["10%","85%"]
-                new_datatype = ["str","markdown"]
-            elif add_timestamp_preview==False and diarize_speakers==True:
-                new_headers = ["Speaker","Text"]
-                new_column_widths = ["15%","80%"]
-                new_datatype = ["str","markdown"]
-            elif add_timestamp_preview==False and diarize_speakers==False:
-                new_headers = ["Text"]
-                new_column_widths = ["95%"]
-                new_datatype = ["markdown"]
-
-            new_data_rows = self.transform_text_to_list(total_result)
-            #updated_data = {"headers": new_headers, "data": new_data_rows}
-            updated_data = {"headers": new_headers, "data": new_data_rows, "column_widths": new_column_widths, "datatype": new_datatype}
-            
-            return [(gr.update(value=updated_data)),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: