From e2dc3bad20731fb3f035531dbd3e257dab245948 Mon Sep 17 00:00:00 2001 From: Andrew Gundersen Date: Mon, 1 Feb 2021 15:42:54 -0600 Subject: [PATCH] ws loop --- ws.py | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) diff --git a/ws.py b/ws.py index 57751f3..7dd8027 100644 --- a/ws.py +++ b/ws.py @@ -19,7 +19,9 @@ config = speech.RecognitionConfig( ) streaming_config = speech.StreamingRecognitionConfig(config=config) -def test(stream): + +def transcribe(stream): + requests = ( speech.StreamingRecognizeRequest(audio_content=chunk) for chunk in stream ) @@ -44,14 +46,22 @@ def test(stream): print(u"Transcript: {}".format(alternative.transcript)) -stream = [] -async def hello(websocket, path): - chunk = await websocket.recv() - print(chunk) +async def session(websocket, path): + buf = [] + + while True: + chunk = await websocket.recv() + + if chunk == "stop": + break + + buf.append(chunk) + + transcribe(buf) -start_server = websockets.serve(hello, "localhost", 8080) +start_server = websockets.serve(session, "localhost", 8080) asyncio.get_event_loop().run_until_complete(start_server) asyncio.get_event_loop().run_forever() -- 2.43.0