app.py updates

app.py

@@ -22,8 +22,8 @@ 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"] = (200,200)
+if "point" not in st.session_state:
+  st.session_state["point"] = (200,200)
 
 if "img" not in st.session_state:
   st.session_state["img"] = 0
@@ -39,8 +39,13 @@ def button_click():
     st.session_state["draw"] = False
 
 def find():
-  for i in range(4):
-    st.session_state["sideimg"][i]+=1
+  point=st.session_state["point"]
+  st.write()
+
+
+
+ # for i in range(4):
+ #   st.session_state["sideimg"][i]+=1
  #   st.image(ds[0]['image'])
 
 
@@ -66,26 +71,28 @@ with col1:
 
   # Draw an ellipse at each coordinate in points
     #for point in st.session_state["points"]:
-    point=st.session_state["points"]
+    point=st.session_state["point"]
     coords = get_ellipse_coords(point)
     draw.rectangle(coords, outline="green",width=2)
 
   value = streamlit_image_coordinates(current_image, key="pil")
 
   if value is not None:
-      point = value["x"], value["y"]
+      point = value["x"]//32*32, value["y"]//32*32
 
-      if point != st.session_state["points"]:
-          st.session_state["points"]=point
+      if point != st.session_state["point"]:
+          st.session_state["point"]=point
           st.session_state["draw"]=True
           st.experimental_rerun()
 
-
-  st.button('Random frame', on_click=button_click)
-  st.button('Find simalar parts', on_click=find)
+  subcol1, subcol2 = st.columns(3)
+  with subcol1:
+    st.button('Previous Frame', on_click=button_click)
+    st.button('Change Batch', on_click=button_click)
+    st.button('Find similar parts', on_click=find)
 
   st.write(st.session_state["img"])
-  st.write(st.session_state["points"])
+  st.write(st.session_state["point"])
   st.write(st.session_state["draw"])
 
 with col2:

pred.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c7dd00bca5acaa98f0b6c16c73eec0c6830d3643d2a273fa90c5ee64b26c1a2
+size 30924928