Allow multiple, distinct, sessions on server.

Dockerfile

@@ -1,6 +1,3 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
-
 FROM python:3.8.12
 
 RUN useradd -m -u 1000 user
@@ -13,11 +10,10 @@ COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-RUN git clone https://github.com/mzimm003/Chess /app/Chess
-RUN pip install /app/Chess
+RUN git clone https://github.com/mzimm003/Chess /home/user/Chess
+RUN pip install /home/user/Chess
 
 COPY --chown=user . /app
-# CMD ["uvicorn", "app:main", "--host", "0.0.0.0", "--port", "7860"]
 
 EXPOSE 8501
 CMD streamlit run app.py \

app.py

@@ -7,20 +7,17 @@ from streamlit_image_coordinates import streamlit_image_coordinates
 
 BOT_SELECITONS = ["DeepChessReplicationMinMax"]
 
-@st.cache_resource
+@st.cache_resource(max_entries=4)
 def load_model(bot_select):
     return DCMinMax.from_pretrained("mzimm003/{}".format(bot_select))
 
-@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()
+class TrackPlay:
+    def __init__(self):
+        self.active_game = False
+        self.bot_select = None
+    def reset(self):
+        self.active_game = False
+        self.bot_select = None
 
 def information(play, t_p):
     display = {
@@ -38,24 +35,26 @@ def reset(t_p):
     st.cache_resource.clear()
 
 def play(bot_select, t_p):
-    play = HumanVsBot(
-        model=load_model(bot_select=bot_select),
-        environment=Chess(render_mode="rgb_array"),
-        extra_model_environment_context=lambda env: {"board":env.board},
-    )
+    if "play" not in st.session_state:
+        st.session_state.play = HumanVsBot(
+            model=load_model(bot_select=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)
+        st.markdown("# {}".format(information(st.session_state.play, t_p)))
+        st.session_state["board"] = streamlit_image_coordinates(st.session_state.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("#### You are playing as {}!".format("white" if st.session_state.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.
+                    2. The piece won't move as you drag, but the move will be registered upon release.
+                    3. Illegal moves will not be registered.
                     """)
-    if not play.is_done():
-        play.run()
+    if not st.session_state.play.is_done():
+        st.session_state.play.run()
 
 def main(kwargs=None):
     css='''
@@ -65,22 +64,23 @@ def main(kwargs=None):
     '''
     st.markdown(css, unsafe_allow_html=True)
 
-    t_p = track_play()
+    if "t_p" not in st.session_state:
+        st.session_state.t_p = TrackPlay()
     placeholder = st.empty()
 
-    if not t_p.active_game:
+    if not st.session_state.t_p.active_game:
         # st.cache_resource.clear()
         with placeholder.container():
-            t_p.bot_select = st.selectbox(
+            st.session_state.t_p.bot_select = st.selectbox(
                 label="bot",
                 options=BOT_SELECITONS,
             )
-            t_p.active_game = st.button("Play bot!")
-            if t_p.active_game:
+            st.session_state.t_p.active_game = st.button("Play bot!")
+            if st.session_state.t_p.active_game:
                 st.rerun()
     else:
         with placeholder.container():
-            play(t_p.bot_select, t_p)
+            play(st.session_state.t_p.bot_select, st.session_state.t_p)
 
 if __name__ == "__main__":
     main()