Update app.py

app.py

@@ -272,15 +272,24 @@ def process_image_with_prompt(image, prompt, model="o3-mini-high"):
     except Exception as e:
         return f"Error processing image with GPT: {str(e)}"
 
-# --- Gallery Update ---
+# --- Sidebar Setup (Add this near the top, before tabs) ---
+st.sidebar.subheader("Gallery Settings")
+if 'gallery_size' not in st.session_state:
+    st.session_state['gallery_size'] = 2  # Default value
+st.session_state['gallery_size'] = st.sidebar.slider(
+    "Gallery Size", 
+    1, 10, st.session_state['gallery_size'], 
+    key="gallery_size_slider"  # Unique key for the slider
+)
+
+# --- Updated Gallery Function ---
 def update_gallery():
     all_files = get_gallery_files()
     if all_files:
         st.sidebar.subheader("Asset Gallery 📸📖")
         cols = st.sidebar.columns(2)
-        for idx, file in enumerate(all_files[:st.sidebar.slider("Gallery Size", 1, 10, 2, key="gallery_size_update")]):
-
-        #for idx, file in enumerate(all_files[:st.sidebar.slider("Gallery Size", 1, 10, 2)]):
+        # Use the stored gallery size from session state
+        for idx, file in enumerate(all_files[:st.session_state['gallery_size']]):
             with cols[idx % 2]:
                 st.session_state['unique_counter'] += 1
                 unique_id = st.session_state['unique_counter']
@@ -293,16 +302,23 @@ def update_gallery():
                     st.image(img, caption=os.path.basename(file), use_container_width=True)
                     doc.close()
                 checkbox_key = f"asset_{file}_{unique_id}"
-                st.session_state['asset_checkboxes'][file] = st.checkbox("Use for SFT/Input", value=st.session_state['asset_checkboxes'].get(file, False), key=checkbox_key)
+                st.session_state['asset_checkboxes'][file] = st.checkbox(
+                    "Use for SFT/Input", 
+                    value=st.session_state['asset_checkboxes'].get(file, False), 
+                    key=checkbox_key
+                )
                 mime_type = "image/png" if file.endswith('.png') else "application/pdf"
                 st.markdown(get_download_link(file, mime_type, "Snag It! 📥"), unsafe_allow_html=True)
                 if st.button("Zap It! 🗑️", key=f"delete_{file}_{unique_id}"):
                     os.remove(file)
                     st.session_state['asset_checkboxes'].pop(file, None)
                     st.sidebar.success(f"Asset {os.path.basename(file)} vaporized! 💨")
-                    st.experimental_rerun()
+                    st.rerun()
+
+# Call update_gallery() once initially or after actions as needed
 update_gallery()
 
+
 # --- Sidebar Logs & History ---
 st.sidebar.subheader("Action Logs 📜")
 with st.sidebar:
@@ -511,7 +527,7 @@ with tab_build:
         if entry not in st.session_state['history']:
             st.session_state['history'].append(entry)
         st.success(f"Model downloaded and saved to {config.model_path}! 🎉")
-        st.experimental_rerun()
+        st.rerun()
 
 # === Tab: Test Image Gen (existing) ===
 with tab_imggen: