updates

app.py

@@ -77,17 +77,8 @@ def patch(ij):
   imc=imm.crop(((x-1)*patch_size,(y-1)*patch_size,(x+2)*patch_size,(y+2)*patch_size))
   return imc
 
-if "sideimg" not in st.session_state:
-  st.session_state["sideimg"] = [patch(i) for i in range(4)]
-
-if "sideix" not in st.session_state:
-  st.session_state["sideix"] = [i for i in range(4)]
-
-def button_click():
-    st.session_state["img"]=np.random.randint(100)
-    st.session_state["draw"] = False
-
 def find():
+  st.session_state["sideix"] = []
   point=st.session_state["point"]
   point=(point[0]//patch_size,point[1]//patch_size)
   #point=point[0]*36+point[1]
@@ -103,8 +94,6 @@ def find():
   batches=[]
   while ix<4:
 
-  #for ix in range(4):
-
     batch=diff.argsort()[i]//(gridsize**2)//20
 
     if batch not in batches:
@@ -116,20 +105,18 @@ def find():
     
     i+=1
   
-    st.session_state["sideix"]=batches
-
-    
-    #st.session_state["sideix"][ix]=diff.argsort()[ix]
-
-    #st.write(diff.argsort()[ix])
-
+  st.session_state["sideix"]=batches
 
 
+def button_click():
+    st.session_state["img"]=np.random.randint(100)
+    st.session_state["draw"] = False
 
- # for i in range(4):
- #   st.session_state["sideimg"][i]+=1
- #   st.image(ds[0]['image'])
+if "sideimg" not in st.session_state:
+  st.session_state["sideimg"] = [patch(i) for i in range(4)]
 
+if "sideix" not in st.session_state:
+  find()
 
 def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
     center = point