Update main.py

main.py

@@ -12,6 +12,7 @@ os.environ['LD_LIBRARY_PATH'] = '/engines/maia/lib:' + os.environ.get('LD_LIBRAR
 user_playing = {
     'total': 0,
     'stockfish': 0,
+    'trollfish': 0,
     'rodent': 0,
     'patricia': 0,
     'maia': 0
@@ -205,6 +206,36 @@ async def websocket_endpoint(websocket: WebSocket, elo: str):
         user_playing['patricia'] -= 1
 
 
+@app.websocket("/trollfish}")
+async def websocket_endpoint(websocket: WebSocket, elo: str):
+    await websocket.accept()
+
+    user_playing['total'] += 1
+    user_playing['trollfish'] += 1
+
+    stockfish = EngineChess([f"./engines/trollfish/trollfish"])
+    
+    async def read_from_socket(websocket: WebSocket):
+        async for data in websocket.iter_text():
+            print(f"Trollfish 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['trollfish'] -= 1
+
+
 @app.get("/")
 async def root(request: Request):
     html_content = """
@@ -272,6 +303,10 @@ async def root(request: Request):
                 <label>Stockfish Users:</label>
                 <span id="stockfish-users">""" + str(user_playing['stockfish']) + """</span>
             </div>
+            <div class="stat">
+                <label>Trollfish Users:</label>
+                <span id="trollfish-users">""" + str(user_playing['trollfish']) + """</span>
+            </div>
             <div class="stat">
                 <label>Rodent-III Users:</label>
                 <span id="rodent3-users">""" + str(user_playing['rodent']) + """</span>