add delete button

app.py

@@ -2437,13 +2437,19 @@ with demo:
                 run_inspection_button = gr.Button("🔴 RUN Inspection", elem_id="run_inspection", variant='primary')
                 inspect_logging_text = gr.Textbox("Logging information", lines=3, label="Logging", elem_id="inspect_logging", type="text", placeholder="Logging information", autofocus=False, autoscroll=False)
                 # output_slot_radio = gr.Radio([1, 2, 3], label="Output Row", value=1, elem_id="output_slot", show_label=True)
-            
+
+                delete_all_output_button = gr.Button("❌ Delete All", elem_id="delete_all_output", variant='secondary')
+                
             image_select.visible = False
             tsne_select.visible = True
             prompt_radio.change(fn=lambda x: gr.update(visible=x=="Tree"), inputs=prompt_radio, outputs=[tsne_select])
             prompt_radio.change(fn=lambda x: gr.update(visible=x=="Image"), inputs=prompt_radio, outputs=[image_select])
             
-        def make_one_output_row(i_row=1):
+        MAX_ROWS = 20
+        current_output_row = gr.State(0)
+        output_row_occupy = gr.State([False] * MAX_ROWS)
+        
+        def make_one_output_row(output_row_occupy, i_row=1):
             with gr.Row() as inspect_output_row:
                 with gr.Column(scale=5, min_width=200):
                     output_tree_image = gr.Image(label=f"spectral-tSNE tree [row#{i_row}]", elem_id="output_image", interactive=False)
@@ -2451,21 +2457,28 @@ with demo:
                     delete_button = gr.Button("❌ Delete", elem_id=f"delete_button_{i_row}", variant='secondary')
                 with gr.Column(scale=10, min_width=200):
                     heatmap_gallery = gr.Gallery(format='png', value=[], label=f"Cluster Heatmap [row#{i_row}]", show_label=True, elem_id="heatmap", columns=[6], rows=[1], object_fit="contain", height="500px", show_share_button=True, interactive=False)
-            delete_button.click(lambda: gr.update(visible=False), outputs=[inspect_output_row])
+            def delete_a_row(output_row_occupy, i_row=1):
+                # output_row_occupy[i_row-1] = False
+                return output_row_occupy, gr.update(visible=False)
+            delete_button.click(partial(delete_a_row, i_row=i_row), output_row_occupy, outputs=[output_row_occupy, inspect_output_row])
             return inspect_output_row, output_tree_image, heatmap_gallery, text_block
         
         gr.Markdown('---')
-        MAX_ROWS = 10
-        current_output_row = gr.State(MAX_ROWS-1)
+        
         inspect_output_rows, output_tree_images, heatmap_galleries, text_blocks = [], [], [], []
         for i_row in range(MAX_ROWS, 0, -1):
-            inspect_output_row, output_tree_image, heatmap_gallery, text_block = make_one_output_row(i_row)
+            inspect_output_row, output_tree_image, heatmap_gallery, text_block = make_one_output_row(output_row_occupy, i_row)
             inspect_output_row.visible = False
             inspect_output_rows.append(inspect_output_row)
             output_tree_images.append(output_tree_image)
             heatmap_galleries.append(heatmap_gallery)
             text_blocks.append(text_block)
             
+        def delete_all_output(output_row_occupy):
+            n_rows = len(output_row_occupy)
+            output_row_occupy = [False] * n_rows
+            return output_row_occupy, *[gr.update(visible=False) for _ in range(n_rows)]
+        delete_all_output_button.click(delete_all_output, inputs=[output_row_occupy], outputs=[output_row_occupy, *inspect_output_rows])
         
         def relative_xy_last_positive(prompts):
             image = prompts['image']
@@ -2522,7 +2535,7 @@ with demo:
             except:
                 raise gr.Error("""No blue point is selected. <br/>Please left-click on the image to select a blue point. <br/>After reloading the image (e.g., change granularity), please use the eraser to remove the previous point, then click on the image to select a blue point.""")
         
