update ui

app.py

@@ -77,6 +77,14 @@ def to_pil_images(images):
         for image in images
     ]
 
+default_images = ['./images/image_0.jpg', './images/image_1.jpg', './images/image_2.jpg', './images/image_3.jpg', './images/image_5.jpg']
+default_outputs = ['./images/ncut_0.jpg', './images/ncut_1.jpg', './images/ncut_2.jpg', './images/ncut_3.jpg', './images/ncut_5.jpg']
+
+downscaled_images = ['./images/image_0_small.jpg', './images/image_1_small.jpg', './images/image_2_small.jpg', './images/image_3_small.jpg', './images/image_5_small.jpg']
+downscaled_outputs = ['./images/ncut_0_small.jpg', './images/ncut_1_small.jpg', './images/ncut_2_small.jpg', './images/ncut_3_small.jpg', './images/ncut_5_small.jpg']
+
+example_items = downscaled_images[:3] + downscaled_outputs[:3]
+
 @spaces.GPU(duration=30)
 def main_fn(
     images,
@@ -94,6 +102,9 @@ def main_fn(
     n_neighbors=500,
     min_dist=0.1,
 ):
+    if len(images) == 0:
+        return [], example_items
+    
     if perplexity >= num_sample_tsne or n_neighbors >= num_sample_tsne:
         # raise gr.Error("Perplexity must be less than the number of samples for t-SNE.")
         gr.Warning("Perplexity/n_neighbors must be less than the number of samples.\n" f"Setting to {num_sample_tsne-1}.")
@@ -125,13 +136,8 @@ def main_fn(
         min_dist=min_dist,
     )
     rgb = dont_use_too_much_green(rgb)
-    return to_pil_images(rgb)
-
-default_images = ['./images/image_0.jpg', './images/image_1.jpg', './images/image_2.jpg', './images/image_3.jpg', './images/image_5.jpg']
-default_outputs = ['./images/ncut_0.jpg', './images/ncut_1.jpg', './images/ncut_2.jpg', './images/ncut_3.jpg', './images/ncut_5.jpg']
+    return to_pil_images(rgb), []
 
-downscaled_images = ['./images/image_0_small.jpg', './images/image_1_small.jpg', './images/image_2_small.jpg', './images/image_3_small.jpg', './images/image_5_small.jpg']
-downscaled_outputs = ['./images/ncut_0_small.jpg', './images/ncut_1_small.jpg', './images/ncut_2_small.jpg', './images/ncut_3_small.jpg', './images/ncut_5_small.jpg']
 
 with gr.Blocks() as demo:
 
@@ -144,7 +150,7 @@ with gr.Blocks() as demo:
         
             gr.Markdown('### Load Examples 👇')
             load_images_button = gr.Button("Load", elem_id="load-images-button")
-            gr.Gallery(value=downscaled_images[:3] + downscaled_outputs[:3], label="Example Set A", show_label=False, columns=[3], rows=[2], object_fit="scale-down", height="200px", show_share_button=False)
+            example_gallery = gr.Gallery(value=example_items, label="Example Set A", show_label=False, columns=[3], rows=[2], object_fit="scale-down", height="200px", show_share_button=False)
         
         with gr.Column(scale=5, min_width=200):
             gr.Markdown('### Output Images')
@@ -166,12 +172,12 @@ with gr.Blocks() as demo:
                 min_dist_slider = gr.Slider(0.1, 1, step=0.1, label="min_dist (UMAP)", value=0.1, elem_id="min_dist")
 
     def load_default_images():
-        return default_images, default_outputs
+        return default_images, default_outputs, []
 
     def empty_input_and_output():
         return [], []
     
-    load_images_button.click(load_default_images, outputs=[input_gallery, output_gallery])
+    load_images_button.click(load_default_images, outputs=[input_gallery, output_gallery, example_gallery])
     clear_images_button.click(empty_input_and_output, outputs=[input_gallery, output_gallery])
     submit_button.click(
         main_fn,
@@ -181,7 +187,7 @@ with gr.Blocks() as demo:
             embedding_method_dropdown, num_sample_tsne_slider, knn_tsne_slider, 
             perplexity_slider, n_neighbors_slider, min_dist_slider
         ],
-        outputs=output_gallery
+        outputs=[output_gallery, example_gallery]
     )