Upload folder using huggingface_hub

app.py

@@ -120,7 +120,7 @@ async def make_move(req: MoveRequest):
                 return JSONResponse({'ai_move': None, 'eval': 0.0})
                 
             # MCTS runs 150 simulations
-            result = await rust_engine_pool.play(board, chess.engine.Limit(time=req.time_limit), info=chess.engine.INFO_SCORE)
+            result = await rust_engine_pool.play(board, chess.engine.Limit(time=req.time_limit), info=chess.engine.INFO_ALL)
             ai_move = result.move.uci()
             
             if "score" in result.info:
@@ -129,14 +129,20 @@ async def make_move(req: MoveRequest):
                     ai_eval = 1.0 if score.mate() > 0 else -1.0
                 else:
                     ai_eval = max(-1.0, min(1.0, score.score() / 1000.0))
+            
+            engine_info = {}
+            if "depth" in result.info: engine_info["depth"] = result.info["depth"]
+            if "nodes" in result.info: engine_info["nodes"] = result.info["nodes"]
+            if "nps" in result.info: engine_info["nps"] = result.info["nps"]
                     
     except Exception as e:
         import traceback
         traceback.print_exc()
         print(f"Engine error: {repr(e)}")
         ai_move = None
+        engine_info = {}
             
-    return JSONResponse({'ai_move': ai_move, 'eval': ai_eval})
+    return JSONResponse({'ai_move': ai_move, 'eval': ai_eval, 'engine_info': engine_info})
 
 if __name__ == '__main__':
     import uvicorn

lichess-bot/remote_engine.py

@@ -80,8 +80,18 @@ def main():
                     response = requests.post(API_URL, json=payload, timeout=30)
                     data = response.json()
                     
-                    if "ai_move" in data:
+                    if "ai_move" in data and data["ai_move"]:
                         move = data["ai_move"]
+                        
+                        if "engine_info" in data and data["engine_info"]:
+                            info = data["engine_info"]
+                            info_str = "info"
+                            if "depth" in info: info_str += f" depth {info['depth']}"
+                            if "nodes" in info: info_str += f" nodes {info['nodes']}"
+                            if "nps" in info: info_str += f" nps {info['nps']}"
+                            if info_str != "info":
+                                print(info_str)
+                                
                         print(f"bestmove {move}")
                     else:
                         print("bestmove 0000")