add image profile

app.py

@@ -1,8 +1,10 @@
 # Author: Huzheng Yang
 # %%
 import copy
+import pickle
 from functools import partial
 from io import BytesIO
+import json
 import os
 import uuid
 
@@ -1152,14 +1154,6 @@ def run_fn(
         raise gr.Error(message, duration=0)
 
 
-
-def make_input_images_section(rows=1, cols=3, height="auto"):
-    gr.Markdown('### Input Images')
-    input_gallery = gr.Gallery(value=None, label="Select images", show_label=True, elem_id="images", columns=[cols], rows=[rows], object_fit="contain", height=height, type="pil", show_share_button=False)
-    submit_button = gr.Button("🔴 RUN", elem_id="submit_button", variant='primary')
-    clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button', variant='stop')
-    return input_gallery, submit_button, clear_images_button
-
 def make_input_video_section():
     # gr.Markdown('### Input Video')
     input_gallery = gr.Video(value=None, label="Select video", elem_id="video-input", height="auto", show_share_button=False, interactive=True)
@@ -1170,8 +1164,33 @@ def make_input_video_section():
     clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button', variant='stop')
     return input_gallery, submit_button, clear_images_button, max_frames_number
 
-def make_dataset_images_section(advanced=False, is_random=False):
 
+def make_input_images_section(rows=1, cols=3, height="auto", advanced=False, is_random=False):
+    gr.Markdown('### Input Images')
+    input_gallery = gr.Gallery(value=None, label="Input images", show_label=True, elem_id="input_images", columns=[cols], rows=[rows], object_fit="contain", height=height, type="pil", show_share_button=False)
+    
+    submit_button = gr.Button("🔴 RUN", elem_id="submit_button", variant='primary')
+    with gr.Row():
+        clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button', variant='stop')
+        clear_images_button.click(fn=lambda: gr.update(value=None), outputs=[input_gallery])
+        upload_button = gr.UploadButton(elem_id="upload_button", label="⬆️Upload", variant='secondary', file_types=["image"], file_count="multiple")
+        
+        def convert_to_pil_and_append(images, new_images):
+            if images is None:
+                images = []
+            if new_images is None:
+                return images
+            if isinstance(new_images, Image.Image):
+                images.append(new_images)
+            if isinstance(new_images, list):
+                images += [Image.open(new_image) for new_image in new_images]
+            if isinstance(new_images, str):
+                images.append(Image.open(new_images))
+            new_images = None
+            return images
+        upload_button.upload(convert_to_pil_and_append, inputs=[input_gallery, upload_button], outputs=[input_gallery])
+        
+    
     gr.Markdown('### Load Datasets')
     load_images_button = gr.Button("🔴 Load Images", elem_id="load-images-button", variant='primary')
     advanced_radio = gr.Radio(["Basic", "Advanced"], label="Datasets", value="Advanced" if advanced else "Basic", elem_id="advanced-radio", show_label=True)
@@ -1204,7 +1223,67 @@ def make_dataset_images_section(advanced=False, is_random=False):
                 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=False)
                 is_random_checkbox = gr.Checkbox(label="Random shuffle", value=True, elem_id="random_seed_checkbox")
                 random_seed_slider = gr.Slider(0, 1000, step=1, label="Random seed", value=42, elem_id="random_seed", visible=True)
+        
+        # add functionality, save and load images to profile
+        with gr.Accordion("Saved Image Profiles", open=False) as profile_accordion:
+            with gr.Row():
+                profile_text = gr.Textbox(label="Profile name", placeholder="Type here: Profile name to save/load/delete", elem_id="profile-name", scale=6, show_label=False)
+                list_profiles_button = gr.Button("📋 List", elem_id="list-profile-button", variant='secondary', scale=3)
+            with gr.Row():
+                save_profile_button = gr.Button("💾 Save", elem_id="save-profile-button", variant='secondary')
+                load_profile_button = gr.Button("📂 Load", elem_id="load-profile-button", variant='secondary')
+                delete_profile_button = gr.Button("🗑️ Delete", elem_id="delete-profile-button", variant='secondary')
+    
+            class OnDiskProfiles:
+                def __init__(self, profile_dir="demo_profiles"):
+                    if not os.path.exists(profile_dir):
+                        os.makedirs(profile_dir)
+                    self.profile_dir = profile_dir
+                
+                def list_profiles(self):
+                    profiles = os.listdir(self.profile_dir)
+                    # remove hidden files
+                    profiles = [p for p in profiles if not p.startswith(".")]
+                    if len(profiles) == 0:
+                        return "No profiles found."
+                    profile_text = "</br>".join(profiles)
+                    n_files = len(profiles)
+                    profile_text = f"Number of profiles: {n_files}</br>---------</br>" + profile_text
+                    return profile_text
+                
+                def save_profile(self, profile_name, images):
+                    profile_path = os.path.join(self.profile_dir, profile_name)
+                    if os.path.exists(profile_path):
+                        raise gr.Error(f"Profile {profile_name} already exists.")
+                    with open(profile_path, "wb") as f:
+                        pickle.dump(images, f)
+                    gr.Info(f"Profile {profile_name} saved.")
+                    return profile_path
                 
