diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py index 9ddf2e12006a64e56958ab6050d013e2aab73c81..b881d8258a59deb7cc26e874a3e199103c42db1e 100644 --- a/TTS/melo_handler.py +++ b/TTS/melo_handler.py @@ -53,11 +53,14 @@ class MeloTTSHandler(BaseHandler): _ = self.model.tts_to_file("text", self.speaker_id, quiet=True) def process(self, llm_sentence): - console.print(f"[green]ASSISTANT: {llm_sentence}") language_id = None + if isinstance(llm_sentence, tuple): + print("llm sentence is tuple!") llm_sentence, language_id = llm_sentence + console.print(f"[green]ASSISTANT: {llm_sentence}") + if language_id is not None and self.language != language_id: self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[self.language], device=self.device) self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[self.language]]