update app.py

app.py

@@ -62,30 +62,48 @@ def move_to_uci(move):
     return cols[fc] + str(8 - fr) + cols[tc] + str(8 - tr)
 
 
-def get_move(board_text, player=None):
-    print(f"[DEBUG] get_move called: board_text={repr(board_text)}, player={player}")
-    lines = [line.strip() for line in board_text.strip().splitlines() if line.strip()]
-    if len(lines) != 8:
-        raise ValueError(f"get_move: board must have exactly 8 rows (got {len(lines)}):\n{board_text}")
-    board_text = "\n".join(lines)
+def fen_to_board(fen):
+    rows_str = fen.strip().split(" ")[0]  # strip off ' w' or ' b'
+    parts = rows_str.split("/")
+    if len(parts) != 8:
+        raise ValueError(f"FEN must have 8 rows, got {len(parts)}")
+    board = np.zeros((8, 8), dtype=np.int32)
+    for r, row in enumerate(parts):
+        c = 0
+        for ch in row:
+            if ch == "B":
+                board[r, c] = 2
+                c += 1
+            elif ch == "W":
+                board[r, c] = 1
+                c += 1
+            elif ch.isdigit():
+                skip = int(ch)
+                c += skip
+            else:
+                raise ValueError(f"Invalid FEN char {ch} in row {r} ({row})")
+        if c != 8:
+            raise ValueError(f"Row {r} has {c} columns, should be 8")
+    return board
 
+def get_move(board_text, player=None):
+    print(f"[DEBUG] get_move FEN board_text={repr(board_text)}, player={player}")
     game = Breakthrough()
-    game.board = parse_board(board_text)
+    game.board = fen_to_board(board_text)
     if player == "1":
         game.move_count = 0
     elif player == "2":
         game.move_count = 1
     else:
-        # Default: white to move
-        game.move_count = 0
+        # infer from the last part of FEN if you want, or keep fixed
+        game.move_count = 0 if player is None or "w" in board_text else 1
     game._cached_matrix = None
     searcher = MCVSSearcher(None, None, zonedb, lambda_zone=1.0, k_zone=5)
     visits, _ = searcher.search_with_time_budget(game, 1.0)
     best_move = max(visits, key=visits.get)
-    move_str = move_to_uci(best_move)
-    print(f"[DEBUG] get_move OK: board={board_text}\nplayer={player} -> {move_str}")
-    return move_str
-
+    move = move_to_uci(best_move)
+    print(f"[DEBUG] get_move OK: FEN={board_text} -> {move}")
+    return move
 
 demo = gr.Interface(
     fn=get_move,