feat: place optimal latent button better

app.py

@@ -72,6 +72,7 @@ with gr.Blocks(
     .coordinate-container {
         width: 600px !important;
         height: 600px !important;
+        position: relative !important;
     }
     .coordinate-container img {
         width: 100% !important;
@@ -84,12 +85,15 @@ with gr.Blocks(
         background-color: #f8f9fa !important;
         border-radius: 8px !important;
     }
+    .button-container {
+        display: flex !important;
+        justify-content: center !important;
+        width: 600px !important;  /* Match coordinate container width */
+    }
     .optimal-button {
         margin-top: 1rem !important;
         margin-bottom: 1rem !important;
         width: 200px !important;
-        margin-left: auto !important;
-        margin-right: auto !important;
     }
 """
 ) as demo:
@@ -150,8 +154,7 @@ with gr.Blocks(
                     show_fullscreen_button=False,
                 )
 
-            # Add the Find Optimal Latent button with a container for centering
-            with gr.Column():
+            with gr.Column(elem_classes="button-container"):
                 optimal_button = gr.Button("Find Optimal Latent", elem_classes="optimal-button")
 
     # Documentation section