Update modules/translation/translation_base.py

modules/translation/translation_base.py

@@ -181,13 +181,12 @@ class TranslationBase(ABC):
                 total_progress = len(input_list_dict)
                 for index, dic in enumerate(input_list_dict):
                     progress(index / total_progress, desc="Translating..")
-                    translated_text = self.translate(dic["text"], max_length=max_length)
-
+                    
                     # Add speaker ID to translated sentence when diarization is enabled
-                    if re.search(': ',dic['text']) != None:
-                        speaker_id = ((dic['text']).split(":", 1)[0]).strip()
-                        if speaker_id != "":
-                            translated_text = speaker_id + ":123 " + translated_text
+                    if speaker_diarization:
+                        translated_text = ((dic['text']).split(":", 1)[0]).strip() + ": " + self.translate(((dic['text']).split(":", 1)[1]).strip(), max_length=max_length)
+                    else:
+                        translated_text = self.translate(dic["text"], max_length=max_length)
                     
                     dic["text"] = translated_text