Update app.py

app.py

@@ -672,14 +672,13 @@ def make_image_sized_pdf(sources):
         return None
 
 
+
 # --- Sidebar Gallery Update Function (Keep from previous) --------
-# ... (update_gallery function assumed here - no changes needed based on AI backend) ...
 def update_gallery():
     container = st.session_state['asset_gallery_container']
     container.empty()
     with container.container():
         st.markdown("### Asset Gallery 📸📖")
-        st.session_state['gallery_size'] = st.slider("Max Items Shown", 2, 50, st.session_state.get('gallery_size', 10), key="gallery_size_slider")
         cols = st.columns(2)
         all_files = get_gallery_files()
         if not all_files: st.info("No assets found."); return
@@ -805,7 +804,24 @@ with st.sidebar.expander("Generation Parameters", expanded=False):
     st.session_state.gen_seed = st.slider("Seed", -1, 65535, st.session_state.get('gen_seed', -1), step=1, key="param_seed", help="-1 for random.")
 
 
-st.sidebar.markdown("---") # Separator before gallery
+
+st.sidebar.markdown("---") # Separator before gallery settings
+
+# --- ADDED: Gallery Settings Section ---
+st.sidebar.subheader("🖼️ Gallery Settings")
+st.slider(
+    "Max Items Shown",
+    min_value=2,
+    max_value=50, # Adjust max if needed
+    value=st.session_state.get('gallery_size', 10),
+    key="gallery_size_slider", # Keep the key, define it ONCE here
+    help="Controls the maximum number of assets displayed in the sidebar gallery."
+)
+st.session_state.gallery_size = st.session_state.gallery_size_slider # Ensure sync
+st.sidebar.markdown("---") # Separator after gallery settings
+
+
+
 
 # --- App Title -------------------------------------------
 st.title("Vision & Layout Titans (HF) 🚀🖼️📄")