Update app.py

app.py

@@ -328,20 +328,32 @@ def main_app():
 
     st.header("Your Generated Images")
 
-     # Load initial batch of images
-    if st.session_state.images_data == [] or st.session_state.load_more_pressed == True:
-        if st.session_state.load_more_pressed == True:
-           st.session_state.start_index += st.session_state.batch_size
-           st.session_state.images_data = [] # Clear the images data when load more is pressed
-        # load all images if new login or load_more is pressed, also reset the flag
-        st.session_state.images_data = load_image_data(st.session_state.current_user, st.session_state.start_index, st.session_state.batch_size)
-        st.session_state.load_more_pressed = False
-
-    # Setup columns for horizontal slider layout
-    if st.session_state.images_data:
-        num_images = len(st.session_state.images_data)
+    # Initialize the current window, if it doesn't exist in session state
+    if "current_window_start" not in st.session_state:
+        st.session_state.current_window_start = 0
+
+
+    # Create left and right arrow buttons
+    col_left, col_center, col_right = st.columns([1,8,1])
+
+
+    with col_left:
+        if st.button("◀️"):
+            st.session_state.current_window_start = max(0, st.session_state.current_window_start - st.session_state.batch_size)
+
+    with col_right:
+        if st.button("▶️"):
+            st.session_state.current_window_start += st.session_state.batch_size
+
+
+    # Dynamically load images for the window
+    images_for_window = load_image_data(st.session_state.current_user, st.session_state.current_window_start, st.session_state.batch_size*3) # Load three batch sizes to cover sliding window
+
+     # Setup columns for horizontal slider layout
+    if images_for_window:
+        num_images = len(images_for_window)
         cols = st.columns(num_images) # dynamically create columns according to data
-        for i, image_data in enumerate(st.session_state.images_data):
+        for i, image_data in enumerate(images_for_window):
            with cols[i]:
                st.image(image_data['image_url'], width = 150)
                st.write(f"**Prompt:** {image_data['prompt']}")
@@ -352,12 +364,6 @@ def main_app():
       st.write("No image generated yet!")
 
 
-    # Load more button
-    if st.button("Load More Images"):
-        st.session_state.load_more_pressed = True
-        # No loading here, we are now loading all the images when `load_more_pressed` flag is set to True
-
-
     # Logout button
     if st.button("Logout", on_click=logout_callback):
         pass