fix AI vs AI bug

const.py

@@ -36,6 +36,11 @@ _PLAYER_COLOR = {
     _BLANK: "Blank",
     _BLACK: "YOU HUMAN",
 }
+_PLAYER_COLOR_AI_VS_AI = {
+    _WHITE: "WHITE",
+    _BLANK: "Blank",
+    _BLACK: "BLACK",
+}
 _HORIZONTAL = np.array(
     [
         [0, 0, 0, 0, 0],
@@ -87,6 +92,6 @@ _ROOM_COLOR = {
 
 _MODEL_PATH = {
     "AlphaZero": "Gomoku_MCTS/checkpoint/2023-12-14-18-17-07_test_teaching_learning_collect_epochs=1000_size=9_model=normal/best_policy.model",
-    "duel": "/Users/husky/GomokuDemo/Gomoku_MCTS/checkpoint/2023-12-14-10-22-12_test_teaching_learning_collect_epochs=1000_size=9_model=duel/best_policy.model",
+    "duel": "Gomoku_MCTS/checkpoint/2023-12-14-10-22-12_test_teaching_learning_collect_epochs=1000_size=9_model=duel/best_policy.model",
     "Gumbel AlphaZero": "Gomoku_MCTS/checkpoint/2023-12-14-21-19-40_selfplay_epochs=1000_size=9_model=gumbel/best_policy.model",
 }

pages/AI_VS_AI.py

@@ -24,7 +24,7 @@ from const import (
     _BLACK,  # 1, for human
     _WHITE,  # 2 , for AI
     _BLANK,
-    _PLAYER_COLOR,
+    _PLAYER_COLOR_AI_VS_AI,
     _PLAYER_SYMBOL1,
     _PLAYER_SYMBOL2,
     _ROOM_COLOR,
@@ -438,38 +438,50 @@ def gomoku():
                                 in session_state.ROOM.COORDINATE_1D
                         ):
                             if i == gpt_i and j == gpt_j:
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[_NEW],
-                                    key=f"{i}:{j}",
-                                    args=(i, j),
-                                    on_click=forbid_click,
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[_NEW],
+                                #     key=f"{i}:{j}",
+                                #     args=(i, j),
+                                #     on_click=forbid_click,
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[_NEW]
                                 )
                             else:
                                 # disable click for GPT choices
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[cell],
-                                    key=f"{i}:{j}",
-                                    args=(i, j),
-                                    on_click=forbid_click
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[cell],
+                                #     key=f"{i}:{j}",
+                                #     args=(i, j),
+                                #     on_click=forbid_click
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[cell]
                                 )
                         else:
                             if session_state.USE_AIAID and i * _BOARD_SIZE + j in top_five_acts and not \
                                     session_state.ROOM.BOARD.game_end()[0]:
                                 # enable click for other cells available for human choices
                                 prob = top_five_probs[top_five_acts.index(i * _BOARD_SIZE + j)]
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[cell] + f"({round(prob, 2)})",
-                                    key=f"{i}:{j}",
-                                    on_click=forbid_click,
-                                    args=(i, j),
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[cell] + f"({round(prob, 2)})",
+                                #     key=f"{i}:{j}",
+                                #     on_click=forbid_click,
+                                #     args=(i, j),
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[cell]
                                 )
                             else:
                                 # enable click for other cells available for human choices
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[cell],
-                                    key=f"{i}:{j}",
-                                    on_click=forbid_click,
-                                    args=(i, j),
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[cell],
+                                #     key=f"{i}:{j}",
+                                #     on_click=forbid_click,
+                                #     args=(i, j),
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[cell]
                                 )
 
             message.markdown(
@@ -512,7 +524,6 @@ def gomoku():
                 print("AI's move is located at ({}, {}) :".format(gpt_i, gpt_j))
                 move = session_state.ROOM.BOARD.location_to_move((gpt_i, gpt_j))
                 print("Location to move: ", move)
-                # print("Location to move: ", move)
                 # MCTS BOARD
                 session_state.ROOM.BOARD.do_move(move)
                 # Gomoku Bot BOARD
@@ -541,38 +552,50 @@ def gomoku():
                                 in session_state.ROOM.COORDINATE_1D
                         ):
                             if i == gpt_i and j == gpt_j:
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[_NEW],
-                                    key=f"{i}:{j}",
-                                    args=(i, j),
-                                    on_click=forbid_click,
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[_NEW],
+                                #     key=f"{i}:{j}",
+                                #     args=(i, j),
+                                #     on_click=forbid_click,
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[_NEW]
                                 )
                             else:
                                 # disable click for GPT choices
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[cell],
-                                    key=f"{i}:{j}",
-                                    args=(i, j),
-                                    on_click=forbid_click
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[cell],
+                                #     key=f"{i}:{j}",
+                                #     args=(i, j),
+                                #     on_click=forbid_click
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[cell]
                                 )
                         else:
                             if session_state.USE_AIAID and i * _BOARD_SIZE + j in top_five_acts and not \
                                     session_state.ROOM.BOARD.game_end()[0]:
                                 # enable click for other cells available for human choices
                                 prob = top_five_probs[top_five_acts.index(i * _BOARD_SIZE + j)]
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[cell] + f"({round(prob, 2)})",
-                                    key=f"{i}:{j}",
-                                    on_click=forbid_click,
-                                    args=(i, j),
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[cell] + f"({round(prob, 2)})",
+                                #     key=f"{i}:{j}",
+                                #     on_click=forbid_click,
+                                #     args=(i, j),
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[cell]
                                 )
                             else:
                                 # enable click for other cells available for human choices
