Update modules/whisper/whisper_base.py

modules/whisper/whisper_base.py

@@ -19,6 +19,8 @@ from modules.utils.files_manager import get_media_files, format_gradio_files, lo
 from modules.whisper.whisper_parameter import *
 from modules.diarize.diarizer import Diarizer
 from modules.vad.silero_vad import SileroVAD
+from modules.translation.translation_base import *
+
 
 
 class WhisperBase(ABC):
@@ -258,6 +260,16 @@ 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 the trandscribed segments
+                translate_input = ""
+            	for item in transcribed_segments:
+            		translate_input += item['text'] + "\n"	
+
+        		translate_input = translate_input.rstrip()
+                translated_text = self.translate(translate_input)
+                print(translated_text)
+                
                 ## Get preview as txt
                 file_name, file_ext = os.path.splitext(os.path.basename(file))
                 subtitle = get_txt(transcribed_segments)