app.py updates

app.py

@@ -3,6 +3,7 @@ from PIL import Image, ImageDraw
 
 from streamlit_image_coordinates import streamlit_image_coordinates
 
+import numpy as np
 
 from datasets import load_dataset
 
@@ -13,11 +14,11 @@ ds = load_dataset("Circularmachines/batch_indexing_machine_100_small_imgs", spli
 
 
 
-st.set_page_config(
-    page_title="Streamlit Image Coordinates: Image Update",
-    page_icon="🎯",
-    layout="wide",
-)
+#st.set_page_config(
+#    page_title="Streamlit Image Coordinates: Image Update",
+#    page_icon="🎯",
+#    layout="wide",
+#)
 
 #"# :dart: Streamlit Image Coordinates: Image Update"
 
@@ -27,8 +28,11 @@ if "points" not in st.session_state:
 
 #"## Click on image"
 
+r=np.random.randint(5000)
+current_image=ds[r]['image']
 
-with ds[0]['image'] as img:
+
+with current_image as img:
     draw = ImageDraw.Draw(img)
 
     def get_ellipse_coords(point):# tuple[int, int]) -> tuple[int, int, int, int]):
@@ -57,7 +61,9 @@ with ds[0]['image'] as img:
             st.experimental_rerun()
 
 
-if st.button('Say hello'):
-    st.write('Why hello there')
-else:
-    st.write('Gohuhuhubye')
+if st.button('Random frame'):
+    r=np.random.randint(5000)
+    current_image=ds[r]['image']
+    st.write(str(r))
+#else:
+#    st.write('Gohuhuhubye')