import streamlit as st
import json
# Initialize session state if not already done
if 'board' not in st.session_state:
st.session_state.board = [
['♜', '♞', '♝', '♛', '♚', '♝', '♞', '♜'],
['♟', '♟', '♟', '♟', '♟', '♟', '♟', '♟'],
[None] * 8,
[None] * 8,
[None] * 8,
[None] * 8,
['♙', '♙', '♙', '♙', '♙', '♙', '♙', '♙'],
['♖', '♘', '♗', '♕', '♔', '♗', '♘', '♖']
]
if 'current_player' not in st.session_state:
st.session_state.current_player = 'white'
if 'selected_piece' not in st.session_state:
st.session_state.selected_piece = None
# Create the HTML/JavaScript for the chess board
chess_board_html = f"""
Current Player: {'White' if st.session_state.current_player == 'white' else 'Black'}
"""
# Generate the squares with pieces
for row in range(8):
for col in range(8):
piece = st.session_state.board[row][col] or ''
is_selected = (st.session_state.selected_piece and
st.session_state.selected_piece['row'] == row and
st.session_state.selected_piece['col'] == col)
square_color = 'white' if (row + col) % 2 == 0 else 'black'
selected_class = ' selected' if is_selected else ''
chess_board_html += f"""
{piece}
"""
# Add JavaScript for handling moves
chess_board_html += """
"""
# Display the chess board
st.components.v1.html(chess_board_html, height=1200)
# Handle moves through Streamlit's session state
if st.session_state.get('clicked_square'):
clicked = st.session_state.clicked_square
row, col = clicked['row'], clicked['col']
if st.session_state.selected_piece:
# Move the piece
selected = st.session_state.selected_piece
st.session_state.board[row][col] = st.session_state.board[selected['row']][selected['col']]
st.session_state.board[selected['row']][selected['col']] = None
st.session_state.selected_piece = None
st.session_state.current_player = 'black' if st.session_state.current_player == 'white' else 'white'
elif st.session_state.board[row][col]:
# Select the piece
st.session_state.selected_piece = {'row': row, 'col': col}
# Clear the clicked square
st.session_state.clicked_square = None
st.rerun()
# Add a reset button
if st.button('Reset Game'):
st.session_state.board = [
['♜', '♞', '♝', '♛', '♚', '♝', '♞', '♜'],
['♟', '♟', '♟', '♟', '♟', '♟', '♟', '♟'],
[None] * 8,
[None] * 8,
[None] * 8,
[None] * 8,
['♙', '♙', '♙', '♙', '♙', '♙', '♙', '♙'],
['♖', '♘', '♗', '♕', '♔', '♗', '♘', '♖']
]
st.session_state.current_player = 'white'
st.session_state.selected_piece = None
st.rerun()