random move play

src/play_interface.py

@@ -61,13 +61,11 @@ def play_user_move(
 def play_ai_move(
     current_board: chess.Board,
     temperature: float = 0.1,
-    top_k: int = 3,
 ):
     print(f"Playing AI move with temperature {temperature}")
     uci_move = inference_fn(
         prompt=f"FEN: {current_board.fen()}\nMOVE:",
         temperature=temperature,
-        top_k=top_k,
     )
     current_board.push_uci(uci_move.strip())
     return current_board
@@ -108,7 +106,9 @@ def try_play_move(
             gr.Warning(f"AI move failed with temperature {temperature}")
     else:
         gr.Warning("AI move failed with all temperatures")
-        current_board.pop()
+        random_move = random.choice(list(current_board.legal_moves))
+        gr.Warning(f"Playing random move {random_move}")
+        current_board.push(random_move)
         return *render_board(current_board), current_board
     return *render_board(current_board), current_board
 
@@ -165,6 +165,11 @@ with gr.Blocks() as interface:
         inputs=[*static_inputs, state_board],
         outputs=[*static_outputs, state_board],
     )
+    move_to_play.submit(
+        try_play_move,
+        inputs=[*static_inputs, state_board],
+        outputs=[*static_outputs, state_board],
+    )
 
     def reset_board():
         board = chess.Board()