app.py updates

app.py

@@ -31,13 +31,17 @@ if "img" not in st.session_state:
 if "draw" not in st.session_state:
   st.session_state["draw"] = False
 
+if "sideimg" not in st.session_state:
+  st.session_state["sideimg"] = [0,1,2,3]
 
 def button_click():
     st.session_state["img"]=np.random.randint(100)
     st.session_state["draw"] = False
 
 def find():
-    st.image(ds[0]['image'])
+  for i in range(4):
+    st.image(ds[0]['image'][i]+=1
+ #   st.image(ds[0]['image'])
 
 
 def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
@@ -51,35 +55,42 @@ def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
     )
 
 
-current_image=ds[st.session_state["img"]]['image']
-draw = ImageDraw.Draw(current_image)
+col1, col2 = st.columns(2)
 
-if st.session_state["draw"]:
+with col1:
 
-# Draw an ellipse at each coordinate in points
-  #for point in st.session_state["points"]:
-  point=st.session_state["points"]
-  coords = get_ellipse_coords(point)
-  draw.rectangle(coords, outline="green",width=2)
+  current_image=ds[st.session_state["img"]]['image']
+  draw = ImageDraw.Draw(current_image)
 
-value = streamlit_image_coordinates(current_image, key="pil")
+  if st.session_state["draw"]:
 
-if value is not None:
-    point = value["x"], value["y"]
+  # Draw an ellipse at each coordinate in points
+    #for point in st.session_state["points"]:
+    point=st.session_state["points"]
+    coords = get_ellipse_coords(point)
+    draw.rectangle(coords, outline="green",width=2)
 
-    if point != st.session_state["points"]:
-        st.session_state["points"]=point
-        st.session_state["draw"]=True
-        st.experimental_rerun()
+  value = streamlit_image_coordinates(current_image, key="pil")
 
+  if value is not None:
+      point = value["x"], value["y"]
 
-st.button('Random frame', on_click=button_click)
-st.button('Find simalar parts', on_click=find)
+      if point != st.session_state["points"]:
+          st.session_state["points"]=point
+          st.session_state["draw"]=True
+          st.experimental_rerun()
 
-st.write(st.session_state["img"])
-st.write(st.session_state["points"])
-st.write(st.session_state["draw"])
-#st.write(value["x"], value["y"])
 
-#else:
-#    st.write('Gohuhuhubye')
+  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"])
+  st.write(st.session_state["draw"])
+
+with col2:
+
+  for i in range(4):
+    st.image(np.array(st.session_state["sideimg"][i])[::4,::4,:])
+
+