Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -266,27 +266,27 @@ class WhisperBase(ABC):
                 # Translate to English using Whisper built-in functionality
                 transcription_note = ""
                 if params.is_translate:
-                    transcription_note = "To English (using Whisper)"
+                    transcription_note = "To English\n\t\t\t(using Whisper)"
                     source_lang = "English"
                 
                 # Translate the transcribed segments
                 translation_note = ""
                 if translate_output:
-                    self.nllb_inf = NLLBInference()
-                    if source_lang in NLLB_AVAILABLE_LANGS.keys():
-                        transcribed_segments = self.nllb_inf.translate_text(
-                            input_list_dict=transcribed_segments,
-                            model_size=translate_model,
-                            src_lang=source_lang,
-                            tgt_lang=target_lang,
-                            speaker_diarization=params.is_diarize
-                        )
-                        if source_lang != target_lang:
+                    if source_lang != target_lang:
+                        self.nllb_inf = NLLBInference()
+                        if source_lang in NLLB_AVAILABLE_LANGS.keys():
+                            transcribed_segments = self.nllb_inf.translate_text(
+                                input_list_dict=transcribed_segments,
+                                model_size=translate_model,
+                                src_lang=source_lang,
+                                tgt_lang=target_lang,
+                                speaker_diarization=params.is_diarize
+                            )
                             translation_note = source_lang + " to " + target_lang + " (using Facebook NLLB)"
                         else:
-                            translation_note = "None (Already in " + target_lang + ")"
+                            translation_note = source_lang + " not supported (using Facebook NLLB)"
                     else:
-                        translation_note = source_lang + " not supported"
+                        translation_note = "None (Already in " + target_lang + ")"
                 
                 ## Get preview as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))