File size: 8,348 Bytes
62698b8 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 ac19806 8ec51c0 62698b8 8ec51c0 90fd6a1 8ec51c0 90fd6a1 8ec51c0 fb800ba 8ec51c0 62698b8 8ec51c0 62698b8 8ec51c0 ac19806 8ec51c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 |
import streamlit as st
import streamlit.components.v1 as components
from streamlit.components.v1 import html
import chess
import streamlit_scrollable_textbox as stx
from st_bridge import bridge
from modules.chess import Chess
from modules.utility import set_page
from modules.states import init_states
import datetime as dt
from gradio_client import Client
import random
set_page(title='Chess vs LLaMA 3.1 405B', page_icon="♟️")
init_states()
st.session_state.board_width = 400
# Initialize the LLaMA 3.1 405B client
llama_client = Client("xianbao/SambaNova-fast")
# Initialize all session state variables
if 'player_color' not in st.session_state:
st.session_state.player_color = 'white'
if 'current_turn' not in st.session_state:
st.session_state.current_turn = 'white'
if 'game_started' not in st.session_state:
st.session_state.game_started = False
if 'curfen' not in st.session_state:
st.session_state.curfen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1"
if 'lastfen' not in st.session_state:
st.session_state.lastfen = None
if 'moves' not in st.session_state:
st.session_state.moves = {}
if 'curside' not in st.session_state:
st.session_state.curside = 'white'
def get_ai_move(fen):
board = chess.Board(fen)
legal_moves = list(board.legal_moves)
if not legal_moves:
return None
prompt = f"You are a chess engine. Given the following chess position in FEN notation: {fen}, suggest a good move. Respond with only the move in UCI notation (e.g., e2e4)."
for _ in range(3): # Try up to 3 times to get a valid move from the AI
try:
response = llama_client.predict(
message=prompt,
system_message="You are a chess engine assistant.",
max_tokens=10,
temperature=0.7, # Increased temperature for more varied moves
top_p=0.9,
top_k=50,
api_name="/chat"
)
move = chess.Move.from_uci(response.strip())
if move in legal_moves:
return move.uci()
except ValueError:
pass # If the AI produces an invalid move, we'll try again
# If the AI fails to produce a valid move after 3 attempts, choose a random legal move
return random.choice(legal_moves).uci()
def reset_game(player_color):
st.session_state.player_color = player_color
st.session_state.curfen = "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1"
st.session_state.moves = {}
st.session_state.current_turn = 'white'
st.session_state.game_started = True
st.session_state.lastfen = None
st.session_state.game_over = False
# If player chose black, make the first move for AI
if st.session_state.player_color == 'black':
ai_move = get_ai_move(st.session_state.curfen)
board = chess.Board(st.session_state.curfen)
if ai_move:
move = chess.Move.from_uci(ai_move)
board.push(move)
st.session_state.curfen = board.fen()
st.session_state.moves.update(
{
st.session_state.curfen : {
'side': 'white',
'curfen': st.session_state.curfen,
'last_fen': "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1",
'last_move': ai_move,
'data': None,
'timestamp': str(dt.datetime.now())
}
}
)
st.session_state.current_turn = 'black'
def check_game_end(board):
outcome = board.outcome()
if outcome:
st.session_state.game_over = True
if outcome.winner is None:
return "Draw"
return "White" if outcome.winner else "Black"
return None
st.title("Chess vs LLaMA 3.1 405B")
# Game controls
col1, col2, col3 = st.columns([1,1,1])
with col1:
player_color = st.selectbox("Choose your color", ['white', 'black'], key='color_select')
with col2:
if st.button('Start New Game', key='start_game'):
reset_game(player_color)
st.rerun()
with col3:
st.write(f"Current turn: {st.session_state.current_turn}")
st.write(f"Your color: {st.session_state.player_color}")
# Get the info from current board after the user made the move.
data = bridge("my-bridge")
if data is not None and st.session_state.game_started and not st.session_state.game_over:
st.session_state.lastfen = st.session_state.curfen
st.session_state.curfen = data['fen']
st.session_state.curside = data['move']['color'].replace('w','white').replace('b','black')
st.session_state.moves.update(
{
st.session_state.curfen : {
'side':st.session_state.curside,
'curfen':st.session_state.curfen,
'last_fen':st.session_state.lastfen,
'last_move':data['pgn'],
'data': None,
'timestamp': str(dt.datetime.now())
}
}
)
st.session_state.current_turn = 'white' if st.session_state.curside == 'black' else 'black'
board = chess.Board(st.session_state.curfen)
game_result = check_game_end(board)
if game_result:
st.success(f"Game Over! Winner: {game_result}")
elif st.session_state.current_turn != st.session_state.player_color:
# AI's turn
ai_move = get_ai_move(st.session_state.curfen)
if ai_move:
move = chess.Move.from_uci(ai_move)
board.push(move)
st.session_state.curfen = board.fen()
st.session_state.moves.update(
{
st.session_state.curfen : {
'side': st.session_state.current_turn,
'curfen': st.session_state.curfen,
'last_fen': st.session_state.lastfen,
'last_move': ai_move,
'data': None,
'timestamp': str(dt.datetime.now())
}
}
)
st.session_state.current_turn = st.session_state.player_color
game_result = check_game_end(board)
if game_result:
st.success(f"Game Over! Winner: {game_result}")
else:
st.error("The AI couldn't make a move. The game may be over.")
# Main game display
cols = st.columns([3, 2])
with cols[0]:
if st.session_state.game_started:
puzzle = Chess(st.session_state.board_width, st.session_state.curfen)
components.html(
puzzle.puzzle_board(),
height=st.session_state.board_width + 75,
scrolling=False
)
board = chess.Board(st.session_state.curfen)
# Game status
status_col1, status_col2 = st.columns(2)
with status_col1:
st.write("Game Status:")
st.write(f"Check: {'Yes' if board.is_check() else 'No'}")
st.write(f"Checkmate: {'Yes' if board.is_checkmate() else 'No'}")
with status_col2:
st.write("\u200B") # Invisible character for alignment
st.write(f"Stalemate: {'Yes' if board.is_stalemate() else 'No'}")
st.write(f"Insufficient material: {'Yes' if board.is_insufficient_material() else 'No'}")
if st.session_state.game_over:
st.success(f"Game Over! Winner: {check_game_end(board)}")
else:
st.info("Welcome to Chess vs LLaMA 3.1 405B!")
st.write("To start a new game:")
st.write("1. Choose your color (white or black)")
st.write("2. Click 'Start New Game'")
st.write("3. Make your moves on the chess board")
st.write("Enjoy playing against the AI!")
with cols[1]:
if st.session_state.game_started:
st.subheader("Move History")
records = [
f"##### {value['timestamp'].split('.')[0]} \n {value['side']} - {value.get('last_move','')}"
for key, value in st.session_state['moves'].items()
]
stx.scrollableTextbox('\n\n'.join(records), height = 400, border=True)
else:
st.image("https://upload.wikimedia.org/wikipedia/commons/6/6f/ChessSet.jpg", caption="Chess pieces", use_column_width=True) |