Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -226,16 +226,17 @@ class WhisperBase(ABC):
             if files and isinstance(files[0], gr.utils.NamedString):
                 files = [file.name for file in files]
 
+            ## Initialization variables & start time
+            files_info = {}
+            files_to_download = {}
+            time_start = datetime.now()
+            
             ## Load parameters related with whisper
             params = WhisperParameters.as_value(*whisper_params)
             
             ## Load model to detect language
             model = whisper.load_model("base")
             
-            files_info = {}
-            files_to_download = {}
-            time_start = datetime.now()
-
             for file in files:
     
                 ## Detect language
@@ -331,17 +332,16 @@ class WhisperBase(ABC):
             for file_name, info in files_info.items():
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
-                total_info += f'Input file: {info["input_source_file"]}\nLanguage: {info["lang"]} (Probability {info["lang_prob"]}%)\n'
+                total_info += f'Input file\t: {info["input_source_file"]}\nLanguage\t: {info["lang"]} (Probability {info["lang_prob"]}%)\n'
                 
                 if params.is_translate:
-                    total_info += f'Translation: {info["transcription"]}\n'
+                    total_info += f'Translation\t: {info["transcription"]}\n'
                 
                 if translate_output:
-                    total_info += f'Translation: {info["translation"]}\n'
+                    total_info += f'Translation\t: {info["translation"]}\n'
             
-            #total_info += f"\nTranscription duration: {self.format_time(total_time)}"
             time_end = datetime.now()
-            total_info += f"\nTranscription duration: {self.format_time((time_end-time_start).total_seconds())}"
+            total_info += f"\nProcessing time\t: {self.format_time((time_end-time_start).total_seconds())}"
             
             result_str = total_result.rstrip("\n")
             result_file_path = [info['path'] for info in files_to_download.values()]