push app

app.py

@@ -88,24 +88,23 @@ if picture is not None:
     # Use the canvas container for all related UI elements
     with canvas.container():
         # canvas.image(img, caption="Captured (512x512) Image", use_container_width=True)
-        
         # Place the interactive component inside the same container
         if 'dotted_image' not in st.session_state:
             img_to_dot = img
         else:
             img_to_dot = st.session_state.dotted_image
         coords = streamlit_image_coordinates(img_to_dot, key="click_img")
-
+        img_with_dot = img.copy()
         if coords:
             cx, cy = int(coords['x']), int(coords['y'])
             # Draw a red circle on the image to indicate the selected point
-            img_with_dot = img.copy()
             draw = ImageDraw.Draw(img_with_dot)
             draw.ellipse((cx-5, cy-5, cx+5, cy+5), fill='red')
             # canvas.image(img_with_dot, caption=f"Selected Point: ({cx}, {cy})", use_container_width=True)
-            st.session_state.dotted_image = img_with_dot
+            
         else:
             cx = cy = None
+        st.session_state.dotted_image = img_with_dot
 
     # Prompt input for inpainting
     prompt = st.text_input("Prompt for inpainting (describe what should replace the selected area):")