update app.py

app.py

@@ -3,37 +3,39 @@ import numpy as np
 from huggingface_hub import hf_hub_download
 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
+    token=TOKEN,
 )
-
-ns = {}
 with open(model_path, "r", encoding="utf-8-sig") as f:
-    exec(f.read(), ns)
+    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
+    token=TOKEN,
 )
 zonedb_data = np.load(db_path, allow_pickle=True)
 
-HilbertOrderedZoneDatabase = ns["HilbertOrderedZoneDatabase"]
-Breakthrough = ns["Breakthrough"]
-MCVSSearcher = ns["MCVSSearcher"]
 
-zonedb = HilbertOrderedZoneDatabase("breakthroughzonedb.npz", max_size=10000)
-zonedb.winningmatrices = list(zonedb_data.get("winning", []))
-zonedb.losingmatrices = list(zonedb_data.get("losing", []))
-zonedb.drawmatrices = list(zonedb_data.get("draw", []))
+# Init YOUR zone DB
+zonedb = HilbertOrderedZoneDatabase()
+zonedb.winning_matrices = list(zonedb_data.get("winning", []))
+zonedb.losing_matrices = list(zonedb_data.get("losing", []))
+zonedb.draw_matrices = list(zonedb_data.get("draw", []))
+
 
 def parse_board(board_text):
     rows = [line.strip() for line in board_text.strip().splitlines() if line.strip()]
@@ -51,39 +53,39 @@ def parse_board(board_text):
             board[r, c] = mapping[cell]
     return board
 
+
+def move_to_uci(move):
+    fr, fc, tr, tc = move
+    cols = "abcdefgh"
+    return cols[fc] + str(8 - fr) + cols[tc] + str(8 - tr)
+
+
 def get_move(board_text, player):
     game = Breakthrough()
     game.board = parse_board(board_text)
-    game.move_count = 0
-    game.cached_matrix = None
+    game.move_count = 0 if str(player) == "1" 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)
-    if not visits:
-        return "No legal move"
     best_move = max(visits, key=visits.get)
-    return str(best_move)
+    return move_to_uci(best_move)
 
 demo = gr.Interface(
     fn=get_move,
-    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"
-            )
+    inputs=gr.Textbox(
+        label="Breakthrough Board (1/2/.)",
+        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"
+    ),
+    outputs=gr.Textbox(label="UCI Move"),
+    title="🎯 BreakthroughMCVS Secure (Model + Dataset)",
 )
-
 demo.launch()