From 362307f9f7436ce82bbd73242d055304692874c3 Mon Sep 17 00:00:00 2001 From: jiegeaiai Date: Sun, 20 Oct 2024 03:28:49 +0800 Subject: [PATCH] modify pause talk --- human/audio_inference_handler.py | 4 ++++ human/audio_mal_handler.py | 3 +++ test/asr_nlp_tts.py | 2 +- 3 files changed, 8 insertions(+), 1 deletion(-) diff --git a/human/audio_inference_handler.py b/human/audio_inference_handler.py index 8b74b25..dd37dc8 100644 --- a/human/audio_inference_handler.py +++ b/human/audio_inference_handler.py @@ -121,3 +121,7 @@ class AudioInferenceHandler(AudioHandler): def stop(self): self._exit_event.clear() self._run_thread.join() + + def pause_talk(self): + self._audio_queue.queue.clear() + self._mal_queue.queue.clear() diff --git a/human/audio_mal_handler.py b/human/audio_mal_handler.py index a9dc50d..008160f 100644 --- a/human/audio_mal_handler.py +++ b/human/audio_mal_handler.py @@ -89,3 +89,6 @@ class AudioMalHandler(AudioHandler): if self._thread.is_alive(): self._thread.join() logging.info('chunk2mal stop') + + def pause_talk(self): + self._queue.queue.clear() diff --git a/test/asr_nlp_tts.py b/test/asr_nlp_tts.py index 528ee03..f5b008b 100644 --- a/test/asr_nlp_tts.py +++ b/test/asr_nlp_tts.py @@ -21,7 +21,7 @@ except ImportError as e: def main(): print("Started! Please speak") - handle = TTSAudioSaveHandle() + handle = TTSAudioSaveHandle(None, None) tts = TTSEdge(handle) split = PunctuationSplit() nlp = DouBao(split, tts)