Add random play bot.

app.py

@@ -5,11 +5,30 @@ from chessmodels import DCMinMax
 import streamlit as st
 from streamlit_image_coordinates import streamlit_image_coordinates
 
-BOT_SELECITONS = ["DeepChessReplicationMinMax"]
+import torch
+
+class RandomPlayer(torch.nn.Module):
+    def __init__(self) -> None:
+        super().__init__()
+
+    def forward(self, board, input):
+        options = torch.arange(input['action_mask'].size)[input['action_mask'].astype(bool)]
+        choice = torch.randint(options.numel(), (1,))
+        return options[choice].item()
+
+BOT_SELECITONS = {
+    "RandomMover":RandomPlayer,
+    "DeepChessReplicationMinMax":DCMinMax,
+    }
 
 @st.cache_resource(max_entries=4)
 def load_model(bot_select):
-    return DCMinMax.from_pretrained("mzimm003/{}".format(bot_select))
+    model = None
+    if bot_select in ["RandomMover"]:
+        model = BOT_SELECITONS[bot_select]()
+    else:
+        model = BOT_SELECITONS[bot_select].from_pretrained("mzimm003/{}".format(bot_select))
+    return model
 
 class TrackPlay:
     def __init__(self):
@@ -73,7 +92,7 @@ def main(kwargs=None):
         with placeholder.container():
             st.session_state.t_p.bot_select = st.selectbox(
                 label="bot",
-                options=BOT_SELECITONS,
+                options=list(BOT_SELECITONS.keys()),
             )
             st.session_state.t_p.active_game = st.button("Play bot!")
             if st.session_state.t_p.active_game: