Improve faceless lineart detail for Space

app.py

@@ -15,6 +15,30 @@ def process_faceless(input_image, prompt="", negative_prompt="", is_cartoon=Fals
     # Configuração base
     cfg = LineartConfig()
     
+    # The detail preset applies to portraits too. Previously it only affected
+    # cartoon mode, which made the public controls misleading.
+    if detail == "detailed":
+        cfg.lineart_threshold = 72
+        cfg.lineart_min_perimeter = 20.0
+        cfg.lineart_min_area = 5.0
+        cfg.min_line_distance = 4
+        cfg.silhouette_line_thickness = 5
+        cfg.interior_line_thickness = 4
+    elif detail == "soft":
+        cfg.lineart_threshold = 100
+        cfg.lineart_min_perimeter = 36.0
+        cfg.lineart_min_area = 10.0
+        cfg.min_line_distance = 7
+        cfg.silhouette_line_thickness = 4
+        cfg.interior_line_thickness = 3
+    else:
+        cfg.lineart_threshold = 88
+        cfg.lineart_min_perimeter = 28.0
+        cfg.lineart_min_area = 8.0
+        cfg.min_line_distance = 5
+        cfg.silhouette_line_thickness = 5
+        cfg.interior_line_thickness = 4
+
     if is_cartoon:
         cfg.is_cartoon = True
         # Afrouxa as regras para não apagar detalhes do desenho, mas com limites seguros para bordar
@@ -59,8 +83,7 @@ def process_faceless(input_image, prompt="", negative_prompt="", is_cartoon=Fals
         cfg.face_min_size = (9999, 9999)
         
         # Mantém grossinho para bordar!
-        cfg.silhouette_line_thickness = 5
-        cfg.interior_line_thickness = 4
+        cfg.stroke_dilate_iterations = 0
 
     # NÃO REDIMENSIONAMOS A IMAGEM – a função interna ( embroidery_processing.py) já lida com tamanhos variados. 
     # Gera lineart e cutout usando a função otimizada
@@ -97,4 +120,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()

faceless_processing.py

@@ -4,6 +4,7 @@ from PIL import Image, ImageFilter
 from dataclasses import dataclass
 from typing import Tuple, Optional
 import io
+
 import base64
 import os
 import urllib.request
@@ -144,16 +145,21 @@ class LineartConfig:
     lineart_coarse: bool = True
     
     # AUMENTADO: Força bruta para ignorar detalhes fracos
-    lineart_threshold: int = 115
+    # Lower threshold keeps subtle clothing/hair contours that were previously
+    # discarded before they could be converted into embroidery lines.
+    lineart_threshold: int = 88
     
     # REDUZIDO: Permite traços de corpos menores (crianças em grupos)
-    lineart_min_perimeter: float = 80.0
-    lineart_min_area: float = 22.0
-    lineart_morph_open: bool = True
+    lineart_min_perimeter: float = 28.0
+    lineart_min_area: float = 8.0
+    lineart_morph_open: bool = False
     force_canny: bool = False
     
     interior_erode_size: int = 1
-    interior_line_thickness: int = 3
+    interior_line_thickness: int = 4
+    # Extra dilation is deliberately opt-in. The old unconditional dilation
+    # made independent contours merge into thick blobs at embroidery scale.
+    stroke_dilate_iterations: int = 0
 
     # ---- Canny fallback parametros ----
     canny_low: int = 50
@@ -688,8 +694,10 @@ def build_interior_edges(rgb: np.ndarray, subject_mask: np.ndarray, cfg: Lineart
                 min_area=min_area
             )
 
-            kernel_dilate = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
-            filtered = cv2.dilate(filtered, kernel_dilate, iterations=1)
+            dilate_iterations = max(0, int(getattr(cfg, 'stroke_dilate_iterations', 0)))
+            if dilate_iterations:
+                kernel_dilate = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
+                filtered = cv2.dilate(filtered, kernel_dilate, iterations=dilate_iterations)
             filtered = cv2.GaussianBlur(filtered, (3, 3), 0)
             
             _, filtered = cv2.threshold(filtered, 127, 255, cv2.THRESH_BINARY)
@@ -821,8 +829,10 @@ def _build_interior_edges_canny(rgb: np.ndarray, subject_mask: np.ndarray, cfg:
         shape=edges.shape
     )
     
-    kernel_dilate = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
-    filtered = cv2.dilate(filtered, kernel_dilate, iterations=1)
+    dilate_iterations = max(0, int(getattr(cfg, 'stroke_dilate_iterations', 0)))
+    if dilate_iterations:
+        kernel_dilate = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (3, 3))
+        filtered = cv2.dilate(filtered, kernel_dilate, iterations=dilate_iterations)
     return filtered
 
 
@@ -1144,11 +1154,6 @@ def remove_background_subject(img: Image.Image) -> Image.Image:
 def make_faceless_cutout(img: Image.Image) -> Image.Image:
     return remove_background_subject(img)
 
-
-
-
-
-
 def build_faceless_embroidery_assets(img: Image.Image, cfg: Optional[LineartConfig] = None, **kwargs) -> Tuple[Image.Image, Image.Image]:
     if cfg is None:
         cfg = LineartConfig()
@@ -1158,4 +1163,4 @@ def build_faceless_embroidery_assets(img: Image.Image, cfg: Optional[LineartConf
         cfg.negative_prompt = kwargs["negative_prompt"]
     lineart = make_faceless_lineart(img, cfg)
     cutout = make_faceless_cutout(img)
-    return lineart, cutout
+    return lineart, cutout