+                def load_profile(self, profile_name):
+                    profile_path = os.path.join(self.profile_dir, profile_name)
+                    if not os.path.exists(profile_path):
+                        raise gr.Error(f"Profile {profile_name} not found.")
+                    with open(profile_path, "rb") as f:
+                        images = pickle.load(f)
+                    gr.Info(f"Profile {profile_name} loaded.")
+                    return images
+                
+                def delete_profile(self, profile_name):
+                    profile_path = os.path.join(self.profile_dir, profile_name)
+                    os.remove(profile_path)
+                    gr.Info(f"Profile {profile_name} deleted.")
+                    return profile_path
+            
+            home_dir = os.path.expanduser("~")
+            defualt_dir = os.path.join(home_dir, ".cache")
+            cache_dir = os.environ.get("DEMO_PROFILE_CACHE_DIR", defualt_dir)
+            cache_dir = os.path.join(cache_dir, "demo_profiles")
+            on_disk_profiles = OnDiskProfiles(cache_dir)
+            save_profile_button.click(fn=lambda name, images: on_disk_profiles.save_profile(name, images), inputs=[profile_text, input_gallery])
+            load_profile_button.click(fn=lambda name: gr.update(value=on_disk_profiles.load_profile(name)), inputs=profile_text, outputs=[input_gallery])
+            delete_profile_button.click(fn=lambda name: on_disk_profiles.delete_profile(name), inputs=profile_text)
+            list_profiles_button.click(fn=lambda: gr.Info(on_disk_profiles.list_profiles(), duration=0))
     
     if advanced:
         advanced_block.visible = True
@@ -1259,8 +1338,7 @@ def make_dataset_images_section(advanced=False, is_random=False):
             key = list(dataset.keys())[0]
             dataset = dataset[key]
         except Exception as e:
-            gr.Error(f"Error loading dataset {dataset_name}: {e}")
-            return None
+            raise gr.Error(f"Error loading dataset {dataset_name}: {e}")
         if num_images > len(dataset):
             num_images = len(dataset)
         
@@ -1310,15 +1388,188 @@ def make_dataset_images_section(advanced=False, is_random=False):
                 return img
             images = [center_crop_image(image) for image in images]
         
