app.py updates

app.py

@@ -28,10 +28,8 @@ if "points" not in st.session_state:
 
 #"## Click on image"
 
-if r not in vars():
-  r=0
 
-#r=0#np.random.randint(100)
+r=0#np.random.randint(100)
 current_image=ds[r]['image']
 
 def button_click():
@@ -40,38 +38,40 @@ def button_click():
     #st.write(str(r))
     st.session_state["points"] = []
 
+while True:
 
-with current_image as img:
-    draw = ImageDraw.Draw(img)
+  with current_image as img:
+      draw = ImageDraw.Draw(img)
 
-    def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
-        center = point
-        radius = 16
-        return (
-            center[0] - radius,
-            center[1] - radius,
-            center[0] + radius,
-            center[1] + radius,
-        )
+      def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
+          center = point
+          radius = 16
+          return (
+              center[0] - radius,
+              center[1] - radius,
+              center[0] + radius,
+              center[1] + radius,
+          )
 
 
-    # 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)
+      # 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)
 
-    value = streamlit_image_coordinates(img, key="pil")
+      value = streamlit_image_coordinates(img, key="pil")
 
-    if value is not None:
-        point = value["x"], value["y"]
+      if value is not None:
+          point = value["x"], value["y"]
 
-        if point not in st.session_state["points"]:
-            st.session_state["points"]=[point]
-            st.experimental_rerun()
+          if point not in st.session_state["points"]:
+              st.session_state["points"]=[point]
+              st.experimental_rerun()
 
 
-    st.button('Random frame', on_click=button_click)
-
+      st.button('Random frame', on_click=button_click)
+      
+  st.write(str(r))
 
 #else:
 #    st.write('Gohuhuhubye')