push app

app.py

@@ -66,6 +66,10 @@ def load_models():
 # Initialize the models (this will run only once thanks to caching)
 # fastsam_model, sd_pipe = load_models()
 
+# Ensure we have a state for removing_dots
+if "is_removing_dot" not in st.session_state:
+    st.session_state.is_removing_dot = False
+
 # Title
 st.subheader("ReDesign")
 
@@ -103,7 +107,7 @@ if "img" in st.session_state:
     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 and new_coord not in st.session_state.coords_list:
+    if new_coord and new_coord not in st.session_state.coords_list and not st.session_state.is_removing_dot:
         is_close = False
         for coord in st.session_state.coords_list:
             existing = np.array([coord["x"], coord["y"]])
@@ -113,8 +117,10 @@ if "img" in st.session_state:
                 break
         if is_close:
             st.session_state.coords_list.remove(coord)
+            st.session_state.is_removing_dot = True
         else:
             st.session_state.coords_list.append(new_coord)
+            st.session_state.is_removing_dot = False
         st.rerun()
 
     st.write("Coordinates:", st.session_state.get("coords"))