push app

app.py

@@ -85,10 +85,6 @@ if picture is not None:
     st.session_state.img = img  # save the processed image
     # st.session_state.coords_list = []
 
-# Ensure we have a widget_key in session state
-if "widget_key" not in st.session_state:
-    st.session_state.widget_key = 0
-
 if "img" in st.session_state:
     img = st.session_state.img
 
@@ -104,27 +100,22 @@ if "img" in st.session_state:
         draw.ellipse((cx - 5, cy - 5, cx + 5, cy + 5), fill="red")
 
     # Use the interactive component as the display canvas, showing the image with all dots.
-    new_coord = streamlit_image_coordinates(img_with_dots, key=f"click_img_{st.session_state.widget_key}")
+    new_coord = streamlit_image_coordinates(img_with_dots, key="click_img")
 
     # If a new coordinate is received and it's not already in our list, add it and force a rerun.
-    if new_coord:
-        is_close = False
-        to_remove = None
+    if new_coord and new_coord not in st.session_state.coords_list:
+        st.session_state.is_close = False
         for coord in st.session_state.coords_list:
             existing = np.array([coord["x"], coord["y"]])
             new = np.array([new_coord["x"], new_coord["y"]])
-            if np.linalg.norm(existing - new) < 10:
-                is_close = True
-                to_remove = coord
+            if  np.linalg.norm(existing - new) < 10:
+                st.session_state.is_close = True
                 break
-        if is_close and to_remove is not None:
-            st.session_state.coords_list.remove(to_remove)
-            # Increment the widget key to reset the widget
-            st.session_state.widget_key += 1
+        if st.session_state.is_close:
+            st.session_state.coords_list.remove(coord)
         else:
             st.session_state.coords_list.append(new_coord)
-            # Also increment the key after a new addition if desired
-            st.session_state.widget_key += 1
+        st.rerun()
 
     st.write("Coordinates:", st.session_state.get("coords"))