Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -280,14 +280,14 @@ class WhisperBase(ABC):
                             tgt_lang=target_lang,
                             speaker_diarization=params.is_diarize
                         )
-                        print (file_language + " found") 
+                        translation_note = "from " + file_language + " to " + target_lang
                     else:
-                        print (file_language + " not supported") 
+                        translation_note = file_language + " not supported"
                 
                 ## Get preview as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle = get_txt(transcribed_segments)
-                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
+                files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext), "translation": translation_note}
                 
                 ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -329,6 +329,8 @@ class WhisperBase(ABC):
                 total_result += f'{info["subtitle"]}'
                 total_time += info["time_for_task"]
                 total_info += f'Input file: {info["input_source_file"]}\nLanguage: {info["lang"]} with probability {info["lang_prob"]}%\n'
+                if translate_output:
+                    total_info += f'Translation: {info["translation"]}\n'
             
             #total_info += f"\nTranscription duration: {self.format_time(total_time)}"
             time_end = datetime.now()