Spaces:
Sleeping
Sleeping
File size: 4,997 Bytes
6ac8ae0 3f02d24 6ac8ae0 3f02d24 6ac8ae0 |
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 |
import streamlit as st
import numpy as np
from objects import Player, HumanPlayer, State
import random
# Create RL bot players
p1 = Player("p1")
p2 = HumanPlayer("p2")
# Initialize the selected opponent in Session State
if "selected_opponent" not in st.session_state:
st.session_state.selected_opponent = 'Computer'
def handle_click(i, j):
if (i, j) not in check_available_moves(extra=True):
st.session_state.warning = True
elif not st.session_state.winner:
st.session_state.warning = False
st.session_state.board[i, j] = st.session_state.player
st.session_state.player = "O" if st.session_state.player == "X" else "X"
winner = check_win(st.session_state.board)
if winner != ".":
st.session_state.winner = winner
if st.session_state.opponent == 'Computer':
# Give reward to the RL bot and update its policy
if winner == 'X':
p1.feedReward(1)
elif winner == 'O':
p1.feedReward(0)
else:
p1.feedReward(0.1)
# Save the RL bot's policy
p1.savePolicy()
def init(post_init=False):
if not post_init:
st.session_state.opponent = 'Human'
st.session_state.win = {'X': 0, 'O': 0}
st.session_state.board = np.full((3, 3), '.', dtype=str)
st.session_state.player = 'X'
st.session_state.warning = False
st.session_state.winner = None
st.session_state.over = False
def check_available_moves(extra=False) -> list:
raw_moves = [row for col in st.session_state.board.tolist() for row in col]
num_moves = [i for i, spot in enumerate(raw_moves) if spot == '.']
if extra:
return [(i // 3, i % 3) for i in num_moves]
return num_moves
def check_rows(board):
for row in board:
if len(set(row)) == 1:
return row[0]
return None
def check_diagonals(board):
if len(set([board[i][i] for i in range(len(board))])) == 1:
return board[0][0]
if len(set([board[i][len(board) - i - 1] for i in range(len(board))])) == 1:
return board[0][len(board) - 1]
return None
def check_state():
if st.session_state.winner:
st.success(f"Congrats! {st.session_state.winner} won the game! 🎈")
if st.session_state.warning and not st.session_state.over:
st.warning('⚠️ This move already exists')
if st.session_state.winner and not st.session_state.over:
st.session_state.over = True
st.session_state.win[st.session_state.winner] = (
st.session_state.win.get(st.session_state.winner, 0) + 1
)
elif not check_available_moves() and not st.session_state.winner:
st.info(f"It's a tie 📍")
st.session_state.over = True
def check_win(board):
for new_board in [board, np.transpose(board)]:
result = check_rows(new_board)
if result:
return result
return check_diagonals(board)
def computer_player():
moves = check_available_moves(extra=True)
if moves:
# Use p1 to choose the action
positions = check_available_moves(extra=True)
p1_action = p1.chooseAction(positions, st.session_state.board, -1)
# Check if the chosen action is valid and make the move
if p1_action in moves:
i, j = p1_action
handle_click(i, j)
def on_opponent_selected():
st.session_state.selected_opponent = st.session_state.opponent
init(True)
def main():
st.write(
"""
# ❎🅾️ Tic Tac Toe
"""
)
if "board" not in st.session_state:
init()
reset, score, player, settings = st.columns([0.5, 0.6, 1, 1])
reset.button('New game', on_click=init, args=(True,))
# Dynamically set the Expander label based on the selected opponent
with settings.expander(f'Opponent ({st.session_state.selected_opponent})'):
st.write('**Warning**: changing this setting will restart your game')
st.selectbox(
'Set opponent',
['Computer','Human'],
key='opponent',
on_change=on_opponent_selected,
)
for i, row in enumerate(st.session_state.board):
cols = st.columns([5, 1, 1, 1, 5])
for j, field in enumerate(row):
cols[j + 1].button(
field,
key=f"{i}-{j}",
on_click=handle_click
if st.session_state.player == 'X'
or st.session_state.opponent == 'Human'
else computer_player(),
args=(i, j),
)
check_state()
score.button(f'❌{st.session_state.win["X"]} 🆚 {st.session_state.win["O"]}⭕')
player.button(
f'{"❌" if st.session_state.player == "X" else "⭕"}\'s turn'
if not st.session_state.winner
else f'🏁 Game finished'
)
if __name__ == '__main__':
main()
|