Update app.py

app.py

@@ -79,25 +79,37 @@ def board_to_image(board, state):
     return img
 
 def click_handler(evt, state):
+    # Ensure state initialized
+    if state is None:
+        state = (initialize_board(), -1, None, None, [])
     board, player, lu, la, history = state
-    x,y=evt.index; cell=360//8; c,r=int(x//cell),int((y-cell*2)//cell)
-    if 0<=r<8 and 0<=c<8 and not is_game_over(board):
-        if apply_move(board,r,c,player): lu,player=( (r,c), -player )
-    # after user
-    yield board_to_image(board,(board,player,lu,la,history)),(board,player,lu,la,history)
-    # if game over
+    x, y = evt.index
+    cell = 360 // 8
+    c, r = int(x // cell), int((y - cell * 2) // cell)
+    # Player move
+    if 0 <= r < 8 and 0 <= c < 8 and not is_game_over(board):
+        if apply_move(board, r, c, player):
+            lu, player = (r, c), -player
+    # Update after player
+    yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
+    # Check end
     if is_game_over(board):
-        history.append(f"Game {len(history)+1}: " + ("Draw" if count_score(board)[0]==count_score(board)[1] else ("Black" if count_score(board)[0]>count_score(board)[1] else "White")))
-        yield board_to_image(board,(board,player,lu,la,history)),(board,player,lu,la,history)
+        winner = "Draw" if count_score(board)[0] == count_score(board)[1] else ("Black" if count_score(board)[0] > count_score(board)[1] else "White")
+        history.append(f"Game {len(history)+1}: {winner}")
+        yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
         return
-    # AI thinking
-    yield board_to_image(board,(board,player,lu,la,history)),(board,player,lu,la,history)
+    # AI thinking indicator
+    yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
     time.sleep(2)
-    ai_mv=choose_move(board,player)
-    if ai_mv: apply_move(board,ai_mv[0],ai_mv[1],player); la,player=( ai_mv, -player )
-    yield board_to_image(board,(board,player,lu,la,history)),(board,player,lu,la,history)
+    # AI move
+    ai_mv = choose_move(board, player)
+    if ai_mv:
+        apply_move(board, ai_mv[0], ai_mv[1], player)
+        la, player = ai_mv, -player
+    yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
 
-def reset_handler():
+# def reset_handler placeholders for replacement
+#def reset_handler():
     return initialize_board(),-1,None,None,[]
 
 with gr.Blocks() as demo: