Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from PIL import Image, ImageDraw
-
+from ai import choose_move
+ 
 def initialize_board():
     board = [[0 for _ in range(8)] for _ in range(8)]
     board[3][3], board[4][4] = 1, 1
@@ -77,7 +78,20 @@ def main():
         status = gr.Text(value="Black: 2 | White: 2 | Black to move", interactive=False)
 
         def click_handler(evt: gr.SelectData, state):
-            return play_move(evt.index[0], evt.index[1], state)
+            # ユーザーの手
+            board_img, status_text, state = play_move(evt.index[0], evt.index[1], state)
+            board, player = state
+            # AI（White）が動く
+            if player == 1:
+                ai_move = choose_move(board, player)
+                if ai_move:
+                    apply_move(board, ai_move[0], ai_move[1], player)
+                    player = -player
+                    black_score, white_score = count_score(board)
+                    status_text = f"Black: {black_score} | White: {white_score} | {'Black' if player == -1 else 'White'} to move"
+                    board_img = board_to_image(board)
+                state = (board, player)
+            return board_img, status_text, state
 
         image_box.select(click_handler, inputs=[state], outputs=[image_box, status, state])
     demo.launch()