Upload ai/utils/record_game.py with huggingface_hub

ai/utils/record_game.py

@@ -0,0 +1,134 @@
+import argparse
+import json
+import os
+import random
+import sys
+
+# Add project root to path
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+import engine_rust
+from ai.benchmark_decks import parse_deck
+
+
+def get_action_name(action_id, game, db_json):
+    """Decodes Action ID into a human-readable name."""
+    if action_id == 0:
+        return "Pass/End Phase"
+
+    phase = game.phase
+    p_idx = game.current_player
+    p_state = game.get_player(p_idx)
+
+    if phase in [-1, 0]:  # Mulligan
+        indices = [i for i in range(len(p_state.hand)) if (action_id >> i) & 1]
+        return f"Mulligan: Discard indices {indices}"
+
+    if phase == 4:  # Main
+        if 1 <= action_id <= 180:
+            adj = action_id - 1
+            hand_idx = adj // 3
+            slot_idx = adj % 3
+            if hand_idx < len(p_state.hand):
+                cid = p_state.hand[hand_idx]
+                card = db_json["member_db"].get(str(cid), {"name": f"Unknown Member {cid}"})
+                return f"Play {card['name']} to Slot {slot_idx}"
+        elif 200 <= action_id < 400:
+            adj = action_id - 200
+            slot_idx = adj // 10
+            ab_idx = adj % 10
+            return f"Activate Ability {ab_idx} in Slot {slot_idx}"
+
+    if phase == 5:  # LiveSet
+        if 400 <= action_id < 500:
+            hand_idx = action_id - 400
+            if hand_idx < len(p_state.hand):
+                cid = p_state.hand[hand_idx]
+                card = db_json["live_db"].get(str(cid), {"name": f"Unknown Live {cid}"})
+                return f"Set Live: {card['name']}"
+
+    return f"Unknown Action ({action_id})"
+
+
+def record_game(output_file, sims=200):
+    with open("data/cards_compiled.json", "r", encoding="utf-8") as f:
+        db_content = f.read()
+    db_json = json.loads(db_content)
+    db = engine_rust.PyCardDatabase(db_content)
+    game = engine_rust.PyGameState(db)
+
+    muse_deck_path = "ai/decks/muse_cup.txt"
+    p_deck, p_lives, p_energy = parse_deck(
+        muse_deck_path, db_json["member_db"], db_json["live_db"], db_json.get("energy_db", {})
+    )
+
+    game.initialize_game(p_deck, p_deck, p_energy, p_energy, p_lives, p_lives)
+
+    with open(output_file, "w", encoding="utf-8") as f:
+        f.write(f"--- GAME RECORD: MCTS ({sims} sims) vs RANDOM ---\n")
+        f.write(f"Deck: {muse_deck_path}\n\n")
+
+        step = 0
+        while not game.is_terminal() and step < 1000:
+            cp = game.current_player
+            phase = game.phase
+            p_state = game.get_player(cp)
+
+            phase_names = {
+                -1: "MulliganP1",
+                0: "MulliganP2",
+                1: "Active",
+                2: "Energy",
+                3: "Draw",
+                4: "Main",
+                5: "LiveSet",
+                6: "PerfP1",
+                7: "PerfP2",
+                8: "LiveResult",
+            }
+            p_name = phase_names.get(phase, f"Phase {phase}")
+
+            f.write(f"[Step {step} | Turn {game.turn} | Player {cp} ({'MCTS' if cp == 0 else 'RANDOM'}) | {p_name}]\n")
+
+            is_interactive = phase in [-1, 0, 4, 5]
+
+            if is_interactive:
+                tapped = sum(1 for t in p_state.tapped_energy if t)
+                total_en = len(p_state.energy_zone)
+                f.write(f"  Energy: {total_en - tapped}/{total_en}\n")
+                f.write(f"  Hand: {p_state.hand}\n")
+                f.write(f"  Stage: {p_state.stage}\n")
+                f.write(f"  Live Zone: {p_state.live_zone} (Revealed: {p_state.live_zone_revealed})\n")
+                f.write(f"  Score: {p_state.score}\n")
+
+                if cp == 0:
+                    suggestions = game.get_mcts_suggestions(sims, engine_rust.SearchHorizon.TurnEnd)
+                    action = suggestions[0][0] if suggestions else 0
+                else:
+                    legal = game.get_legal_action_ids()
+                    action = random.choice(legal) if legal else 0
+
+                f.write(f"  Action: {action} ({get_action_name(action, game, db_json)})\n")
+                game.step(action)
+            else:
+                game.step(0)
+
+            f.write("\n")
+            step += 1
+
+        f.write("\n--- GAME COMPLETE ---\n")
+        p0 = game.get_player(0)
+        p1 = game.get_player(1)
+        f.write(f"Final Outcome: winner={game.get_winner()}\n")
+        f.write(f"P0 (MCTS): Score {p0.score}, Success Lives: {len(p0.success_lives)}\n")
+        f.write(f"P1 (RANDOM): Score {p1.score}, Success Lives: {len(p1.success_lives)}\n")
+
+    print(f"Game record saved to {output_file}")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--output", default="game_record.txt")
+    parser.add_argument("--sims", type=int, default=200)
+    args = parser.parse_args()
+    record_game(args.output, sims=args.sims)