Update app.py

app.py

@@ -5,18 +5,35 @@ import io
 import os
 
 # Importa a função melhorada do seu módulo (que já está ajustado )
-from faceless_processing import build_faceless_embroidery_assets
+from faceless_processing import build_faceless_embroidery_assets, LineartConfig
 
 @spaces.GPU
-def process_faceless(input_image, prompt="", negative_prompt=""):
+def process_faceless(input_image, prompt="", negative_prompt="", is_cartoon=False):
     if input_image is None:
         return None, None
 
+    # Configuração base
+    cfg = LineartConfig()
+    
+    if is_cartoon:
+        # Afrouxa as regras para não apagar detalhes do desenho
+        cfg.lineart_threshold = 60
+        cfg.lineart_min_perimeter = 25.0
+        cfg.detect_eyebrows = False
+        
+        # Impede de achar rosto e apagar os olhos/boca do desenho
+        cfg.face_min_size = (9999, 9999) 
+        
+        # Mantém grossinho para bordar!
+        cfg.silhouette_line_thickness = 5
+        cfg.interior_line_thickness = 4
+
     # 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
     try:
         lineart, cutout = build_faceless_embroidery_assets(
             input_image,
+            cfg=cfg,
             use_huggingface=False,   # Desativa Hugging Face (pode ativar se tiver token)
             prompt=prompt or None,
             negative_prompt=negative_prompt or None,
@@ -33,14 +50,15 @@ demo = gr.Interface(
     inputs=[
         gr.Image(type="pil", label="Imagem de Entrada"),
         gr.Textbox(label="Prompt (Opcional)", value=""),
-        gr.Textbox(label="Prompt Negativo (Opcional)", value="")
+        gr.Textbox(label="Prompt Negativo (Opcional)", value=""),
+        gr.Checkbox(label="Modo Cartoon / Desenho (Mantém detalhes e rostos)", value=False)
     ],
     outputs=[
         gr.Image(label="Lineart Final (Transparente)"),
         gr.Image(label="Recorte do Rosto (Sem Fundo)")
     ],
     title="Faceless Embroidery Studio Generator",
-    description="Gera contornos e linearts minimalistas (Faceless) para bordados a partir de retratos.\n\n**Nota:** O rosto é completamente apagado e apenas a silhueta externa é desenhada."
+    description="Gera contornos e linearts minimalistas (Faceless) para bordados a partir de retratos.\n\n**Nota:** O rosto é completamente apagado e apenas a silhueta externa é desenhada (exceto no Modo Cartoon)."
 )
 
 if __name__ == "__main__":