Update app.py

app.py

@@ -330,43 +330,48 @@ def main_app():
 
     # 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
-    if "max_images_to_display" not in st.session_state:
-      st.session_state.max_images_to_display = 5 #maximum number of images to display
+      st.session_state.current_window_start = 0
+
+    if "window_size" not in st.session_state:
+      st.session_state.window_size = 5 # The number of images to display at a time
 
     # 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)
+            st.session_state.current_window_start = max(0, st.session_state.current_window_start - st.session_state.window_size)
 
     with col_right:
         if st.button("▶️"):
-            st.session_state.current_window_start += st.session_state.batch_size
+            st.session_state.current_window_start += st.session_state.window_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
+    all_images = load_image_data(st.session_state.current_user, 0, 1000) # load all images
+    
+    if all_images:
+        num_images = len(all_images)
+
+        # Calculate the range for images to display
+        start_index = st.session_state.current_window_start
+        end_index = min(start_index + st.session_state.window_size, num_images)
+        
+        images_for_window = all_images[start_index:end_index]
 
-    # Setup columns for horizontal slider layout
-    if images_for_window:
-        num_images = len(images_for_window)
-        num_images_to_display = min(st.session_state.max_images_to_display, num_images) # To limit max images to display
+        # Setup columns for horizontal slider layout
+        num_images_to_display = len(images_for_window)
 
         cols = st.columns(num_images_to_display)
-        
-        start_index = max(0, num_images - num_images_to_display)
-        
-        for i, image_data in enumerate(images_for_window[start_index:]):
-          if i < num_images_to_display:
+        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']}")
-              st.write(f"**Aspect Ratio:** {image_data['aspect_ratio']}")
-              st.write(f"**Realism:** {image_data['realism']}")
-              st.markdown("---")
+                st.image(image_data['image_url'], width = 150)
+                st.write(f"**Prompt:** {image_data['prompt']}")
+                st.write(f"**Aspect Ratio:** {image_data['aspect_ratio']}")
+                st.write(f"**Realism:** {image_data['realism']}")
+                st.markdown("---")
+
     else:
-      st.write("No image generated yet!")
+        st.write("No image generated yet!")
 
     # Logout button
     if st.button("Logout", on_click=logout_callback):