Update app_pages/image_generation.py

app_pages/image_generation.py

@@ -39,7 +39,7 @@ def _zip_gallery_images(gallery_items):
         shutil.rmtree(temp_dir, ignore_errors=True)
 
 
-def render_bulk_image_generator(prefix: str = "ig_img"):
+def render_bulk_image_generator(uid: str, prefix: str = "ig_img"):
     zip_file = st.file_uploader(
         "Upload Zip or Single File",
         type=["zip", "png", "jpg", "jpeg"],
@@ -99,6 +99,7 @@ def render_bulk_image_generator(prefix: str = "ig_img"):
                         demo_btn,
                         gallery,
                         blur,
+                        uid,
                     )
                     st.session_state[gallery_key] = images or []
                     images = st.session_state[gallery_key]