Add PGN util

src/util/pgn_util.py

@@ -0,0 +1,39 @@
+import chess.pgn
+import tempfile
+
+
+# Load game
+def read_pgn(pgn_filepath):
+    with open(pgn_filepath) as f:
+        game = chess.pgn.read_game(f)
+    return game
+
+
+def export_pgn(game):
+    # Create a temp file with .pgn extension that won't be auto-deleted
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".pgn", mode="w") as tmp:
+        exporter = chess.pgn.FileExporter(tmp)
+        game.accept(exporter)
+        return tmp.name  # Return the file path so Gradio can serve it
+
+
+def format_pv(pv_moves, board):
+    temp_board = board.copy()
+    san_moves = []
+    for move in pv_moves:
+        san_moves.append(temp_board.san(move))
+        temp_board.push(move)
+    return " ".join(san_moves)
+
+
+def add_variation(node, variation):
+    board = node.board()
+    current = node
+
+    for move in variation:
+        # Ensure move is legal in current position
+        if move in board.legal_moves:
+            current = current.add_variation(move)
+            board.push(move)
+        else:
+            break  # Stop if PV deviates from legal moves