Update app.py

app.py

@@ -27,7 +27,13 @@ def check_draw(board):
     return all(cell != "" for row in board for cell in row)
 
 # Handle a move
-def handle_move(board, current_player, button_idx):
+def handle_move(board, current_player, button_idx, game_status):
+    # If the game is already over (winner or draw), don't allow further moves
+    if "wins" in game_status or "draw" in game_status:
+        status = game_status  # No change to status if the game is over
+        buttons = [gr.Button(value=board[i//3][i%3], elem_classes=["cell-btn"], interactive=False) for i in range(9)]
+        return board, current_player, status, *buttons
+
     row, col = divmod(button_idx, 3)
     if board[row][col] != "":
         status = f"Invalid move! Player {1 if current_player == 'X' else 2}'s turn ({current_player})"
@@ -38,12 +44,12 @@ def handle_move(board, current_player, button_idx):
     winner = check_winner(board)
     if winner:
         status = f"Player {1 if winner == 'X' else 2} ({winner}) wins! 🎉"
-        buttons = [gr.Button(value=board[i//3][i%3], elem_classes=["cell-btn"]) for i in range(9)]
+        buttons = [gr.Button(value=board[i//3][i%3], elem_classes=["cell-btn"], interactive=False) for i in range(9)]
         return board, current_player, status, *buttons
 
     if check_draw(board):
         status = "It's a draw! 🤝"
-        buttons = [gr.Button(value=board[i//3][i%3], elem_classes=["cell-btn"]) for i in range(9)]
+        buttons = [gr.Button(value=board[i//3][i%3], elem_classes=["cell-btn"], interactive=False) for i in range(9)]
         return board, current_player, status, *buttons
 
     current_player = "O" if current_player == "X" else "X"
@@ -74,7 +80,7 @@ with gr.Blocks(css=".cell-btn {height: 100px; width: 100px; font-size: 2em; text
     for idx, btn in enumerate(buttons):
         btn.click(
             handle_move,
-            inputs=[board_state, current_player, gr.Number(idx, visible=False)],
+            inputs=[board_state, current_player, gr.Number(idx, visible=False), game_status],
             outputs=[board_state, current_player, game_status, *buttons],  # Return all outputs
         )