Update app.py

app.py

@@ -52,24 +52,28 @@ def play_game(board, col, state):
     if state["game_over"]:
         return board, "Game is over! Click 'New Game' to play again."
     
+    # Convert Dataframe to numpy array for game logic
+    board_array = np.array([[0 if cell == "⚪" else 1 if cell == "🔴" else 2 for cell in row] 
+                           for row in board.values.tolist()])
+    
     # Player move
-    board = make_move(board, col, 1)
-    if check_winner(board):
+    board_array = make_move(board_array, col, 1)
+    if check_winner(board_array):
         state["game_over"] = True
-        return format_board(board), "You win! 🎉"
+        return format_board(board_array), "You win! 🎉"
     
     # AI move
     config = Configuration({"rows": 6, "columns": 7, "inarow": 4})
-    obs = Observation({"board": board.flatten().tolist(), "mark": 2})
+    obs = Observation({"board": board_array.flatten().tolist(), "mark": 2})
     
     ai_col = my_agent(obs, config)
-    board = make_move(board, ai_col, 2)
+    board_array = make_move(board_array, ai_col, 2)
     
-    if check_winner(board):
+    if check_winner(board_array):
         state["game_over"] = True
-        return format_board(board), "AI wins! 🤖"
+        return format_board(board_array), "AI wins! 🤖"
     
-    return format_board(board), "Your turn!"
+    return format_board(board_array), "Your turn!"
 
 def create_ui():
     with gr.Blocks(css=css) as demo:
@@ -78,7 +82,11 @@ def create_ui():
         
         state = gr.State({"game_over": False})
         
-        # Using Grid instead of DataFrame
+        with gr.Row():
+            # Add column buttons at the top
+            buttons = [gr.Button(str(i)) for i in range(7)]
+        
+        # Board display below the buttons
         board = gr.Dataframe(
             value=format_board(initialize_game()),
             interactive=False,
@@ -89,10 +97,6 @@ def create_ui():
         )
         
         message = gr.Textbox(value="Your turn!", label="Status")
-        
-        with gr.Row():
-            buttons = [gr.Button(str(i)) for i in range(7)]
-        
         new_game = gr.Button("New Game")
         
         def reset_game():