Update app.py

app.py

@@ -63,6 +63,8 @@ if 'asset_checkboxes' not in st.session_state:
     st.session_state['asset_checkboxes'] = {}
 if 'downloaded_pdfs' not in st.session_state:
     st.session_state['downloaded_pdfs'] = {}
+if 'unique_counter' not in st.session_state:
+    st.session_state['unique_counter'] = 0  # For generating unique keys
 
 @dataclass
 class ModelConfig:
@@ -531,6 +533,8 @@ def update_gallery():
         cols = st.sidebar.columns(2)
         for idx, file in enumerate(all_files[:gallery_size * 2]):
             with cols[idx % 2]:
+                st.session_state['unique_counter'] += 1  # Increment counter for uniqueness
+                unique_id = st.session_state['unique_counter']
                 if file.endswith('.png'):
                     st.image(Image.open(file), caption=os.path.basename(file), use_container_width=True)
                 else:
@@ -539,7 +543,7 @@ def update_gallery():
                     img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
                     st.image(img, caption=os.path.basename(file), use_container_width=True)
                     doc.close()
-                checkbox_key = f"asset_{file}_{idx}"  # Unique key with index
+                checkbox_key = f"asset_{file}_{unique_id}"  # Unique key with counter
                 st.session_state['asset_checkboxes'][file] = st.checkbox(
                     "Use for SFT/Input", 
                     value=st.session_state['asset_checkboxes'].get(file, False), 
@@ -547,7 +551,7 @@ def update_gallery():
                 )
                 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}_{idx}"):  # Unique key with index
+                if st.button("Zap It! 🗑️", key=f"delete_{file}_{unique_id}"):  # Unique key with counter
                     os.remove(file)
                     if file in st.session_state['asset_checkboxes']:
                         del st.session_state['asset_checkboxes'][file]