-        return images   
-    
-    load_images_button.click(load_dataset_images, 
-                        inputs=[advanced_radio, dataset_dropdown, num_images_slider,
+        return images 
+    
+    def load_and_append(existing_images, *args, **kwargs):
+        new_images = load_dataset_images(*args, **kwargs)
+        if new_images is None:
+            return existing_images
+        if len(new_images) == 0:
+            return existing_images
+        if existing_images is None:
+            existing_images = []
+        existing_images += new_images
+        return existing_images
+    
+    load_images_button.click(load_and_append, 
+                        inputs=[input_gallery, advanced_radio, dataset_dropdown, num_images_slider,
                                 filter_by_class_checkbox, filter_by_class_text, 
                                 is_random_checkbox, random_seed_slider],
                         outputs=[input_gallery])
     
-    return dataset_dropdown, num_images_slider, random_seed_slider, load_images_button
+    
+    
+    return input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button
+    
+
+
+# def make_input_images_section(rows=1, cols=3, height="auto"):
+#     gr.Markdown('### Input Images')
+#     input_gallery = gr.Gallery(value=None, label="Select images", show_label=True, elem_id="images", columns=[cols], rows=[rows], object_fit="contain", height=height, type="pil", show_share_button=False)
+#     submit_button = gr.Button("🔴 RUN", elem_id="submit_button", variant='primary')
+#     clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button', variant='stop')
+#     return input_gallery, submit_button, clear_images_button
+
+
+# def make_dataset_images_section(advanced=False, is_random=False):
+
+#     gr.Markdown('### Load Datasets')
+#     load_images_button = gr.Button("🔴 Load Images", elem_id="load-images-button", variant='primary')
+#     advanced_radio = gr.Radio(["Basic", "Advanced"], label="Datasets", value="Advanced" if advanced else "Basic", elem_id="advanced-radio", show_label=True)
+#     with gr.Column() as basic_block:
+#         example_gallery = gr.Gallery(value=example_items, label="Example Images", show_label=True, columns=[3], rows=[2], object_fit="scale-down", height="200px", show_share_button=False, elem_id="example-gallery")
+#     with gr.Column() as advanced_block:
+#         # dataset_names = DATASET_NAMES
+#         # dataset_classes = DATASET_CLASSES
+#         dataset_categories = list(DATASETS.keys())
+#         defualt_cat = dataset_categories[0]
+#         def get_choices(cat):
+#             return [tup[0] for tup in DATASETS[cat]]
+#         defualt_choices = get_choices(defualt_cat)
+#         with gr.Row():
+#             dataset_radio = gr.Radio(dataset_categories, label="Dataset Category", value=defualt_cat, elem_id="dataset-radio", show_label=True, min_width=600)
+#             # dataset_dropdown = gr.Dropdown(dataset_names, label="Dataset name", value="mrm8488/ImageNet1K-val", elem_id="dataset", min_width=300)
+#             dataset_dropdown = gr.Dropdown(defualt_choices, label="Dataset name", value=defualt_choices[0], elem_id="dataset", min_width=400)
+#             dataset_radio.change(fn=lambda x: gr.update(choices=get_choices(x), value=get_choices(x)[0]), inputs=dataset_radio, outputs=dataset_dropdown)
+#             # num_images_slider = gr.Number(10, label="Number of images", elem_id="num_images")
+#             num_images_slider = gr.Slider(1, 1000, step=1, label="Number of images", value=10, elem_id="num_images", min_width=200)
+#             if not is_random:
+#                 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)
+#                 is_random_checkbox = gr.Checkbox(label="Random shuffle", value=True, elem_id="random_seed_checkbox")
+#                 random_seed_slider = gr.Slider(0, 1000, step=1, label="Random seed", value=1, elem_id="random_seed", visible=True)
+#             if is_random:
+#                 filter_by_class_checkbox = gr.Checkbox(label="Filter by class", value=False, 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=False)
+#                 is_random_checkbox = gr.Checkbox(label="Random shuffle", value=True, elem_id="random_seed_checkbox")
+#                 random_seed_slider = gr.Slider(0, 1000, step=1, label="Random seed", value=42, elem_id="random_seed", visible=True)
+                
+    
+#     if advanced:
+#         advanced_block.visible = True
+#         basic_block.visible = False
+#     else:
+#         advanced_block.visible = False
+#         basic_block.visible = True
+        
+#     # change visibility 
+#     advanced_radio.change(fn=lambda x: gr.update(visible=x=="Advanced"), inputs=advanced_radio, outputs=[advanced_block])
+#     advanced_radio.change(fn=lambda x: gr.update(visible=x=="Basic"), inputs=advanced_radio, outputs=[basic_block])
+    
+#     def find_num_classes(dataset_name):
+#         num_classes = None
+#         for cat, datasets in DATASETS.items():
+#             datasets = [tup[0] for tup in datasets]
+#             if dataset_name in datasets:
+#                 num_classes = DATASETS[cat][datasets.index(dataset_name)][1]
+#                 break
+#         return num_classes
+    
+#     def change_filter_options(dataset_name):
+#         num_classes = find_num_classes(dataset_name)
+#         if num_classes is None:
+#             return (gr.Checkbox(label="Filter by class", value=False, elem_id="filter_by_class_checkbox", visible=False),
+#                     gr.Textbox(label="Class to select", value="0,1,2", elem_id="filter_by_class_text", info="e.g. `0,1,2`. This dataset has no class label", visible=False))
+#         return (gr.Checkbox(label="Filter by class", value=True, elem_id="filter_by_class_checkbox", visible=True),
+#                 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=True))
+#     dataset_dropdown.change(fn=change_filter_options, inputs=dataset_dropdown, outputs=[filter_by_class_checkbox, filter_by_class_text])
+    
+#     def change_filter_by_class(is_filter, dataset_name):
+#         num_classes = find_num_classes(dataset_name)
+#         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)
+    
+#     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)
+#     is_random_checkbox.change(fn=change_random_seed, inputs=is_random_checkbox, outputs=random_seed_slider)
+    
+        
+#     def load_dataset_images(is_advanced, dataset_name, num_images=10, 
+#                             is_filter=True, filter_by_class_text="0,1,2", 
+#                             is_random=False, seed=1):
+#         progress = gr.Progress()
+#         progress(0, desc="Loading Images")
+#         if is_advanced == "Basic":
+#             gr.Info("Loaded images from Ego-Exo4D")
+#             return default_images
+#         try:
+#             progress(0.5, desc="Downloading Dataset")
+#             dataset = load_dataset(dataset_name, trust_remote_code=True)
+#             key = list(dataset.keys())[0]
+#             dataset = dataset[key]
+#         except Exception as e:
+#             gr.Error(f"Error loading dataset {dataset_name}: {e}")
+#             return None
+#         if num_images > len(dataset):
+#             num_images = len(dataset)
+        
+#         if is_filter:
+#             progress(0.8, desc="Filtering Images")
+#             classes = [int(i) for i in filter_by_class_text.split(",")]
+#             labels = np.array(dataset['label'])
+#             unique_labels = np.unique(labels)
+#             valid_classes = [i for i in classes if i in unique_labels]
+#             invalid_classes = [i for i in classes if i not in unique_labels]
+#             if len(invalid_classes) > 0:
+#                 gr.Warning(f"Classes {invalid_classes} not found in the dataset.")
+#             if len(valid_classes) == 0:
+#                 gr.Error(f"Classes {classes} not found in the dataset.")
+#                 return None
+#             # shuffle each class
+#             chunk_size = num_images // len(valid_classes)
+#             image_idx = []
+#             for i in valid_classes:
+#                 idx = np.where(labels == i)[0]
+#                 if is_random:
+#                     idx = np.random.RandomState(seed).choice(idx, chunk_size, replace=False)
+#                 else:
+#                     idx = idx[:chunk_size]
+#                 image_idx.extend(idx.tolist())
+#         if not is_filter:
+#             if is_random:
+#                 image_idx = np.random.RandomState(seed).choice(len(dataset), num_images, replace=False).tolist()
+#             else:
+#                 image_idx = list(range(num_images))
+#         key = 'image' if 'image' in dataset[0] else list(dataset[0].keys())[0]
+#         images = [dataset[i][key] for i in image_idx]
+#         gr.Info(f"Loaded {len(images)} images from {dataset_name}")
+#         del dataset
+        
+#         if dataset_name in CENTER_CROP_DATASETS:
+#             def center_crop_image(img):
+#                 # image: PIL image
+#                 w, h = img.size
+#                 min_hw = min(h, w)
+#                 # center crop
+#                 left = (w - min_hw) // 2
+#                 top = (h - min_hw) // 2
+#                 right = left + min_hw
+#                 bottom = top + min_hw
+#                 img = img.crop((left, top, right, bottom))
+#                 return img
+#             images = [center_crop_image(image) for image in images]
+        
+#         return images   
+    
+#     load_images_button.click(load_dataset_images, 
+#                         inputs=[advanced_radio, dataset_dropdown, num_images_slider,
+#                                 filter_by_class_checkbox, filter_by_class_text, 
+#                                 is_random_checkbox, random_seed_slider],
+#                         outputs=[input_gallery])
+    
+#     return dataset_dropdown, num_images_slider, random_seed_slider, load_images_button
     
 
 # def random_rotate_rgb_gallery(images):
@@ -1501,8 +1752,7 @@ 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()
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 num_images_slider.value = 30
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information", autofocus=False, autoscroll=False)
                 
@@ -1516,10 +1766,7 @@ with demo:
                     perplexity_slider, n_neighbors_slider, min_dist_slider, 
                     sampling_method_dropdown, ncut_metric_dropdown, positive_prompt, negative_prompt
                 ] = make_parameters_section()
-                num_eig_slider.value = 30
 
-        clear_images_button.click(lambda x: ([], [], []), outputs=[input_gallery, output_gallery, cluster_gallery])
-        
         false_placeholder = gr.Checkbox(label="False", value=False, elem_id="false_placeholder", visible=False)
         no_prompt = gr.Textbox("", label="", elem_id="empty_placeholder", type="text", placeholder="", visible=False)
         
@@ -1542,8 +1789,7 @@ 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()
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 num_images_slider.value = 30
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information", autofocus=False, autoscroll=False, lines=20)
                 
@@ -1559,8 +1805,6 @@ with demo:
                     sampling_method_dropdown, ncut_metric_dropdown, positive_prompt, negative_prompt
                 ] = make_parameters_section()
                 num_eig_slider.value = 30
