push app

app.py

@@ -83,29 +83,34 @@ if picture is not None:
     img = picture.copy()
     img = crop_resize_image(img, size=512)
     st.session_state.img = img  # save the processed image
+    st.session_state.coords_list = []
 
 if "img" in st.session_state:
     img = st.session_state.img
 
-    # If we have coordinates stored, create an image with a red dot overlay.
-    if "coords" in st.session_state:
-        cx, cy = int(st.session_state.coords["x"]), int(st.session_state.coords["y"])
-        img_with_dot = img.copy()
-        draw = ImageDraw.Draw(img_with_dot)
-        draw.ellipse((cx-5, cy-5, cx+5, cy+5), fill="red")
-        display_img = img_with_dot
-    else:
-        display_img = img
-
-    # Call the streamlit_image_coordinates component with our display image.
-    coords = streamlit_image_coordinates(display_img, key="click_img")
-
-    # If new coordinates are returned and differ from what we have stored, update them and rerun.
-    if coords and st.session_state.get("coords") != coords:
-        st.session_state.coords = coords
+    # Initialize the coordinates list if it doesn't exist.
+    if "coords_list" not in st.session_state:
+        st.session_state.coords_list = []
+
+    # Create a copy of the image and draw red dots for every stored coordinate.
+    img_with_dots = img.copy()
+    draw = ImageDraw.Draw(img_with_dots)
+    for coord in st.session_state.coords_list:
+        cx, cy = int(coord["x"]), int(coord["y"])
+        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="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:
+        st.session_state.coords_list.append(new_coord)
         st.rerun()
 
     st.write("Coordinates:", st.session_state.get("coords"))
+
+
+    
     # Prompt input for inpainting
     prompt = st.text_input("Prompt for inpainting (describe what should replace the selected area):")