-                                BOARD_PLATE[i][j].button(
-                                    session_state.Symbol[cell],
-                                    key=f"{i}:{j}",
-                                    on_click=forbid_click,
-                                    args=(i, j),
+                                # BOARD_PLATE[i][j].button(
+                                #     session_state.Symbol[cell],
+                                #     key=f"{i}:{j}",
+                                #     on_click=forbid_click,
+                                #     args=(i, j),
+                                # )
+                                BOARD_PLATE[i][j].write(
+                                    session_state.Symbol[cell]
                                 )
 
             message.markdown(
@@ -623,21 +646,23 @@ def gomoku():
     # Infos
     def update_info() -> None:
         # Additional information
-        SCORE_PLATE[0].metric("Gomoku-Agent", session_state.ROOM.HISTORY[0])
-        SCORE_PLATE[1].metric("Black", session_state.ROOM.HISTORY[1])
         if session_state.ROOM.WINNER != _BLANK:
             st.balloons()
             ROUND_INFO.write(
-                f"#### **{_PLAYER_COLOR[session_state.ROOM.WINNER]} WIN!**\n**Click buttons on the left for more plays.**"
+                f"#### **{_PLAYER_COLOR_AI_VS_AI[session_state.ROOM.WINNER]} WIN!**\n**Click buttons on the left for more plays.**"
             )
         st.markdown("<br>", unsafe_allow_html=True)
         st.markdown("<br>", unsafe_allow_html=True)
         chart_data = pd.DataFrame(session_state.ROOM.ai_simula_time_list, columns=["Simulation Time"])
-        st.line_chart(chart_data)
-
-    game_control()
-    update_info()
-
+        chart_placeholder.line_chart(chart_data)
+
+    
+    chart_placeholder = st.empty()
+    while session_state.ROOM.WINNER == _BLANK:
+        game_control()
+        update_info()
+    SCORE_PLATE[0].metric("Gomoku-Agent", session_state.ROOM.HISTORY[0])
+    SCORE_PLATE[1].metric("Black", session_state.ROOM.HISTORY[1])
 
 if __name__ == "__main__":
     gomoku()