WS: accept both binary audio chunks and text control messages (FINISH/RESET)

app.py

@@ -89,21 +89,37 @@ async def websocket_audio(websocket: WebSocket):
         server_wrapper.reset()
         
         while True:
-            data = await websocket.receive_bytes()
-            
-            if data == b"FINISH":
-                # Finish transcription
-                result = await asyncio.get_event_loop().run_in_executor(None, server_wrapper.finish)
-                await websocket.send_json({"type": "finish", **result})
-                break
-            elif data == b"RESET":
-                server_wrapper.reset()
-                await websocket.send_json({"type": "reset", "status": "ok"})
+            # Accept either binary frames (audio) or text frames (control messages like FINISH)
+            message = await websocket.receive()
+            data = None
+            is_text = False
+            if 'bytes' in message and message['bytes'] is not None:
+                data = message['bytes']
+            elif 'text' in message and message['text'] is not None:
+                data = message['text']
+                is_text = True
+
+            if is_text:
+                # Control messages
+                if data == "FINISH":
+                    result = await asyncio.get_event_loop().run_in_executor(None, server_wrapper.finish)
+                    await websocket.send_json({"type": "finish", **(result or {})})
+                    break
+                elif data == "RESET":
+                    server_wrapper.reset()
+                    await websocket.send_json({"type": "reset", "status": "ok"})
+                else:
+                    # Unknown text message - ignore or log
+                    logger.debug(f"Unknown WS text message: {data}")
             else:
-                # Process audio chunk
-                result = await asyncio.get_event_loop().run_in_executor(None, server_wrapper.process_chunk_from_bytes, data)
-                if result and result.get("text"):
-                    await websocket.send_json({"type": "update", "text": result["text"]})
+                # Binary audio chunk
+                try:
+                    result = await asyncio.get_event_loop().run_in_executor(None, server_wrapper.process_chunk_from_bytes, data)
+                    if result and result.get("text"):
+                        await websocket.send_json({"type": "update", "text": result["text"]})
+                except Exception as e:
+                    logger.error(f"Error processing audio chunk via websocket: {e}")
+                    await websocket.send_json({"type": "error", "message": str(e)})
     
     except Exception as e:
         logger.error(f"WebSocket error: {e}")