Update main.py

main.py

@@ -80,11 +80,11 @@ app.add_middleware(
 async def websocket_endpoint(websocket: WebSocket, version: str):
     await websocket.accept()
     
-    stockfish = EngineChess(f"engines/stockfish-{version}-uci")
+    stockfish = EngineChess(f"engines/stockfish/stockfish-{version}-uci")
 
     async def read_from_socket(websocket: WebSocket):
         async for data in websocket.iter_text():
-            print(f"Client: {data}")
+            print(f"Stockfish Client: {data}")
             stockfish.put(data)
                     
     asyncio.create_task(read_from_socket(websocket))
@@ -92,8 +92,30 @@ async def websocket_endpoint(websocket: WebSocket, version: str):
     while True:
         while True:
             res = stockfish.read_line()
-            if res != "":
+            if res:
                 await websocket.send_text(f"{res}")
             else:
                 break
-        await asyncio.sleep(0.1)
+        await asyncio.sleep(0.1)
+
+@app.websocket("/maia-{elo}")
+async def websocket_endpoint(websocket: WebSocket, version: str):
+    await websocket.accept()
+    
+    stockfish = EngineChess(["./engines/maia/lc0", f"--weights=engines/maia-{elo}.pb.gz", "--backend=trivial"])
+
+    async def read_from_socket(websocket: WebSocket):
+        async for data in websocket.iter_text():
+            print(f"Maia Client: {data}")
+            stockfish.put(data)
+                    
+    asyncio.create_task(read_from_socket(websocket))
+
+    while True:
+        while True:
+            res = stockfish.read_line()
+            if res:
+                await websocket.send_text(f"{res}")
+            else:
+                break
+        await asyncio.sleep(0.1)