Upload app.py

app.py

@@ -443,10 +443,12 @@ def update_all_points(visible_points, occlusion_points):
     # 返回更新字典来明确设置 choices 和 value
     return text, gr.Dropdown(label="Select Visible Point to Delete", choices=visible_dropdown_choices, value=None, interactive=True), gr.Dropdown(label="Select Occlusion Point to Delete", choices=occlusion_dropdown_choices, value=None, interactive=True)
 
-def delete_selected_visible(image, visible_points, occlusion_points, selected_index):
-    """
-    删除 visible 点列表中指定索引的点，并更新图像和显示信息。
-    """
+def delete_selected_visible(image, visible_points, occlusion_points, selected_value):
+    # selected_value 是类似 "(x, y)" 的字符串
+    try:
+        selected_index = [f"({p[0]}, {p[1]})" for p in visible_points].index(selected_value)
+    except ValueError:
+        selected_index = None
     if selected_index is not None and 0 <= selected_index < len(visible_points):
         visible_points.pop(selected_index)
     updated_image = image.copy()
@@ -456,10 +458,11 @@ def delete_selected_visible(image, visible_points, occlusion_points, selected_in
     updated_text, vis_dropdown, occ_dropdown = update_all_points(visible_points, occlusion_points)
     return updated_image, visible_points, occlusion_points, updated_text, vis_dropdown, occ_dropdown
 
-def delete_selected_occlusion(image, visible_points, occlusion_points, selected_index):
-    """
-    删除 occlusion 点列表中指定索引的点，并更新图像和显示信息。
-    """
+def delete_selected_occlusion(image, visible_points, occlusion_points, selected_value):
+    try:
+        selected_index = [f"({p[0]}, {p[1]})" for p in occlusion_points].index(selected_value)
+    except ValueError:
+        selected_index = None
     if selected_index is not None and 0 <= selected_index < len(occlusion_points):
         occlusion_points.pop(selected_index)
     updated_image = image.copy()