app.py updates

app.py

@@ -28,6 +28,9 @@ if "points" not in st.session_state:
 if "img" not in st.session_state:
   st.session_state["img"] = 0
 
+if "draw" not in st.session_state:
+  st.session_state["draw"] = False
+
 #"## Click on image"
 
 
@@ -40,7 +43,7 @@ def button_click():
     #st.write(str(r))
     st.session_state["points"] = []
     #value = streamlit_image_coordinates(current_image, key="pil")
-
+    st.session_state["draw"]=False
 
 
 
@@ -57,10 +60,12 @@ def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
     )
 
 
+if st.session_state["draw"]:
+
 # Draw an ellipse at each coordinate in points
-for point in st.session_state["points"]:
-    coords = get_ellipse_coords(point)
-    draw.rectangle(coords, outline="green",width=2)
+  for point in st.session_state["points"]:
+      coords = get_ellipse_coords(point)
+      draw.rectangle(coords, outline="green",width=2)
 
 value = streamlit_image_coordinates(current_image, key="pil")
 
@@ -69,6 +74,7 @@ if value is not None:
 
     if point not in st.session_state["points"]:
         st.session_state["points"]=[point]
+        st.session_state["draw"]=True
         st.experimental_rerun()