update app.py

app.py

@@ -5,39 +5,86 @@ import os
 
 TOKEN = os.environ.get("HF_TOKEN")
 
-# MODEL repo (model type)
-model_repo = "test1978/breakthrough-model"
-model_path = hf_hub_download(model_repo, "breakthrough_mcvs.py", repo_type="model", token=TOKEN)
-with open(model_path, 'r', encoding='utf-8-sig') as f:
-    exec(f.read())
-
-# DB dataset (dataset type!)
-db_repo = "test1978/breakthrough-data"
-db_path = hf_hub_download(db_repo, "breakthrough_zone_db.npz", repo_type="dataset", token=TOKEN)
+model_repo = "typical-cyber/breakthrough-model"
+model_path = hf_hub_download(
+    model_repo,
+    "breakthrough_mcvs.py",
+    repo_type="model",
+    token=TOKEN
+)
+
+ns = {}
+with open(model_path, "r", encoding="utf-8-sig") as f:
+    exec(f.read(), ns)
+
+db_repo = "typical-cyber/breakthrough-data"
+db_path = hf_hub_download(
+    db_repo,
+    "breakthrough_zone_db.npz",
+    repo_type="dataset",
+    token=TOKEN
+)
 zonedb_data = np.load(db_path, allow_pickle=True)
 
-# Init YOUR zone DB
-zonedb = HilbertOrderedZoneDatabase()
-zonedb.winningmatrices = list(zonedb_data.get('winning', []))
-zonedb.losingmatrices = list(zonedb_data.get('losing', []))
-zonedb.drawmatrices = list(zonedb_data.get('draw', []))
+HilbertOrderedZoneDatabase = ns["HilbertOrderedZoneDatabase"]
+Breakthrough = ns["Breakthrough"]
+MCVSSearcher = ns["MCVSSearcher"]
 
-def get_move(fen, player):
-    board = Breakthrough.Board(fen)
+zonedb = HilbertOrderedZoneDatabase("breakthroughzonedb.npz", maxsize=10000)
+zonedb.winningmatrices = list(zonedb_data.get("winning", []))
+zonedb.losingmatrices = list(zonedb_data.get("losing", []))
+zonedb.drawmatrices = list(zonedb_data.get("draw", []))
+
+def parse_board(board_text):
+    rows = [line.strip() for line in board_text.strip().splitlines() if line.strip()]
+    if len(rows) != 8:
+        raise ValueError("Board must have exactly 8 rows.")
+    board = np.zeros((8, 8), dtype=np.int32)
+    mapping = {".": 0, "1": 1, "2": 2}
+    for r, line in enumerate(rows):
+        parts = line.split()
+        if len(parts) != 8:
+            raise ValueError("Each row must have 8 space-separated cells.")
+        for c, cell in enumerate(parts):
+            if cell not in mapping:
+                raise ValueError("Use '.', '1', or '2' only.")
+            board[r, c] = mapping[cell]
+    return board
+
+def get_move(board_text, player):
     game = Breakthrough()
-    game.board = board
+    game.board = parse_board(board_text)
+    game.movecount = 0
+    game.cachedmatrix = None
+
     searcher = MCVSSearcher(None, None, zonedb, lambda_zone=1.0, k_zone=5)
     visits, _ = searcher.search_with_time_budget(game, 1.0)
+    if not visits:
+        return "No legal move"
     best_move = max(visits, key=visits.get)
-    return best_move.uci()
+    return str(best_move)
 
 demo = gr.Interface(
     fn=get_move,
-    inputs=gr.Textbox(
-        label="FEN", 
-        value="rnbqkbnr/pppppppp/8/8/4P3/8/PPPP1PPP/RNBQKBNR b KQkq e3 0 1"
-    ),
-    outputs=gr.Textbox(label="UCI Move"),
-    title="🎯 ChessMCVS Secure (Model + Dataset)"
+    inputs=[
+        gr.Textbox(
+            label="Board",
+            lines=8,
+            value=(
+                "1 1 1 1 1 1 1 1\n"
+                "1 1 1 1 1 1 1 1\n"
+                ". . . . . . . .\n"
+                ". . . . . . . .\n"
+                ". . . . . . . .\n"
+                ". . . . . . . .\n"
+                "2 2 2 2 2 2 2 2\n"
+                "2 2 2 2 2 2 2 2"
+            )
+        ),
+        gr.Dropdown(choices=["1", "2"], value="1", label="Player")
+    ],
+    outputs=gr.Textbox(label="Best Move"),
+    title="Breakthrough MCVS Secure"
 )
+
 demo.launch()