Update app.py

app.py

@@ -11,47 +11,43 @@ from diffusers import AutoPipelineForInpainting, AutoencoderKL
 # Pure-CPU helpers (no CUDA here)
 # -----------------------------
 from PIL import Image, ImageChops
+import math
+
+def _round_up(x, m=8):
+    return int(math.ceil(x / m) * m)
 
 def autocrop_content(img: Image.Image, bg_color=(255, 255, 255), tol=12) -> Image.Image:
-    """
-    Trim uniform white (or near-white) margins before we center/pad.
-    Works for RGB and RGBA. `tol` is tolerance for "how white".
-    """
     if img.mode in ("RGBA", "LA"):
-        # If we have transparency, use alpha channel bbox
         alpha = img.split()[-1]
         bbox = alpha.getbbox()
         return img.crop(bbox) if bbox else img
 
-    # RGB: compare to a solid background
     bg = Image.new(img.mode, img.size, bg_color)
     diff = ImageChops.difference(img, bg).convert("L")
-    # Binarize with tolerance to catch JPEG off-white backgrounds
     mask = diff.point(lambda p: 255 if p > tol else 0, mode="1")
     bbox = mask.getbbox()
     return img.crop(bbox) if bbox else img
 
 def squarify_image(img: Image.Image, color="white") -> Image.Image:
-    """
-    1) Trim outer white margins.
-    2) Pad to square, centered.
-    """
+    # 1) trim margins
     img = autocrop_content(img, bg_color=(255, 255, 255), tol=12)
 
     w, h = img.size
-    size = max(w, h)
-    bg = Image.new("RGB", (size, size), color=color)
-    bg.paste(img, ((size - w) // 2, (size - h) // 2))
-    return bg
+    # 2) target square side rounded **up** to /8
+    side = _round_up(max(w, h), 8)
 
+    bg = Image.new("RGB", (side, side), color=color)
+    bg.paste(img, ((side - w) // 2, (side - h) // 2))
+    return bg
 
 def divisible_by_8(image: Image.Image) -> Image.Image:
     w, h = image.size
-    new_w = max(8, (w // 8) * 8)
-    new_h = max(8, (h // 8) * 8)
-    if (new_w, new_h) != (w, h):
-        image = image.resize((new_w, new_h), Image.BICUBIC)
-    return image
+    # round **up** so we never shrink (keeps content + avoids 1012-style errors)
+    nw = _round_up(w, 8)
+    nh = _round_up(h, 8)
+    if (nw, nh) == (w, h):
+        return image
+    return image.resize((nw, nh), Image.LANCZOS)
 
 # -----------------------------
 # Lazy singletons (created inside GPU context)