cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

wait to consume server responses

Browse Source
This commit is contained in:
makaveli10
2023-08-24 10:07:44 +05:30
parent 155aaca7a7
commit 82a1aa449d
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/client.py
+6
View File
@@ -11,6 +11,7 @@ import textwrap
import json import json
import websocket import websocket
import uuid import uuid
import time
def resample(file: str, sr: int = 16000): def resample(file: str, sr: int = 16000):
@@ -54,6 +55,8 @@ class Client:
task = "transcribe" task = "transcribe"
uid = str(uuid.uuid4()) uid = str(uuid.uuid4())
WAITING = False WAITING = False
LAST_RESPONSE_RECIEVED = None
DISCONNECT_IF_NO_RESPONSE_FOR = 15
def __init__(self, host=None, port=None, is_multilingual=False, lang=None, translate=False): def __init__(self, host=None, port=None, is_multilingual=False, lang=None, translate=False):
Client.multilingual = is_multilingual Client.multilingual = is_multilingual
@@ -92,6 +95,7 @@ class Client:
@staticmethod @staticmethod
def on_message(ws, message): def on_message(ws, message):
Client.LAST_RESPONSE_RECIEVED = time.time()
message = json.loads(message) message = json.loads(message)
if message.get('uid')!=Client.uid: if message.get('uid')!=Client.uid:
print("[ERROR]: invalid client uid") print("[ERROR]: invalid client uid")
@@ -203,6 +207,8 @@ class Client:
print("[INFO]: Keyboard interrupt.") print("[INFO]: Keyboard interrupt.")
def close_websocket(self): def close_websocket(self):
while time.time() - self.LAST_RESPONSE_RECIEVED < self.DISCONNECT_IF_NO_RESPONSE_FOR:
continue
try: try:
self.client_socket.close() # Close the WebSocket connection self.client_socket.close() # Close the WebSocket connection
except Exception as e: except Exception as e: