app.py updates

app.py

@@ -35,11 +35,10 @@ if "draw" not in st.session_state:
 def button_click():
     st.session_state["img"]=np.random.randint(100)
     st.session_state["draw"] = False
-    #current_image=ds[r]['image']
-    #st.write(str(r))
-    #st.session_state["points"] = (200,200)
-    #value = streamlit_image_coordinates(current_image, key="pil")
-    
+
+def find():
+    st.image(ds[st.session_state[0]]['image'])
+
 
 def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
     center = point
@@ -75,6 +74,7 @@ if value is not None:
 
 
 st.button('Random frame', on_click=button_click)
+st.button('Find simalar parts', on_click=find)
 
 st.write(st.session_state["img"])
 st.write(st.session_state["points"])