Update app.py

app.py

@@ -1,87 +1,37 @@
 import gradio as gr
 import random
 
-# Initialize the game board and state
-def initialize_game():
-    board = [["" for _ in range(3)] for _ in range(3)]
-    current_player = "X"
-    status = "Player 1's turn (X)"
-    buttons = [gr.Button(value="", elem_classes=["cell-btn"], interactive=True) for _ in range(9)]
-    return board, current_player, status, *buttons
-
-# Check for a winner
+
+# Check if there's a winner
 def check_winner(board):
-    for i in range(3):
-        if board[i][0] == board[i][1] == board[i][2] and board[i][0] != "":
-            return board[i][0]
-        if board[0][i] == board[1][i] == board[2][i] and board[0][i] != "":
-            return board[0][i]
-    if board[0][0] == board[1][1] == board[2][2] and board[0][0] != "":
+    for row in board:
+        if row[0] == row[1] == row[2] != "":
+            return row[0]
+    for col in range(3):
+        if board[0][col] == board[1][col] == board[2][col] != "":
+            return board[0][col]
+    if board[0][0] == board[1][1] == board[2][2] != "":
         return board[0][0]
-    if board[0][2] == board[1][1] == board[2][0] and board[0][2] != "":
+    if board[0][2] == board[1][1] == board[2][0] != "":
         return board[0][2]
     return None
 
-# Check for a draw
+
+# Check if it's a draw
 def check_draw(board):
-    return all(cell != "" for row in board for cell in row)
+    for row in board:
+        if "" in row:
+            return False
+    return True
 
-# Minimax algorithm for AI's move
-def minimax(board, depth, is_maximizing, difficulty):
-    winner = check_winner(board)
-    if winner == "X":
-        return -10 + depth
-    elif winner == "O":
-        return 10 - depth
-    elif check_draw(board):
-        return 0
-
-    if is_maximizing:
-        best = -float('inf')
-        for i in range(3):
-            for j in range(3):
-                if board[i][j] == "":
-                    board[i][j] = "O"
-                    best = max(best, minimax(board, depth + 1, False, difficulty))
-                    board[i][j] = ""
-        return best
-    else:
-        best = float('inf')
-        for i in range(3):
-            for j in range(3):
-                if board[i][j] == "":
-                    board[i][j] = "X"
-                    best = min(best, minimax(board, depth + 1, True, difficulty))
-                    board[i][j] = ""
-        return best
-
-# Find the best move for AI
+
+# AI move (random for simplicity)
 def get_best_move(board, difficulty):
