chessXAI / svg_pieces.py
seredapj's picture
Upload 19 files
0c536cc verified
from python_chess_customized_svg import piece
import python_chess_customized_svg as svg
import chess
import base64
#def get_svg_piece(symbol):
# img = piece(chess.Piece.from_symbol(symbol))
# svg_str = str(img)
# svg_byte = svg_str.encode()
# encoded = base64.b64encode(svg_byte)
# svg_piece = 'data:image/svg+xml;base64,{}'.format(encoded.decode())
# return svg_piece
def get_svg_board(board, focused_square_ind, only_pieces):
if focused_square_ind is not None:
squares = [focused_square_ind]
else:
squares = []
if board.move_stack:
print('board stack YES')
lastmove = board.peek()
else:
print('board stack NO')
lastmove = None
svg_str = str(svg.board(board, squares=squares, arrows=[], lastmove=lastmove, coordinates=False, only_pieces=only_pieces))
svg_byte = svg_str.encode()
encoded = base64.b64encode(svg_byte)
svg_board = 'data:image/svg+xml;base64,{}'.format(encoded.decode())
return svg_board
#SVG_PIECES = {piece: get_svg_piece(piece) for piece in ('b', 'k', 'n', 'p', 'q', 'r', 'B', 'K', 'N', 'P', 'Q', 'R')}