Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
 
+# 🧠 Game logic
 def check_winner(board):
     for i in range(3):
         if abs(sum(board[i, :])) == 3:
@@ -16,7 +17,7 @@ def check_winner(board):
     return None
 
 def ai_move(board):
-    # Smart AI: Win if possible, block if needed
+    # Smart AI: try to win or block
     for player in [-1, 1]:
         for i in range(3):
             for j in range(3):
@@ -33,32 +34,32 @@ def ai_move(board):
     open_cells = np.argwhere(board == 0)
     return tuple(open_cells[0])
 
-def play(user_row, user_col, board_flat):
+def play(board_flat, r, c):
     board = np.array(board_flat).reshape(3, 3)
-    if board[user_row, user_col] != 0:
-        return board.flatten().tolist(), "Cell already occupied! Try another."
+    if board[r, c] != 0:
+        return board_flat, "🚫 Cell already taken! Try another.", board_flat
 
-    board[user_row, user_col] = 1
+    board[r, c] = 1
     winner = check_winner(board)
     if winner == 1:
-        return board.flatten().tolist(), "✅ You win!"
+        return board.flatten().tolist(), "✅ You win!", board.flatten().tolist()
     elif winner == 0:
-        return board.flatten().tolist(), "🤝 It's a draw!"
+        return board.flatten().tolist(), "🤝 It's a draw!", board.flatten().tolist()
 
     ai_r, ai_c = ai_move(board)
     board[ai_r, ai_c] = -1
     winner = check_winner(board)
     if winner == -1:
-        return board.flatten().tolist(), "🤖 AI wins!"
+        return board.flatten().tolist(), "🤖 AI wins!", board.flatten().tolist()
     elif winner == 0:
-        return board.flatten().tolist(), "🤝 It's a draw!"
+        return board.flatten().tolist(), "🤝 It's a draw!", board.flatten().tolist()
 
-    return board.flatten().tolist(), "Your turn!"
+    return board.flatten().tolist(), "Your turn!", board.flatten().tolist()
 
 def new_game():
     return [0]*9, "🎮 New game started! You go first."
 
-def update_button_labels(board):
+def update_board_labels(board):
     labels = []
     for v in board:
         if v == 1:
@@ -66,9 +67,10 @@ def update_button_labels(board):
         elif v == -1:
             labels.append("⭕")
         else:
-            labels.append("")
+            labels.append("⬜")
     return labels
 
+# 🎨 Gradio interface
 with gr.Blocks() as demo:
     board_state = gr.State([0]*9)
 
@@ -79,22 +81,22 @@ with gr.Blocks() as demo:
         for r in range(3):
             with gr.Row():
                 for c in range(3):
-                    btn = gr.Button("", elem_id=f"cell-{r}-{c}", size="lg")
+                    btn = gr.Button("⬜", elem_id=f"cell-{r}-{c}", size="lg")
                     btns.append((btn, r, c))
 
     reset = gr.Button("🔄 Reset Game")
 
-    def handle_click(r, c, board_flat):
-        board, msg = play(r, c, board_flat)
-        return board, msg, board
-
     for btn, r, c in btns:
+        def on_click(board_flat, r=r, c=c):
+            board, msg, _ = play(board_flat, r, c)
+            return board, msg, board
+
         btn.click(
-            handle_click,
-            inputs=[gr.Number(r), gr.Number(c), board_state],
+            on_click,
+            inputs=[board_state],
             outputs=[board_state, status, board_state],
         ).then(
-            update_button_labels,
+            update_board_labels,
             inputs=board_state,
             outputs=[b for b, _, _ in btns]
         )
@@ -103,7 +105,7 @@ with gr.Blocks() as demo:
         new_game,
         outputs=[board_state, status]
     ).then(
-        update_button_labels,
+        update_board_labels,
         inputs=board_state,
         outputs=[b for b, _, _ in btns]
     )