diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py index 0b1ed7ee306f87cb161922a0269ed271a7ba5f4e..e964a7c9f66fc33e69374b2246114a7642e62d73 100644 --- a/STT/whisper_stt_handler.py +++ b/STT/whisper_stt_handler.py @@ -8,6 +8,7 @@ import torch from baseHandler import BaseHandler from rich.console import Console import logging +from shared_variables import current_language logger = logging.getLogger(__name__) console = Console() diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py index 24e366d446990630585f090a672013c35fdffc22..06afcc635af4a209400cb039378ac401625db8de 100644 --- a/TTS/melo_handler.py +++ b/TTS/melo_handler.py @@ -5,6 +5,7 @@ import librosa import numpy as np from rich.console import Console import torch +from shared_variables import current_language logger = logging.getLogger(__name__) @@ -53,7 +54,6 @@ class MeloTTSHandler(BaseHandler): def process(self, llm_sentence): console.print(f"[green]ASSISTANT: {llm_sentence}") - global current_language if self.language != current_language: 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]] diff --git a/s2s_pipeline.py b/s2s_pipeline.py index d85ade8a3e5e5308f9d36104e58a973cb40ffa48..2438060982149f1ccf24073ddbaa72322ef950b1 100644 --- a/s2s_pipeline.py +++ b/s2s_pipeline.py @@ -48,7 +48,6 @@ os.environ["TORCHINDUCTOR_CACHE_DIR"] = os.path.join(CURRENT_DIR, "tmp") console = Console() logging.getLogger("numba").setLevel(logging.WARNING) # quiet down numba logs -current_language = "en" def prepare_args(args, prefix): diff --git a/shared_variables.py b/shared_variables.py new file mode 100644 index 0000000000000000000000000000000000000000..649f0f30f718c7210580681832f49643872caab5 --- /dev/null +++ b/shared_variables.py @@ -0,0 +1 @@ +current_language = "en"