Remove duplicate load_gallery_images definition to fix UnboundLocalError

ui/layout.py

@@ -9,6 +9,17 @@ MAX_DYNAMIC_CONTROLS = 10
 def build_ui(event_handler_function):
     ui_components = {}
 
+    # Helper function to load thumbnails from the output folder.
+    def load_gallery_images():
+        output_dir = os.path.abspath(OUTPUT_DIR)
+        if not os.path.isdir(output_dir):
+            return []
+        image_paths = []
+        for fname in sorted(os.listdir(output_dir)):
+            if fname.lower().endswith((".png", ".jpg", ".jpeg", ".gif", ".webp")):
+                image_paths.append(os.path.join(output_dir, fname))
+        return image_paths
+
     # All UI components, including tabs, must be created inside the same Gradio Blocks context.
     with gr.Blocks() as demo:
         gr.Markdown("# ImageGen")
@@ -55,19 +66,7 @@ def build_ui(event_handler_function):
 
         gr.Markdown("<div style='text-align: center; margin-top: 20px;'>Made by RioShiina with ❤️<br><a href='https://github.com/RioShiina47' target='_blank'>GitHub</a> | <a href='https://huggingface.co/RioShiina' target='_blank'>Hugging Face</a> | <a href='https://civitai.com/user/RioShiina' target='_blank'>Civitai</a></div>")
 
-        # Load gallery images on startup
-        def load_gallery_images():
-            # Resolve absolute path for the output directory
-            output_dir = os.path.abspath(OUTPUT_DIR)
-            if not os.path.isdir(output_dir):
-                return []
-            # Collect image files (common formats)
-            image_paths = []
-            for fname in sorted(os.listdir(output_dir)):
-                if fname.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.webp')):
-                    image_paths.append(os.path.join(output_dir, fname))
-            return image_paths
-
+        # Load gallery images on startup using the helper defined earlier.
         demo.load(fn=load_gallery_images, outputs=ui_components["gallery"])
 
         # Connect event handlers (e.g., button clicks) with the UI components.