Update app.py

app.py

@@ -331,12 +331,12 @@ 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
 
     # 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)
@@ -345,25 +345,29 @@ def main_app():
         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
+    # 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(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("---")
+        num_images_to_display = min(st.session_state.max_images_to_display, num_images) # To limit max images to display
+
+        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:
+            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("---")
     else:
       st.write("No image generated yet!")
 
-
     # Logout button
     if st.button("Logout", on_click=logout_callback):
         pass