Inpainting Lines

app.py

@@ -135,7 +135,7 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
     l_clahe = clahe.apply(l_channel)
     shading_map = l_clahe / 255.0
 
-        # ==========================================================
+    # ==========================================================
     # [PATCHED] TILE PATTERN (soft crossfade) + BLEND
     #    Replace the block from "# Tile pattern" through
     #    "pattern_folded = np.clip(pattern_folded, 0, 1)"
@@ -302,6 +302,45 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
     # END NEW: SEAM-FIX UPSTREAM
     # ================================
 
+        # ================================
+    # POST-PROCESS: SEAM / BLACK-LINE FILL
+    # (place this right before: "Premultiplied → Straight alpha")
+    # ================================
+
+    # Build a temporary straight-RGB for detection (do NOT export this)
+    _alpha_tmp = np.clip(mask_expanded, eps, 1.0)                   # [H,W]
+    _alpha_tmp3 = _alpha_tmp[..., None]                              # [H,W,1]
+    straight_tmp = np.clip(pattern_final / _alpha_tmp3, 0.0, 1.0)    # [H,W,3], float
+
+    # Convert to 8-bit for OpenCV ops
+    straight_u8 = (straight_tmp * 255).astype(np.uint8)
+    gray8 = cv2.cvtColor(straight_u8, cv2.COLOR_RGB2GRAY)
+
+    # Local median to find narrow dark troughs (seams)
+    med8 = cv2.medianBlur(gray8, 5)
+
+    # Heuristics: dark + locally darker than neighbors + inside garment
+    dark_thr = int(0.22 * 255)       # tune: 0.18..0.28
+    delta_thr = int(0.08 * 255)      # tune: 0.06..0.10
+    dark = gray8 < dark_thr
+    contrast = (med8.astype(np.int16) - gray8.astype(np.int16)) > delta_thr
+    inside = (mask_expanded > 0.06)   # avoid true background
+
+    seam_mask = (dark & contrast & inside).astype(np.uint8) * 255
+
+    # Keep seams thin (remove blobs); nudge toward single‑pixel cores
+    k3 = np.ones((3, 3), np.uint8)
+    seam_mask = cv2.morphologyEx(seam_mask, cv2.MORPH_OPEN, k3, iterations=1)
+    seam_mask = cv2.erode(seam_mask, np.ones((2, 2), np.uint8), iterations=1)
+
+    # Inpaint on straight space (Telea = good for thin scratches)
+    inpaint_radius = 3  # try 2..4
+    inpainted_u8 = cv2.inpaint(straight_u8, seam_mask, inpaint_radius, cv2.INPAINT_TELEA)
+    inpainted = inpainted_u8.astype(np.float32) / 255.0
+
+    # Re-premultiply and replace
+
+
     # Premultiplied → Straight alpha
     eps = 1e-6
     # _alpha = np.clip(mask_blurred, eps, 1.0)