Update app.py

app.py

@@ -12,16 +12,15 @@ For batch processing, upload a .zip of images (PNG/JPG/JPEG) and download the ou
 """
 
 EXAMPLES_DIR = "examples"
+CREATE_DEFAULTS_IF_EMPTY = True  # set False if you never want auto-generated examples
 
-def _ensure_examples():
+# ---------- Examples handling ----------
+def _make_default_examples():
     os.makedirs(EXAMPLES_DIR, exist_ok=True)
-    paths = [
-        os.path.join(EXAMPLES_DIR, "checkerboard.png"),
-        os.path.join(EXAMPLES_DIR, "gradient.png"),
-        os.path.join(EXAMPLES_DIR, "shapes.png"),
-    ]
+
     # 1) Checkerboard
-    if not os.path.exists(paths[0]):
+    cb_path = os.path.join(EXAMPLES_DIR, "checkerboard.png")
+    if not os.path.exists(cb_path):
         cb = Image.new("RGB", (512, 512), "white")
         draw = ImageDraw.Draw(cb)
         tile = 64
@@ -29,18 +28,22 @@ def _ensure_examples():
             for x in range(0, 512, tile):
                 if (x//tile + y//tile) % 2 == 0:
                     draw.rectangle([x, y, x+tile-1, y+tile-1], fill=(30, 30, 30))
-        cb.save(paths[0])
+        cb.save(cb_path)
+
     # 2) Gradient
-    if not os.path.exists(paths[1]):
+    grad_path = os.path.join(EXAMPLES_DIR, "gradient.png")
+    if not os.path.exists(grad_path):
         arr = np.zeros((360, 640, 3), dtype=np.uint8)
         for x in range(640):
             arr[:, x, 0] = int(255 * x / 639)
         for y in range(360):
             arr[y, :, 1] = int(255 * y / 359)
         arr[:, :, 2] = 160
-        Image.fromarray(arr).save(paths[1])
+        Image.fromarray(arr).save(grad_path)
+
     # 3) Shapes
-    if not os.path.exists(paths[2]):
+    shapes_path = os.path.join(EXAMPLES_DIR, "shapes.png")
+    if not os.path.exists(shapes_path):
         sh = Image.new("RGB", (512, 384), "white")
         d = ImageDraw.Draw(sh)
         colors = [(220,20,60),(65,105,225),(60,179,113),(255,165,0),(148,0,211)]
@@ -48,9 +51,29 @@ def _ensure_examples():
             d.rectangle([20+90*i, 30, 80+90*i, 180], fill=c, outline=(0,0,0), width=3)
         for i in range(6):
             d.ellipse([40+80*i, 200, 90+80*i, 350], fill=colors[i%len(colors)], outline=(0,0,0), width=3)
-        sh.save(paths[2])
-    return [[p] for p in paths]
-
+        sh.save(shapes_path)
+
+def _list_example_images():
+    """Return [[path], [path], ...] for all images under examples/ (recursive)."""
+    exts = {".png", ".jpg", ".jpeg", ".bmp", ".webp"}
+    items = []
+    if os.path.isdir(EXAMPLES_DIR):
+        for root, _, files in os.walk(EXAMPLES_DIR):
+            for f in files:
+                if os.path.splitext(f)[1].lower() in exts:
+                    items.append([os.path.join(root, f)])
+    # sort by path for stable order
+    items.sort(key=lambda x: x[0].lower())
+    return items
+
+def _get_examples():
+    items = _list_example_images()
+    if not items and CREATE_DEFAULTS_IF_EMPTY:
+        _make_default_examples()
+        items = _list_example_images()
+    return items
+
+# ---------- App logic ----------
 def _parse_grid(grid_choice: str) -> int:
     # Expect strings like "2x2", "4x4", "8x8", "16x16"
     try:
@@ -130,11 +153,12 @@ def run_batch(zip_file, grid_choice, use_overlap, overlap_pct, mix_prob, beta_a,
     msg = f"Processed {count_out}/{count_in} files."
     return out_zip, msg
 
+# ---------- UI ----------
 with gr.Blocks() as demo:
     gr.Markdown(f"# {TITLE}")
     gr.Markdown(DESC)
 
-    examples = _ensure_examples()
+    examples = _get_examples()
 
     with gr.Tabs():
         with gr.TabItem("Single Image"):