Update app.py

app.py

@@ -103,9 +103,12 @@ css = """
     height: 40px;
     vertical-align: middle;
 }
-.button-row {
+.button-container {
     max-width: 400px;
     margin: 0 auto 20px auto;
+    padding: 10px;
+}
+.button-row {
     display: flex;
     justify-content: space-between;
     gap: 10px;
@@ -124,7 +127,8 @@ def create_ui():
         
         state = gr.State({"game_over": False})
         
-        with gr.Box(elem_classes="button-container"):
+        # Simplified button container
+        with gr.Row(elem_classes="button-container"):
             with gr.Row(elem_classes="button-row"):
                 buttons = [gr.Button(str(i), size="sm") for i in range(BOARD_SIZE[1])]
         
@@ -137,8 +141,7 @@ def create_ui():
             elem_id="board",
             row_count=BOARD_SIZE[0],
             col_count=BOARD_SIZE[1],
-            max_rows=BOARD_SIZE[0],
-            height=300
+            max_rows=BOARD_SIZE[0]
         )
         
         message = gr.Textbox(value="Your turn!", label="Status")