-    if difficulty == "easy":
-        # Random move for easy level
-        empty_cells = [(i, j) for i in range(3) for j in range(3) if board[i][j] == ""]
-        if empty_cells:
-            return random.choice(empty_cells)
-
-    if difficulty == "intermediate":
-        # Mixed strategy for intermediate
-        if random.random() < 0.5:
-            return get_best_move(board, "easy")
-    
-    best_val = -float('inf')
-    best_move = (-1, -1)
-    for i in range(3):
-        for j in range(3):
-            if board[i][j] == "":
-                board[i][j] = "O"
-                move_val = minimax(board, 0, False, difficulty)
-                board[i][j] = ""
-                if move_val > best_val:
-                    best_move = (i, j)
-                    best_val = move_val
-    return best_move
-
-# Handle a move
+    empty_cells = [(row, col) for row in range(3) for col in range(3) if board[row][col] == ""]
+    return random.choice(empty_cells) if empty_cells else None
+
+
+# Handle the move
 def handle_move(board, current_player, button_idx, game_status, difficulty):
     if "wins" in game_status or "draw" in game_status:
         buttons = [gr.Button(value=board[i // 3][i % 3], elem_classes=["cell-btn"], interactive=False) for i in range(9)]
@@ -97,13 +47,15 @@ def handle_move(board, current_player, button_idx, game_status, difficulty):
     winner = check_winner(board)
     if winner:
         status = f"Player {1 if winner == 'X' else 2} ({winner}) wins! 🎉"
+        winner_message = f"<h1 style='font-size: 2.5em; color: green; text-align: center;'>Player {1 if winner == 'X' else 2} ({winner}) Wins! 🎉</h1>"
         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
+        return board, current_player, status, winner_message, *buttons
 
     if check_draw(board):
         status = "It's a draw! 🤝"
+        winner_message = "<h1 style='font-size: 2.5em; color: blue; text-align: center;'>It's a Draw! 🤝</h1>"
         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
+        return board, current_player, status, winner_message, *buttons
 
     # AI's turn
     if current_player == "X":
@@ -113,68 +65,59 @@ def handle_move(board, current_player, button_idx, game_status, difficulty):
         winner = check_winner(board)
         if winner:
             status = f"AI ({winner}) wins! 🎉"
+            winner_message = f"<h1 style='font-size: 2.5em; color: red; text-align: center;'>AI ({winner}) Wins! 🎉</h1>"
             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
+            return board, current_player, status, winner_message, *buttons
 
         if check_draw(board):
             status = "It's a draw! 🤝"
+            winner_message = "<h1 style='font-size: 2.5em; color: blue; text-align: center;'>It's a Draw! 🤝</h1>"
             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
+            return board, current_player, status, winner_message, *buttons
 
         current_player = "X"
         status = f"Player 1's turn (X)"
 
+    winner_message = ""  # No winner yet
     buttons = [gr.Button(value=board[i // 3][i % 3], elem_classes=["cell-btn"]) for i in range(9)]
-    return board, current_player, status, "", *buttons
-
-
-# Build the Gradio UI
-with gr.Blocks(css="""
-.cell-btn {
-    height: 100px;
-    width: 100px;
-    font-size: 2em;
-    text-align: center;
-}
-.winner-highlight {
-    font-size: 3em;
-    font-weight: bold;
-    color: green;
-    text-align: center;
-    margin-top: 20px;
-}
-""") as tic_tac_toe:
-    gr.Markdown("## Tic-Tac-Toe with AI 🎮")
-    difficulty = gr.Radio(["easy", "intermediate", "impossible"], value="impossible", label="Difficulty Level")
-
-    # Initialize states
-    board_state = gr.State([["" for _ in range(3)] for _ in range(3)])
+    return board, current_player, status, winner_message, *buttons
+
+
+# Reset the game
+def reset_game():
+    board = [[""] * 3 for _ in range(3)]
+    buttons = [gr.Button(value="", elem_classes=["cell-btn"]) for _ in range(9)]
+    return board, "X", "Player 1's turn (X)", "", *buttons
+
+
+# Initial game setup
+board = [[""] * 3 for _ in range(3)]
+buttons = [gr.Button(value="", elem_classes=["cell-btn"]) for _ in range(9)]
+
+with gr.Blocks(css=".cell-btn {height: 100px; width: 100px; font-size: 2em; margin: 5px;}") as app:
+    gr.Markdown("<h1 style='text-align: center;'>Tic Tac Toe Game</h1>")
+    game_status = gr.Textbox("Player 1's turn (X)", interactive=False, label="Game Status")
+    winner_display = gr.HTML()
+
+    with gr.Row():
+        button_components = []
+        for i in range(3):
+            with gr.Row():
+                for j in range(3):
+                    button_components.append(gr.Button("", elem_classes=["cell-btn"], interactive=True))
+
     current_player = gr.State("X")
-    game_status = gr.Textbox(value="Player 1's turn (X)", label="Game Status", interactive=False)
-    winner_display = gr.HTML("", elem_classes=["winner-highlight"])
-
-    # Create grid buttons
-    buttons = []
-    for i in range(3):
-        with gr.Row():
-            for j in range(3):
-                btn = gr.Button(value="", elem_classes=["cell-btn"])
-                buttons.append(btn)
-
-    # Update buttons dynamically on click
-    for idx, btn in enumerate(buttons):
-        btn.click(
-            handle_move,
-            inputs=[board_state, current_player, gr.Number(idx, visible=False), game_status, difficulty],
-            outputs=[board_state, current_player, game_status, winner_display, *buttons],
-        )
+    board_state = gr.State([[""] * 3 for _ in range(3)])
+    difficulty = gr.Dropdown(choices=["Easy", "Medium", "Hard"], value="Medium", label="AI Difficulty")
 
-    # Reset game button
     reset_button = gr.Button("Reset Game")
-    reset_button.click(
-        initialize_game,
-        inputs=[],
-        outputs=[board_state, current_player, game_status, winner_display, *buttons],
-    )
+    reset_button.click(reset_game, [], outputs=[board_state, current_player, game_status, winner_display, *button_components])
+
+    for idx, button in enumerate(button_components):
+        button.click(
+            handle_move,
+            inputs=[board_state, current_player, gr.State(idx), game_status, difficulty],
+            outputs=[board_state, current_player, game_status, winner_display, *button_components]
+        )
 
-tic_tac_toe.launch()
+app.launch()