Update main.py

main.py

@@ -13,6 +13,7 @@ user_playing = {
     'total': 0,
     'stockfish': 0,
     'rodent': 0,
+    'patricia': 0,
     'maia': 0
 }
 
@@ -170,6 +171,36 @@ async def websocket_endpoint(websocket: WebSocket, personality: str):
         user_playing['rodent'] -= 1
 
 
+@app.websocket("/patricia")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+
+    user_playing['total'] += 1
+    user_playing['patricia'] += 1
+
+    stockfish = EngineChess([f"./engines/Patricia/patricia"])
+    
+    async def read_from_socket(websocket: WebSocket):
+        async for data in websocket.iter_text():
+            print(f"Patricia Client: {data}")
+            stockfish.put(data)
+
+    asyncio.create_task(read_from_socket(websocket))
+
+    try:
+        while True:
+            while True:
+                res = stockfish.read_line()
+                if res:
+                    await websocket.send_text(f"{res}")
+                else:
+                    break
+            await asyncio.sleep(0.1)
+    finally:
+        user_playing['total'] -= 1
+        user_playing['patricia'] -= 1
+
+
 @app.get("/")
 async def root(request: Request):
     html_content = """
@@ -238,9 +269,13 @@ async def root(request: Request):
                 <span id="stockfish-users">""" + str(user_playing['stockfish']) + """</span>
             </div>
             <div class="stat">
-                <label>RodentIII Users:</label>
+                <label>Rodent-III Users:</label>
                 <span id="rodent3-users">""" + str(user_playing['rodent']) + """</span>
             </div>
+            <div class="stat">
+                <label>Patricia Users:</label>
+                <span id="patricia-users">""" + str(user_playing['patricia']) + """</span>
+            </div>
             <div class="stat">
                 <label>Maia Users:</label>
                 <span id="maia-users">""" + str(user_playing['maia']) + """</span>