From e960b7ad5d1c05821bb40d2bd281810a96b49e0c Mon Sep 17 00:00:00 2001 From: XpycTee Date: Tue, 14 May 2024 12:55:52 +0500 Subject: [PATCH] Refactor message processing in process_chunk function to handle JSON message objects and improve readability --- websocket/asr_server.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/websocket/asr_server.py b/websocket/asr_server.py index f2c6e99..eae894f 100755 --- a/websocket/asr_server.py +++ b/websocket/asr_server.py @@ -11,9 +11,11 @@ from vosk import Model, SpkModel, KaldiRecognizer def process_chunk(rec, message): - if message == '{"eof" : 1}': + message = json.loads(message) if isinstance(message, str) else message + + if message == {"eof": 1}: return rec.FinalResult(), True - if message == '{"reset" : 1}': + if message == {"reset": 1}: return rec.FinalResult(), False elif rec.AcceptWaveform(message): return rec.Result(), False