fix AI win bug

Gomoku_MCTS/__init__.py

@@ -1,7 +1,7 @@
 from .mcts_pure import MCTSPlayer as MCTSpure
 from .mcts_alphaZero import MCTSPlayer as alphazero
-from .dueling_net import PolicyValueNet
-# from .policy_value_net_pytorch import PolicyValueNet
+# from .dueling_net import PolicyValueNet
+from .policy_value_net_pytorch import PolicyValueNet
 import numpy as np
 
 

pages/Player_VS_AI.py

@@ -332,11 +332,15 @@ def gomoku():
                 # session_state.ROOM.BOARD[gpt_i][gpt_j] = session_state.ROOM.TURN
                 session_state.ROOM.COORDINATE_1D.append(gpt_i * _BOARD_SIZE + gpt_j)
 
-                copy_mcts = deepcopy(session_state.ROOM.MCTS.mcts)
-                _, acts, probs, simul_mean_time = copy_mcts.get_move_probs(session_state.ROOM.BOARD)
-                sorted_acts_probs = sorted(zip(acts, probs), key=lambda x: x[1], reverse=True)
-                top_five_acts = [act for act, prob in sorted_acts_probs[:5]]
-                top_five_probs = [prob for act, prob in sorted_acts_probs[:5]]
+                if not session_state.ROOM.BOARD.game_end()[0]:
+                    copy_mcts = deepcopy(session_state.ROOM.MCTS.mcts)
+                    _, acts, probs, simul_mean_time = copy_mcts.get_move_probs(session_state.ROOM.BOARD)
+                    sorted_acts_probs = sorted(zip(acts, probs), key=lambda x: x[1], reverse=True)
+                    top_five_acts = [act for act, prob in sorted_acts_probs[:5]]
+                    top_five_probs = [prob for act, prob in sorted_acts_probs[:5]]
+                else:
+                    top_five_acts = []
+                    top_five_probs = []
 
                 # construction of clickable buttons
                 for i, row in enumerate(session_state.ROOM.BOARD.board_map):
@@ -354,7 +358,7 @@ def gomoku():
                                 on_click=forbid_click
                             )
                         else:
-                            if session_state.USE_AIAID and i * _BOARD_SIZE + j in top_five_acts:
+                            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(