v12.1: Fix QR alignment (496->512px resize), reduce feather to 16px, bump overlay to 60%

handler.py

@@ -1,5 +1,5 @@
 """
-QR-Verse AI Art Generator — HuggingFace Inference Endpoint Handler v12
+QR-Verse AI Art Generator — HuggingFace Inference Endpoint Handler v12.1
 
 Art + QR overlay pipeline: ControlNet art generation + post-processing QR composite.
 
@@ -66,10 +66,10 @@ P2_STRENGTH = 0.15
 # ---------------------------------------------------------------------------
 # QR overlay post-processing
 # ---------------------------------------------------------------------------
-OVERLAY_OPACITY = 0.55       # Alpha for QR modules (0=invisible, 1=solid black)
+OVERLAY_OPACITY = 0.60       # Alpha for QR modules (0=invisible, 1=solid black)
 OVERLAY_BG_RATIO = 0.6       # Background alpha = opacity * ratio (lighter than modules)
 OVERLAY_BLUR_SIGMA = 1.0     # Gaussian blur on overlay for softer edges
-OVERLAY_FEATHER_PX = 40      # Fade-out at overlay borders (px)
+OVERLAY_FEATHER_PX = 16      # Fade-out at overlay borders (1 QR module width)
 
 # ---------------------------------------------------------------------------
 # Quality tags — NO QR tags (QR structure from ControlNet only)
@@ -121,7 +121,7 @@ class EndpointHandler:
 
     def __init__(self, path: str = ""):
         """Load models on endpoint startup."""
-        logger.info("Loading QR Art Generator pipeline v12 (Art+Overlay)...")
+        logger.info("Loading QR Art Generator pipeline v12.1 (Art+Overlay)...")
         start = time.time()
 
         device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -182,7 +182,7 @@ class EndpointHandler:
         self.device = device
         self.dtype = dtype
         elapsed = time.time() - start
-        logger.info(f"Pipeline v12 loaded in {elapsed:.1f}s on {device}")
+        logger.info(f"Pipeline v12.1 loaded in {elapsed:.1f}s on {device}")
 
     def _generate_qr_images(self, data: str):
         """
@@ -212,8 +212,8 @@ class EndpointHandler:
 
         qr_w, qr_h = qr_gray.size
 
-        # Resize to target size
-        if qr_w > QR_TARGET_SIZE or qr_h > QR_TARGET_SIZE:
+        # Always resize to exact target size for consistent alignment
+        if qr_w != QR_TARGET_SIZE or qr_h != QR_TARGET_SIZE:
             qr_gray = qr_gray.resize(
                 (QR_TARGET_SIZE, QR_TARGET_SIZE), Image.NEAREST
             )
@@ -223,6 +223,7 @@ class EndpointHandler:
             logger.info(f"QR resized from {qr_w}x{qr_h} to {QR_TARGET_SIZE}x{QR_TARGET_SIZE}")
 
         # Conditioning: center on gray canvas + pre-blur
+        # Both conditioning and overlay MUST use the same offset for alignment
         conditioning = Image.new("RGB", (QR_CANVAS_SIZE, QR_CANVAS_SIZE), (128, 128, 128))
         offset = (QR_CANVAS_SIZE - QR_TARGET_SIZE) // 2
         conditioning.paste(qr_gray, (offset, offset))
@@ -288,9 +289,9 @@ class EndpointHandler:
             ov_arr[:, :, 3] = (ov_arr[:, :, 3].astype(np.float32) * fade).astype(np.uint8)
             overlay_img = Image.fromarray(ov_arr, "RGBA")
 
-        # Center overlay on full canvas
+        # Center overlay on full canvas — MUST match conditioning offset
         canvas = Image.new("RGBA", (QR_CANVAS_SIZE, QR_CANVAS_SIZE), (0, 0, 0, 0))
-        offset = (QR_CANVAS_SIZE - qr_size) // 2
+        offset = (QR_CANVAS_SIZE - QR_TARGET_SIZE) // 2
         canvas.paste(overlay_img, (offset, offset))
 
         return canvas
@@ -489,7 +490,7 @@ class EndpointHandler:
             "image": result_b64,
             "seed": seed,
             "parameters": {
-                "pipeline": f"{'two' if passes >= 2 else 'single'}-pass-v12-overlay",
+                "pipeline": f"{'two' if passes >= 2 else 'single'}-pass-v12.1-overlay",
                 "passes": passes,
                 "category": category,
                 "p1_monster": p1_monster,