Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -411,7 +411,7 @@ class WhisperBase(ABC):
                 
                 # Add filename & info as first line
                 title_line = ""
-                tabbed_space = ("\t")*5
+                tabbed_space = ("\t")*3
                 if file_count_total > 1 and file_count > 1: title_line += "■"
                 if add_timestamp_preview: title_line += "□"
                 if diarize_speakers: title_line += "□"
@@ -430,9 +430,16 @@ class WhisperBase(ABC):
             
             time_end = datetime.now()
             #total_info += f"Total processing time:\t{self.format_time((time_end-time_start).total_seconds())}"
-            total_info = f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
 
-            print("data to process: "+total_result)
+            total_info = ""
+            if params.is_translate or translate_output:
+                total_info = "Translation: Enabled"
+            if diarize_speakers:
+                if total_info != "": total_info += "\n"
+                total_info += "Diarization: Enabled"
+            if total_info != "": total_info += "\n"
+            total_info += f"Processed {file_count} {'file' if file_count == 1 else 'files'} in {self.format_time((time_end-time_start).total_seconds())}"
+
             result_str = self.transform_text_to_list(total_result,"□","■") #Transform to gr.Dataframe format
             result_file_path = [info['path'] for info in files_to_download.values()]