Initial commit

.gitattributes

@@ -0,0 +1,36 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+bin/stockfish filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: ThinkSquare
+emoji: 🏆
+colorFrom: blue
+colorTo: green
+sdk: gradio
+sdk_version: 5.32.1
+app_file: app.py
+pinned: false
+license: apache-2.0
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,79 @@
+from typing import Optional, Tuple
+import gradio as gr
+
+from ThinkSquare.src.thinksqure_engine import ThinkSquareEngine
+
+
+def play_chess(
+    move: Optional[str] = "", fen: Optional[str] = "", draw_ascii: bool = True
+) -> Tuple:
+    """Play a move in a chess game.
+    Prerequisites:
+        - User must be asked if they want to play as white or black. If user chooses black, pass an empty string in the first move for the engine to play as white.
+        - User must be asked if they want a board drawn in ASCII format.
+        - If a move is provided, it must be in long algebraic notation (e.g., "e4", "Nf3", "Bb5").
+
+    To start a new game:
+        - Pass empty string for fen.
+        - Pass empty string for engine to play as white.
+        - Pass a move in long algebraic notation (e.g., "e4", "Nf3", "Bb5") for engine to play as black.
+
+    To coninue a game:
+        - Pass the FEN string representing the board state prior to the user's last move.
+        - Pass a move in long algebraic notation (e.g., "e4", "Nf3", "Bb5") for engine to play the next move.
+
+    Args:
+        move: The move to play in long algebraic notation. If None, the engine will play a move.
+        fen: The FEN string representing the board state prior to the user's last move. If None, the game starts from the initial position.
+        draw_ascii: Whether to draw the board in ASCII format. Defaults to True.
+
+    Returns:
+        The best move played by the engine, the updated board state in FEN notation, and  a string representation of the board if draw_ascii is True else None.
+    """
+
+    if move is None or move == "" or move.lower() == "none" or move.lower() == "null":
+        move = None
+    if fen is None or fen == "" or fen.lower() == "none" or fen.lower() == "null":
+        fen = None
+
+    if move is not None:
+        is_valid = ThinkSquareEngine.is_valid_move(move, fen)
+
+        if not is_valid:
+            return "Invalid move", "", ""
+
+        fen = ThinkSquareEngine.get_fen_after_move(move, fen)
+
+        assert fen is not None, "FEN after move should not be None"
+
+    bestmove_san = ThinkSquareEngine.get_best_move(fen)
+    fen_after_move = ThinkSquareEngine.get_fen_after_move(bestmove_san, fen)
+
+    board_str = (
+        ThinkSquareEngine.get_ascii_board(fen_after_move) if draw_ascii else None
+    )
+
+    return bestmove_san, fen_after_move, board_str
+
+
+demo = gr.Interface(
+    fn=play_chess,
+    inputs=[
+        gr.Textbox(
+            label="Your Move (in standard algebraic notation - SAN) (Optional)",
+            placeholder="e.g. e4, Nf3, Bb5, etc. Optional - ",
+            value=None,
+        ),
+        gr.Textbox(
+            label="FEN String (optional)",
+            placeholder="Optional - Leave blank to start from initial position",
+            value=None,
+        ),
+        gr.Checkbox(label="Draw ASCII Board", value=True),
+    ],
+    outputs=("text", "text", "text"),
+    title="Play Chess with an engine",
+    description="This interface allows you to play chess against the Stockfish engine. You can provide a move in standard algebraic notation and an optional FEN string representing the board state. The engine will respond with its best move, the updated FEN string, and an ASCII representation of the board if requested.",
+)
+
+demo.launch(mcp_server=True)

bin/stockfish

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7fecbc0b26454b62be5e3b237b58dc5666401b56e520aeb1b0bf8f53fa8f2ef3
+size 78768416

requirements.txt

@@ -0,0 +1 @@
+chess==1.11.2

src/thinksqure_engine.py

@@ -0,0 +1,67 @@
+from typing import Optional
+import chess
+import chess.engine
+
+
+class ThinkSquareEngine:
+    _ENGINE = "ThinkSquare/bin/stockfish"
+
+    @staticmethod
+    def get_best_move(fen: Optional[str] = None, time_limit=0.1):
+        if fen is None:
+            fen = chess.STARTING_FEN
+
+        board = chess.Board(fen)
+
+        with chess.engine.SimpleEngine.popen_uci(ThinkSquareEngine._ENGINE) as engine:
+            result = engine.play(board, chess.engine.Limit(time=time_limit))
+            best_move = result.move
+            bestmove_san = board.san(best_move)
+
+        return bestmove_san
+
+    @staticmethod
+    def is_valid_move(
+        move_san: str,
+        fen: Optional[str] = None,
+    ) -> bool:
+        if fen is None:
+            fen = chess.STARTING_FEN
+
+        board = chess.Board(fen)
+
+        try:
+            move = board.parse_san(move_san)
+            return board.is_legal(move)
+        except ValueError:
+            return False
+
+    @staticmethod
+    def get_fen_after_move(
+        move_san: str,
+        fen: Optional[str] = None,
+    ) -> Optional[str]:
+        if fen is None:
+            fen = chess.STARTING_FEN
+
+        board = chess.Board(fen)
+
+        try:
+            move = board.parse_san(move_san)
+            if board.is_legal(move):
+                board.push(move)
+                return board.fen()
+            else:
+                return None
+        except ValueError:
+            return None
+
+    @staticmethod
+    def get_ascii_board(fen: Optional[str] = None) -> str:
+        if fen is None:
+            fen = chess.STARTING_FEN
+
+        board = chess.Board(fen)
+        ascii_representation = str(board)
+
+        return ascii_representation