updates

app.py

@@ -177,10 +177,15 @@ with col1:
   st.write(st.session_state["draw"])
 
 with col2:
-
+ # st.write("current selection:")
   for i in range(4):
     st.image(st.session_state["sideimg"][i].resize((128,128)))
-    st.write("batch: "+str(st.session_state["sideix"][i]))#//(gridsize**2)//20))
+    if i==0:
+      st.write("current batch: "+str(st.session_state["sideix"][i]))#//(gridsize**2)//20))
+    else:
+      st.write("other batch: "+str(st.session_state["sideix"][i]))#//(gridsize**2)//20))
+   
+    
     #st.write(st.session_state["sideix"][i])