Create app.py

Play TicTacToe with AI

app.py

@@ -0,0 +1,82 @@
+# app.py
+import gradio as gr
+import numpy as np
+from tictactoe import X, O, EMPTY, initial_state, result, terminal, winner, minimax
+
+def initialize_board():
+    return initial_state()
+
+def make_move(board_state, row, col, choice):
+    if not isinstance(board_state, list):
+        board_state = initial_state()
+    
+    # Convert string representation back to None for EMPTY cells
+    board = [[cell if cell in [X, O] else EMPTY for cell in row] for row in board_state]
+    
+    # Player move
+    if board[row][col] != EMPTY:
+        return board_state, "Invalid move! Cell already taken."
+    
+    # Update board with player's move
+    board = result(board, (row, col))
+    
+    # Check if game is over after player's move
+    if terminal(board):
+        game_winner = winner(board)
+        if game_winner:
+            return board, f"Game Over! {game_winner} wins!"
+        return board, "Game Over! It's a tie!"
+    
+    # AI move
+    ai_move = minimax(board)
+    if ai_move:
+        board = result(board, ai_move)
+        
+        # Check if game is over after AI's move
+        if terminal(board):
+            game_winner = winner(board)
+            if game_winner:
+                return board, f"Game Over! {game_winner} wins!"
+            return board, "Game Over! It's a tie!"
+        
+        return board, f"AI moved to position {ai_move}"
+    
+    return board, "Your turn!"
+
+def create_board_interface():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Tic Tac Toe vs AI")
+        gr.Markdown("You play as X, AI plays as O")
+        
+        board_state = gr.State(initialize_board())
+        message = gr.Textbox(label="Game Status", value="Your turn!")
+        
+        board = gr.DataFrame(
+            value=initialize_board(),
+            headers=False,
+            datatype=["str", "str", "str"],
+            row_count=3,
+            col_count=3,
+            interactive=False
+        )
+        
+        def on_cell_click(evt: gr.SelectData, board_state):
+            row, col = evt.index
+            new_board, msg = make_move(board_state, row, col, X)
+            return new_board, new_board, msg
+        
+        board.select(on_cell_click, [board_state], [board_state, board, message])
+        
+        restart_btn = gr.Button("Restart Game")
+        
+        def restart_game():
+            new_board = initialize_board()
+            return new_board, new_board, "Your turn!"
+        
+        restart_btn.click(restart_game, None, [board_state, board, message])
+        
+    return demo
+
+if __name__ == "__main__":
+    demo = create_board_interface()
+    demo.launch()