From 3860c67bef133bb5f4b3a60599de76a7bb3aa5de Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Fri, 25 Aug 2023 00:05:34 +0530 Subject: [PATCH] check elapsed time before closing stream --- whisper_live/client.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/whisper_live/client.py b/whisper_live/client.py index 85c9f10..663f579 100644 --- a/whisper_live/client.py +++ b/whisper_live/client.py @@ -195,6 +195,9 @@ class Client: self.stream.write(data) self.wf.close() + elapsed_time = time.time() - self.LAST_RESPONSE_RECIEVED + while elapsed_time < self.DISCONNECT_IF_NO_RESPONSE_FOR: + continue self.stream.close() self.close_websocket() @@ -207,8 +210,6 @@ class Client: print("[INFO]: Keyboard interrupt.") def close_websocket(self): - while time.time() - self.LAST_RESPONSE_RECIEVED < self.DISCONNECT_IF_NO_RESPONSE_FOR: - continue try: self.client_socket.close() # Close the WebSocket connection except Exception as e: