Spaces:
Sleeping
Sleeping
File size: 2,959 Bytes
d288960 770f950 0154f58 e7988f7 b3c5de3 770f950 22beef3 b3c5de3 770f950 22beef3 770f950 e7988f7 22beef3 e7988f7 770f950 22beef3 |
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 |
from my_chess.scripts.scripts import HumanVsBot
from my_chess.learner.environments import Chess
from chessmodels import DCMinMax
import streamlit as st
from streamlit_image_coordinates import streamlit_image_coordinates
class HvB(HumanVsBot):
@staticmethod
@st.cache_resource()
def get_hvb_manager(
_model,
_environment,
_extra_model_environment_context,
**kwargs):
return HvB(
model = _model,
environment = _environment,
extra_model_environment_context = _extra_model_environment_context,
**kwargs)
@st.cache_resource
def track_play():
class trackPlay:
def __init__(self):
self.active_game = False
self.bot_select = None
def reset(self):
self.active_game = False
self.bot_select = None
return trackPlay()
def information(play, t_p):
display = {
True:{True:"Your turn!",False:"Bot is thinking..."},
False:{"1-0":"White Wins!","0-1":"Black Wins!","1/2-1/2":"It's a Draw!"},
}
hum_turn = play.get_human_player() == play.get_curr_player()
game_over = play.is_done()
choose1 = not game_over
choose2 = play.get_result() if game_over else hum_turn
return display[choose1][choose2]
def reset(t_p):
t_p.reset()
st.cache_resource.clear()
def play(bot_select, t_p):
play = HvB.get_hvb_manager(
_model=DCMinMax.from_pretrained("mzimm003/{}".format(bot_select)),
_environment=Chess(render_mode="rgb_array"),
_extra_model_environment_context=lambda env: {"board":env.board}
)
col1, col2 = st.columns([3,1])
with col1:
st.markdown("# {}".format(information(play, t_p)))
st.session_state["board"] = streamlit_image_coordinates(play.render_board(), key="brd", click_and_drag=True)
with col2:
st.button("Reset", on_click=reset, args=[t_p])
st.markdown("#### You are playing as {}!".format("white" if play.get_human_player() == "player_0" else "black"))
st.markdown("""
1. Play by dragging the piece you want to move.
2. Illegal moves will not be registered.
""")
if not play.is_done():
play.run()
def main(kwargs=None):
css='''
<style>
section.main > div {max-width:75rem}
</style>
'''
st.markdown(css, unsafe_allow_html=True)
t_p = track_play()
placeholder = st.empty()
if not t_p.active_game:
# st.cache_resource.clear()
with placeholder.container():
t_p.bot_select = st.selectbox(
label="bot",
options=["DeepChessReplicationMinMax"],
)
t_p.active_game = st.button("Play bot!")
if t_p.active_game:
st.rerun()
else:
with placeholder.container():
play(t_p.bot_select, t_p)
if __name__ == "__main__":
main() |