Update app.py

app.py

@@ -359,8 +359,10 @@ CSS = """
   line-height: 1.2;
 }
 
-/* 让第三行控制更像一条 toolbar，而不是竖菜单 */
+/* Index 一行，按钮单独一行并向下留间距 */
 #toolbar .gr-row { align-items: end; }
+#toolbar-btns { margin-top: 12px; }
+#toolbar-btns .gr-row { align-items: end; }
 
 /* Gallery 更像 viewer */
 #viewer { margin-top: 10px; }
@@ -389,17 +391,19 @@ def build_ui():
                 parquet_tip = gr.HTML(value="<div id='badges'></div>")
                 summary_badge = gr.HTML(value="<div id='badges'><span class='badge'>No parquet selected</span></div>")
                 scan_info = gr.HTML(value="<div id='badges'><span class='badge'>Indexing dataset repo…</span></div>")
+                
 
             with gr.Row():
                 shape_dd = gr.Dropdown(label="Shape", choices=SHAPES, value="circle", scale=1)
                 split_dd = gr.Dropdown(label="Split", choices=SPLITS, value="test", scale=1)
                 size_dd = gr.Dropdown(label="Maze size", choices=MAZE_SIZE_CHOICES, value="All", scale=1)
-                parquet_dd = gr.Dropdown(label="Parquet", choices=[], value=None, scale=3)
+                parquet_dd = gr.Dropdown(label="Parquet", choices=[], value=None, scale=2)
+                
 
             with gr.Row(elem_id="toolbar"):
-                id_query = gr.Textbox(label="Find by id", placeholder="UUID or substring", scale=3)
-                idx_slider = gr.Slider(label="Index", minimum=0, maximum=0, value=0, step=1, scale=4)
-
+                id_query = gr.Textbox(label="Find by id", placeholder="UUID or substring", scale=2)
+                idx_slider = gr.Slider(label="Index", minimum=0, maximum=0, value=0, step=1, scale=2)
+            with gr.Row():
                 prev_btn = gr.Button("⬅ Prev", variant="secondary", scale=1)
                 next_btn = gr.Button("Next ➡", variant="secondary", scale=1)
                 random_btn = gr.Button("🎲 Random", variant="primary", scale=1)