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

Merge pull request #43 from makaveli10/close_websocket

Browse Source 
close websocket after consuming file
This commit is contained in:
makaveli
2023-08-28 11:20:33 +05:30
committed by GitHub
parent 71caaffa45 3860c67bef
commit d319aa2fc9
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
whisper_live/client.py
+8
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")
@@ -191,7 +195,11 @@ class Client:
self.stream.write(data) self.stream.write(data)
self.wf.close() 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.stream.close()
self.close_websocket()
except KeyboardInterrupt: except KeyboardInterrupt:
self.wf.close() self.wf.close()