-
-        clear_images_button.click(lambda x: ([], [], [], []), outputs=[input_gallery, output_gallery, cluster_gallery, norm_gallery])
         
         false_placeholder = gr.Checkbox(label="False", value=False, elem_id="false_placeholder", visible=False)
         no_prompt = gr.Textbox("", label="", elem_id="empty_placeholder", type="text", placeholder="", visible=False)
@@ -1604,8 +1848,7 @@ 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()
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 
             with gr.Column(scale=5, min_width=200):
                 output_gallery = make_output_images_section()
@@ -1624,7 +1867,6 @@ with demo:
                 # logging text box
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
             
-        clear_images_button.click(lambda x: ([], []), outputs=[input_gallery, output_gallery])
         false_placeholder = gr.Checkbox(label="False", value=False, elem_id="false_placeholder", visible=False)
         no_prompt = gr.Textbox("", label="", elem_id="empty_placeholder", type="text", placeholder="", visible=False)
         
@@ -1651,8 +1893,7 @@ 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(advanced=True)
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 num_images_slider.value = 100
                 clear_images_button.visible = False
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
@@ -1695,7 +1936,6 @@ 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])
         no_prompt = gr.Textbox("", label="", elem_id="empty_placeholder", type="text", placeholder="", visible=False)
         
         submit_button.click(
@@ -1719,8 +1959,7 @@ with demo:
 
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button = make_input_images_section(rows=5, cols=3, height='auto')
-                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 num_images_slider.value = 100
                 clear_images_button.visible = False
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information", lines=20)
@@ -1769,7 +2008,6 @@ 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])
         no_prompt = gr.Textbox("", label="", elem_id="empty_placeholder", type="text", placeholder="", visible=False)
         
         submit_button.click(
@@ -1859,9 +2097,7 @@ 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(advanced=False)
-                clear_images_button.click(lambda x: ([], [], [], []), outputs=[input_gallery, l1_gallery, l2_gallery, l3_gallery])
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 
             with gr.Column(scale=5, min_width=200):
                 [
@@ -1898,8 +2134,7 @@ with demo:
         gr.Markdown('---')
         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(advanced=True, is_random=True)
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 num_images_slider.value = 100
                 
                 
@@ -1926,9 +2161,7 @@ with demo:
                 num_sample_tsne_slider.value = 1000
                 # logging text box
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
-                
-        clear_images_button.click(lambda x: ([], []), outputs=[input_gallery, output_gallery])
-        
+                        
         false_placeholder = gr.Checkbox(label="False", value=False, elem_id="false_placeholder", visible=False)
         no_prompt = gr.Textbox("", label="", elem_id="empty_placeholder", type="text", placeholder="", visible=False)
         
@@ -1980,8 +2213,7 @@ 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(advanced=True, is_random=True)
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 num_images_slider.value = 100
                 
                 
@@ -2014,9 +2246,7 @@ with demo:
                 num_sample_tsne_slider.value = 1000
                 # logging text box
                 logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
-                
-        clear_images_button.click(lambda x: ([],), outputs=[input_gallery])
-        
+                        
         true_placeholder = gr.Checkbox(label="True placeholder", value=True, elem_id="true_placeholder")
         true_placeholder.visible = False
         false_placeholder = gr.Checkbox(label="False placeholder", value=False, elem_id="false_placeholder")
@@ -2078,10 +2308,8 @@ with demo:
 
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button = make_input_images_section()
-                clear_images_button.click(lambda x: [], outputs=[input_gallery])
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 submit_button.visible = False
-                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
 
                 
             for i in range(3):
@@ -2153,10 +2381,8 @@ with demo:
 
         with gr.Row():
             with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button = make_input_images_section()
-                clear_images_button.click(lambda x: [], outputs=[input_gallery])
+                input_gallery, submit_button, clear_images_button, dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_input_images_section()
                 submit_button.visible = False
-                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
 
                 
             for i in range(3):