| |
| """Interactive Vela-Chess. Run this in your terminal.""" |
| import chess, sys |
| from pathlib import Path |
| sys.path.insert(0, str(Path(__file__).parent)) |
| from vela_chess_engine_v2 import VelaChessV2 |
|
|
| UPPER = {'P':'P','N':'N','B':'B','R':'R','Q':'Q','K':'K'} |
| LOWER = {'p':'p','n':'n','b':'b','r':'r','q':'q','k':'k'} |
|
|
| def show(board): |
| print() |
| print(" a b c d e f g h") |
| print(" +---------------+") |
| for r in range(7, -1, -1): |
| row = "%d |" % (r + 1) |
| for f in range(8): |
| p = board.piece_at(chess.square(f, r)) |
| if p: |
| sym = UPPER.get(p.symbol()) if p.color == chess.WHITE else LOWER.get(p.symbol()) |
| row += " " + sym |
| else: |
| row += " ." |
| row += " |" |
| print(row) |
| print(" +---------------+") |
| print(" a b c d e f g h") |
| print() |
|
|
| engine = VelaChessV2(str(Path(__file__).parent.parent / "checkpoints" / "vela_chess_v2" / "final.pt")) |
| board = chess.Board() |
|
|
| print("=== VELA-CHESS ===") |
| print("You = WHITE (uppercase)") |
| print("Vela = BLACK (lowercase)") |
| print("Commands: UCI move (e2e4), 'moves', 'quit'") |
| print() |
|
|
| show(board) |
|
|
| while not board.is_game_over(): |
| if board.turn == chess.WHITE: |
| cmd = input("Your move: ").strip().lower() |
| if cmd == "quit": |
| break |
| if cmd == "moves": |
| print(", ".join(m.uci() for m in board.legal_moves)) |
| continue |
| try: |
| move = chess.Move.from_uci(cmd) |
| except: |
| print("Bad format. Use UCI: e2e4") |
| continue |
| if move not in board.legal_moves: |
| print("Illegal! Type 'moves' for legal moves") |
| continue |
| else: |
| print("Vela thinking...") |
| move = engine.choose_move(board) |
| cmd = move.uci() |
| print("Vela plays: %s" % cmd) |
|
|
| board.push(move) |
| show(board) |
|
|
| print("Game over: %s" % board.result()) |
|
|
