diff --git a/handle_client.py b/handle_client.py new file mode 100644 index 0000000..2d4e114 --- /dev/null +++ b/handle_client.py @@ -0,0 +1,61 @@ +import asyncio +import websockets +import cv2 +import numpy as np +from loguru import logger + + +async def handle_client(websocket): + client = websocket.remote_address + logger.info(f"Client connected: {client}") + + frame_count = 0 + + try: + async for message in websocket: + frame_count += 1 + + data = np.frombuffer(message, dtype=np.uint8) + frame = cv2.imdecode(data, cv2.IMREAD_COLOR) + + if frame is None: + logger.warning(f"Decode frame failed from client={client}") + continue + + if frame_count % 100 == 0: + logger.info(f"Received frames={frame_count}, client={client}, size={len(message)} bytes") + + cv2.imshow("Android Camera", frame) + + if cv2.waitKey(1) & 0xFF == 27: + logger.info("ESC pressed, closing display") + break + + except websockets.ConnectionClosed: + logger.info(f"Client disconnected: {client}") + + except Exception as e: + logger.exception(f"WebSocket error, client={client}, error={e}") + + finally: + logger.info(f"Connection closed: client={client}, total_frames={frame_count}") + cv2.destroyAllWindows() + + +async def main(): + host = "0.0.0.0" + port = 8765 + + logger.info(f"WebSocket server started: ws://{host}:{port}") + + async with websockets.serve( + handle_client, + host, + port, + max_size=10 * 1024 * 1024 + ): + await asyncio.Future() + + +if __name__ == "__main__": + asyncio.run(main()) \ No newline at end of file