better UI

app.py

@@ -8,7 +8,7 @@ import time
 
 import gradio as gr
 
-from backbone import extract_features
+from draft_gradio_backbone import extract_features
 from ncut_pytorch import NCUT, rgb_from_tsne_3d, rgb_from_umap_3d
 
 
@@ -138,7 +138,7 @@ def ncut_run(
     )
     logging_str += _logging_str
     rgb = dont_use_too_much_green(rgb)
-    return to_pil_images(rgb), [], logging_str
+    return to_pil_images(rgb), logging_str
 
 @spaces.GPU(duration=15)
 def quick_run(images, **kwargs):
@@ -173,7 +173,7 @@ def run_fn(
     min_dist=0.1,
 ):
     if images is None:
-        return [], example_items, "No images selected."
+        return [], "No images selected."
     
     kwargs = {
         "model_name": model_name,
@@ -217,8 +217,14 @@ with gr.Blocks() as demo:
         
             gr.Markdown('### Load Examples 👇')
             load_images_button = gr.Button("Load", elem_id="load-images-button")
-            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)
-        
+            hide_button = gr.Button("Hide", elem_id="hide-button")
+            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, elem_id="example-gallery")
+
+            hide_button.click(
+                fn=lambda: gr.update(visible=False),
+                outputs=example_gallery
+            )
+
         with gr.Column(scale=5, min_width=200):
             gr.Markdown('### Output Images')
             output_gallery = gr.Gallery(value=[], label="NCUT Embedding", show_label=False, elem_id="ncut", columns=[3], rows=[1], object_fit="contain", height="auto")
@@ -242,11 +248,18 @@ with gr.Blocks() as demo:
             logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
             
     def load_default_images():
-        return default_images, default_outputs, []
+        return default_images, default_outputs, example_items
 
     def empty_input_and_output():
         return [], [], example_items
     
+    def hide_example_gallery(images):
+        if images is None:
+            return 
+        
+        # hide the example gallery, use css to hide
+        
+    
     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, example_gallery])
     submit_button.click(
@@ -257,7 +270,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, example_gallery, logging_text]
+        outputs=[output_gallery, logging_text]
     )