update msg

app.py

@@ -2465,7 +2465,7 @@ with demo:
                     text_block = gr.Textbox("", label="Logging", elem_id=f"logging_{i_row}", type="text", placeholder="Logging information", autofocus=False, autoscroll=False, lines=2, show_label=False)
                     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)
+                    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="550px", show_share_button=True, interactive=False)
             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)
@@ -2513,6 +2513,7 @@ with demo:
         
         def find_closest_fps_point_for_tsne_tree_plot(tsne_prompt, tsne2d_embed):
             x, y = relative_xy_last_positive(tsne_prompt)
+            _x_ratio, _y_ratio = x, y
             x_vmax = tsne2d_embed[:, 0].max() * 1.1
             x_vmin = tsne2d_embed[:, 0].min() * 1.1
             y_vmax = tsne2d_embed[:, 1].max() * 1.1
@@ -2522,10 +2523,11 @@ with demo:
             y = y * (y_vmax - y_vmin) + y_vmin
             dist = np.linalg.norm(tsne2d_embed - np.array([x, y]), axis=1)
             closest_idx = np.argmin(dist)
-            return closest_idx
+            return closest_idx, (_x_ratio, _y_ratio)
         
         def find_closest_fps_point_for_image_prompt(image_prompt, i_image, eigvecs, fps_eigvecs):
             x, y = relative_xy_last_positive(image_prompt)
+            _x_ratio, _y_ratio = x, y
             _eigvec = eigvecs[i_image]
             h, w = _eigvec.shape[:2]
             x = int(x * w)
@@ -2533,7 +2535,7 @@ with demo:
             eigvec = _eigvec[y, x]
             sim = fps_eigvecs @ eigvec
             closest_idx = np.argmax(sim)
-            return closest_idx
+            return closest_idx, (_x_ratio, _y_ratio)
         
         def find_closest_fps_point(prompt_radio, tsne_prompt, image_prompt, i_image, tsne2d_embed, eigvecs, fps_eigvecs):
             try:
@@ -2547,15 +2549,15 @@ with demo:
         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)
+            closest_idx, (_x, _y) = find_closest_fps_point(prompt_radio, tsne_prompt, image_prompt, i_image, tsne2d_embed, eigvecs, fps_eigvecs)
             closest_rgb = fps_tsne_rgb[closest_idx]
             closest_rgb = (closest_rgb * 255).astype(np.uint8)
             
             from fps_cluster import find_connected_component
             connected_idxs = find_connected_component(edges[granularity:], closest_idx)
             
-            logging_text = f"Clicked: idx={closest_idx}, RGB: {closest_rgb.tolist()}\n"
-            logging_text += f"Granularity: k={granularity}, Connected: n={len(connected_idxs)}"
+            logging_text = f"Clicked: idx={closest_idx}, xy=[{_x:.2f}, {_y:.2f}], RGB={closest_rgb}"
+            logging_text += f"\nGranularity: k={granularity}, Connected: n={len(connected_idxs)}"
             
             output_tsne_plot = plot_tsne_tree(tsne2d_embed, edges, fps_tsne_rgb, granularity, closest_idx, highlight_connections=True)
             
@@ -2610,7 +2612,7 @@ with demo:
             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#{current_output_row+1}]"
-            heatmap_label = f"Cluster Heatmap [row#{current_output_row+1}] k={granularity} idx={closest_idx} n={len(connected_idxs)}"
+            heatmap_label = f"Cluster Heatmap [row#{current_output_row+1}] k={granularity} n={len(connected_idxs)} xy=[{_x:.2f}, {_y:.2f}] idx={closest_idx}"
             # update the output slots
             output_rows = [gr.update() for _ in range(max_rows)]
             output_tsne_plots = [gr.update() for _ in range(max_rows)]