Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -269,6 +269,12 @@ class WhisperBase(ABC):
                 #)
                 #files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
 
+                # Translate to English using Whisper built-in functionality
+                transcription_note = ""
+                if params.is_translate:
+                    transcription_note = "Translated to English"
+                    file_language = "English"
+                
                 # Translate the transcribed segments
                 translation_note = ""
                 if translate_output:
@@ -288,7 +294,7 @@ class WhisperBase(ABC):
                 ## 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), "translation": translation_note}
+                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, "transcription": transcription_note}
                 
                 ## Add output file as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
@@ -330,6 +336,10 @@ 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 params.is_translate:
+                    total_info += f'transcription: {info["transcription"]}\n'
+                
                 if translate_output:
                     total_info += f'Translation: {info["translation"]}\n'