app.py updates

app.py

@@ -33,12 +33,12 @@ current_image=ds[0]['image']
 
 def button_click():
     r=np.random.randint(100)
-    #current_image=ds[r]['image']
-    st.write(str(r))
+    current_image=ds[r]['image']
+    #st.write(str(r))
     st.session_state["points"] = []
 
 
-with ds[r]['image'] as img:
+with current_image as img:
     draw = ImageDraw.Draw(img)
 
     def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):