app.py updates

app.py

@@ -22,24 +22,24 @@ ds = load_dataset("Circularmachines/batch_indexing_machine_100_small_imgs", spli
 
 #"# :dart: Streamlit Image Coordinates: Image Update"
 
-#if "points" not in st.session_state:
-st.session_state["points"] = []
+if "points" not in st.session_state:
+  st.session_state["points"] = []
 
-
-#st.session_state["img"] = 0
+if "img" not in st.session_state:
+  st.session_state["img"] = 0
 
 #"## Click on image"
 
 
 r=0#np.random.randint(100)
-current_image=ds[r]['image']
+current_image=ds[st.session_state["img"]]['image']
 
 def button_click():
-    r=np.random.randint(100)
+    st.session_state["img"]=np.random.randint(100)
     current_image=ds[r]['image']
     #st.write(str(r))
     st.session_state["points"] = []
-    value = streamlit_image_coordinates(current_image, key="pil")
+    #value = streamlit_image_coordinates(current_image, key="pil")