From 0741df8fdef23ad48b2b22658d2282e65c4cac4e Mon Sep 17 00:00:00 2001
From: jiegeaiai <jiegeaiai@163.com>
Date: Wed, 6 Nov 2024 07:43:03 +0800
Subject: [PATCH] modify stop function
---
asr/sherpa_ncnn_asr.py | 2 +-
human/audio_inference_handler.py | 2 ++
human/human_render.py | 2 +-
nlp/nlp_base.py | 2 ++
nlp/nlp_doubao.py | 6 ++++++
5 files changed, 12 insertions(+), 2 deletions(-)
diff --git a/asr/sherpa_ncnn_asr.py b/asr/sherpa_ncnn_asr.py
index eb77bf7..8abda87 100644
--- a/asr/sherpa_ncnn_asr.py
+++ b/asr/sherpa_ncnn_asr.py
@@ -65,7 +65,7 @@ class SherpaNcnnAsr(AsrBase):
self._notify_complete('介绍中国5000年历史文学')
logger.info(f'_recognize_loop111')
segment_id += 1
- time.sleep(50)
+ time.sleep(5)
logger.info(f'_recognize_loop222')
logger.info(f'_recognize_loop exit')
'''
diff --git a/human/audio_inference_handler.py b/human/audio_inference_handler.py
index 0a67f79..f867614 100644
--- a/human/audio_inference_handler.py
+++ b/human/audio_inference_handler.py
@@ -135,4 +135,6 @@ class AudioInferenceHandler(AudioHandler):
def pause_talk(self):
print('AudioInferenceHandler pause_talk', self._audio_queue.size(), self._mal_queue.size())
self._audio_queue.clear()
+ print('AudioInferenceHandler111')
self._mal_queue.clear()
+ print('AudioInferenceHandler222')
diff --git a/human/human_render.py b/human/human_render.py
index 8189e7b..f0afe8f 100644
--- a/human/human_render.py
+++ b/human/human_render.py
@@ -86,7 +86,7 @@ class HumanRender(AudioHandler):
# return self._voice_render.size()
def pause_talk(self):
- pass
+ logging.info('hunan pause_talk')
# self._voice_render.pause_talk()
# self._video_render.pause_talk()
diff --git a/nlp/nlp_base.py b/nlp/nlp_base.py
index c545211..13bb4e7 100644
--- a/nlp/nlp_base.py
+++ b/nlp/nlp_base.py
@@ -51,7 +51,9 @@ class NLPBase(AsrObserver):
logger.info('NLPBase stop')
self._is_running = False
self._ask_queue.add_task(self._on_close)
+ logger.info('NLPBase add close')
self._ask_queue.stop()
+ logger.info('NLPBase _ask_queue stop')
def pause_talk(self):
logger.info('NLPBase pause_talk')
diff --git a/nlp/nlp_doubao.py b/nlp/nlp_doubao.py
index d1b41da..794bdb5 100644
--- a/nlp/nlp_doubao.py
+++ b/nlp/nlp_doubao.py
@@ -66,3 +66,9 @@ class DouBao(NLPBase):
logger.info('AsyncArk close')
if self.__client is not None and not self.__client.is_closed():
await self.__client.close()
+
+ def stop(self):
+ print('doubao stop00')
+ self.__client.close()
+ print('doubao stop11')
+ super().stop()