From 1e037854fd55ffb40d8eda461d74978c9c506ffa Mon Sep 17 00:00:00 2001
From: IrvingGao <1729854488@qq.com>
Date: Tue, 18 Jun 2024 17:31:00 +0800
Subject: [PATCH] [client] fix bugs

---
 takway/clients/web_socket_client_utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/takway/clients/web_socket_client_utils.py b/takway/clients/web_socket_client_utils.py
index 1a36c2d..ec0916c 100644
--- a/takway/clients/web_socket_client_utils.py
+++ b/takway/clients/web_socket_client_utils.py
@@ -370,6 +370,9 @@ class WebSocketClinet:
                     client.close_client()
                     break
                 
+                if self.interrupt_event.is_set():
+                    break
+                
                 response, data_type = client.receive_per_data()
                 if data_type == dict:
                     print(f"{datetime.now()}: receive json data: {response}")  # 打印接收到的消息
@@ -408,6 +411,7 @@ class WebSocketClinet:
                     # print(f"listening_event: {self.listening_event.is_set()}, speaking_event: {self.speaking_event.is_set()}")
                     break  # 如果没有接收到消息，则退出循环
             client.close_client()
+            self.interrupt_event.clear()
     
     
     def audio_play_process(self):
@@ -462,7 +466,6 @@ class WebSocketClinet:
                             time.sleep(0.5)
                             self.listening_event.set()
                             self.speaking_event.clear()
-                            self.interrupt_event.clear()
                             break
                         # if self.listening_event.is_set() and not self.speaking_event.is_set():
                         #     print(f"{datetime.now()}: speaker interrupt.")