Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -413,13 +413,15 @@ class WhisperBase(ABC):
                 if file_count > 1:
                     total_info += f'\n' 
 
-                # Add filename as firstline
+                # Add filename & info as first line
                 add_tabs = 0
                 title_line = ""
                 if add_timestamp_preview: add_tabs += 1
                 if diarize_speakers: add_tabs += 1
                 title_line = (add_tabs*"\t")+f'Media file: {info["input_source_file"]}\n'
                 title_line += (add_tabs*"\t")+f'Language: {info["lang"]} (probability {info["lang_prob"]}%)\n'
+                if params.is_translate: title_line += f'Translation: {info["transcription"]} (Handled by OpenAI Whisper)\n'
+                if translate_output: title_line += f'Translation: {info["translation"]} (Handled by Facebook NLLB)\n'
                 
                 total_result += title_line+info["subtitle"]
                 total_time += info["time_for_task"]