diff --git a/whispering/serve.py b/whispering/serve.py index 1399a3c..1c4bdad 100644 --- a/whispering/serve.py +++ b/whispering/serve.py @@ -28,6 +28,19 @@ async def serve_with_websocket_main(websocket): if isinstance(message, str): logger.debug(f"Got str: {message}") + d = json.loads(message) + v = d.get("context") + if v is not None: + ctx = Context.parse_obj(v) + else: + await websocket.send( + json.dumps( + { + "error": "unsupported message", + } + ) + ) + return continue logger.debug(f"Message size: {len(message)}") diff --git a/whispering/websocket_client.py b/whispering/websocket_client.py index f59de38..1bcbe26 100644 --- a/whispering/websocket_client.py +++ b/whispering/websocket_client.py @@ -38,6 +38,9 @@ async def transcribe_from_mic_and_send( callback=sd_callback, ): async with websockets.connect(uri, max_size=999999999) as ws: # type:ignore + logger.debug("Sent context") + await ws.send(ctx.json()) + idx: int = 0 while True: