update UI

app.py

@@ -484,14 +484,13 @@ def make_dataset_images_section(advanced=False):
     with gr.Column() as advanced_block:
         dataset_names = DATASET_NAMES
         dataset_classes = DATASET_CLASSES
-        dataset_dropdown = gr.Dropdown(dataset_names, label="Dataset name", value="mrm8488/ImageNet1K-val", elem_id="dataset")
-        num_images_slider = gr.Number(10, label="Number of images", elem_id="num_images")
-        # with gr.Row():
-        filter_by_class_checkbox = gr.Checkbox(label="Filter by class", value=True, elem_id="filter_by_class_checkbox")
-        filter_by_class_text = gr.Textbox(label="Class to select", value="0,33,99", elem_id="filter_by_class_text", info=f"e.g. `0,1,2`. (1000 classes)", visible=True)
-        # with gr.Row():
-        is_random_checkbox = gr.Checkbox(label="Random shuffle", value=False, elem_id="random_seed_checkbox")
-        random_seed_slider = gr.Slider(0, 1000, step=1, label="Random seed", value=1, elem_id="random_seed", visible=False)
+        with gr.Row():
+            dataset_dropdown = gr.Dropdown(dataset_names, label="Dataset name", value="mrm8488/ImageNet1K-val", elem_id="dataset", min_width=300)
+            num_images_slider = gr.Number(10, label="Number of images", elem_id="num_images")
+            filter_by_class_checkbox = gr.Checkbox(label="Filter by class", value=True, elem_id="filter_by_class_checkbox")
+            filter_by_class_text = gr.Textbox(label="Class to select", value="0,33,99", elem_id="filter_by_class_text", info=f"e.g. `0,1,2`. (1000 classes)", visible=True)
+            is_random_checkbox = gr.Checkbox(label="Random shuffle", value=False, elem_id="random_seed_checkbox")
+            random_seed_slider = gr.Slider(0, 1000, step=1, label="Random seed", value=1, elem_id="random_seed", visible=False)
     
     if advanced:
         advanced_block.visible = True
@@ -519,7 +518,7 @@ def make_dataset_images_section(advanced=False):
         idx = dataset_names.index(dataset_name)
         num_classes = dataset_classes[idx]
         return gr.Textbox(label="Class to select", value="0,1,2", elem_id="filter_by_class_text", info=f"e.g. `0,1,2`. ({num_classes} classes)", visible=is_filter)
-    filter_by_class_checkbox.change(fn=change_filter_by_class, inputs=[filter_by_class_checkbox, dataset_dropdown], outputs=filter_by_class_text)
+    # filter_by_class_checkbox.change(fn=change_filter_by_class, inputs=[filter_by_class_checkbox, dataset_dropdown], outputs=filter_by_class_text)
     
     def change_random_seed(is_random):
         return gr.Slider(0, 1000, step=1, label="Random seed", value=1, elem_id="random_seed", visible=is_random)
@@ -727,8 +726,9 @@ with demo:
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
                 input_gallery, submit_button, clear_images_button = make_input_images_section()
-                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section()
+                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
                 num_images_slider.value = 100
+                clear_images_button.visible = False
                 
             with gr.Column(scale=5, min_width=200):
                 with gr.Accordion("➡️ Recursion config", open=True):
@@ -769,7 +769,7 @@ with demo:
         false_placeholder.visible = False
         number_placeholder = gr.Number(0, label="Number placeholder", elem_id="number_placeholder")
         number_placeholder.visible = False
-        clear_images_button.click(lambda x: ([], [], [], []), outputs=[input_gallery, l1_gallery, l2_gallery, l3_gallery])
+        clear_images_button.click(lambda x: ([],), outputs=[input_gallery])
         submit_button.click(
             run_fn,
             inputs=[