-        def run_inspection(tsne_prompt, image_prompt, prompt_radio, output_slot, tsne2d_embed, edges, fps_eigvecs, fps_tsne_rgb, fps_indices, granularity, eigvecs, i_image, tsne3d_rgb, input_gallery, max_rows=MAX_ROWS):
+        def run_inspection(tsne_prompt, image_prompt, prompt_radio, current_output_row, tsne2d_embed, edges, fps_eigvecs, fps_tsne_rgb, fps_indices, granularity, eigvecs, i_image, tsne3d_rgb, input_gallery, output_row_occupy, max_rows=MAX_ROWS):
             if len(tsne2d_embed) == 0:
                 raise gr.Error("Please run FPS+Cluster first.")
             closest_idx = find_closest_fps_point(prompt_radio, tsne_prompt, image_prompt, i_image, tsne2d_embed, eigvecs, fps_eigvecs)
@@ -2575,31 +2588,40 @@ with demo:
                 blend = Image.fromarray(blend.astype(np.uint8))
                 heatmap_images[i] = blend
             
+            # find the output slot
+            # search from the last row
+            found_flag = False
+            for i_slot in range(max_rows-1, -1, -1):
+                if not output_row_occupy[i_slot]:
+                    found_flag = True
+                    break
+            if not found_flag:
+                i_slot = 0
+                gr.Warning("Output slots are full, Overwriting the first row. Please use '❌ Delete All' to clear all outputs.")
+            output_row_occupy[i_slot] = True
             # tree_label = f"spectral-tSNE tree [row#{max_rows-output_slot}] k={granularity} idx={closest_idx} n={len(connected_idxs)}"
-            tree_label = f"spectral-tSNE tree [row#{max_rows-output_slot}]"
-            heatmap_label = f"Cluster Heatmap [row#{max_rows-output_slot}] k={granularity} idx={closest_idx} n={len(connected_idxs)}"
+            tree_label = f"spectral-tSNE tree [row#{current_output_row+1}]"
+            heatmap_label = f"Cluster Heatmap [row#{current_output_row+1}] k={granularity} idx={closest_idx} n={len(connected_idxs)}"
             # update the output slots
             output_rows = [gr.update() for _ in range(max_rows)]
             output_tsne_plots = [gr.update() for _ in range(max_rows)]
             output_heatmaps = [gr.update() for _ in range(max_rows)]
             output_texts = [gr.update() for _ in range(max_rows)]
-            output_rows[output_slot] = gr.update(visible=True)
-            output_tsne_plots[output_slot] = gr.update(value=output_tsne_plot, label=tree_label)
-            output_heatmaps[output_slot] = gr.update(value=heatmap_images, label=heatmap_label)
-            output_texts[output_slot] = gr.update(value=logging_text)
+            output_rows[i_slot] = gr.update(visible=True)
+            output_tsne_plots[i_slot] = gr.update(value=output_tsne_plot, label=tree_label)
+            output_heatmaps[i_slot] = gr.update(value=heatmap_images, label=heatmap_label)
+            output_texts[i_slot] = gr.update(value=logging_text)
             # gr.Info(f"Output in [row#{max_rows-output_slot}]", 3)
-            logging_text += f"\nOutput: [row#{max_rows-output_slot}]"
-            output_slot -= 1
-            if output_slot < 0:
-                output_slot = max_rows - 1
+            logging_text += f"\nOutput: [row#{current_output_row+1}]"
+            current_output_row += 1
             
-            return *output_rows, *output_tsne_plots, *output_heatmaps, *output_texts, output_slot, logging_text
+            return *output_rows, *output_tsne_plots, *output_heatmaps, *output_texts, current_output_row, output_row_occupy, logging_text
             
             
         run_inspection_button.click(
             run_inspection,
-            inputs=[tsne_prompt_image, image_plot, prompt_radio, current_output_row, tsne_2d_points, edges, fps_eigvecs, fps_tsne_rgb, fps_indices, granularity_slider, eigvecs, image_slider, tsne3d_rgb, input_gallery],
-            outputs=inspect_output_rows + output_tree_images + heatmap_galleries + text_blocks + [current_output_row, inspect_logging_text],
+            inputs=[tsne_prompt_image, image_plot, prompt_radio, current_output_row, tsne_2d_points, edges, fps_eigvecs, fps_tsne_rgb, fps_indices, granularity_slider, eigvecs, image_slider, tsne3d_rgb, input_gallery, output_row_occupy],
+            outputs=inspect_output_rows + output_tree_images + heatmap_galleries + text_blocks + [current_output_row, output_row_occupy, inspect_logging_text],
         )
         
     with gr.Tab('PlayGround (eig)', visible=True) as test_playground_tab2: