Update app.py

app.py

@@ -3,42 +3,29 @@ import gradio as gr
 from PIL import Image
 import io
 import os
-import tempfile
 
-# Importa a função melhorada do seu módulo (que já está ajustado)
-from faceless_processing import build_faceless_embroidery_assets, LineartConfig
+# Importa a função melhorada do seu módulo (que já está ajustado )
+from faceless_processing import build_faceless_embroidery_assets
 
 @spaces.GPU
 def process_faceless(input_image, prompt="", negative_prompt=""):
     if input_image is None:
         return None, None
 
-    # NÃO REDIMENSIONAMOS A IMAGEM – a função interna já lida com tamanhos variados.
+    # 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:
-        cfg = LineartConfig(
-            prompt=prompt or "technical line art, clean vector, simple continuous lines, white background",
-            negative_prompt=negative_prompt or "shading, realism, messy lines, noisy, double lines, background clutter"
+        lineart, cutout = build_faceless_embroidery_assets(
+            input_image,
+            use_huggingface=False,   # Desativa Hugging Face (pode ativar se tiver token)
+            prompt=prompt or None,
+            negative_prompt=negative_prompt or None,
         )
-        lineart, cutout = build_faceless_embroidery_assets(input_image, cfg)
-        
-        # Save to temp files to preserve RGBA transparency channels
-        lineart_path = os.path.join(tempfile.gettempdir(), f"lineart_{os.getpid()}_{os.urandom(4).hex()}.png")
-        cutout_path = os.path.join(tempfile.gettempdir(), f"cutout_{os.getpid()}_{os.urandom(4).hex()}.png")
-        
-        lineart.save(lineart_path, "PNG")
-        if cutout:
-            cutout.save(cutout_path, "PNG")
-            return lineart_path, cutout_path
-        else:
-            return lineart_path, None
+        return lineart, cutout
     except Exception as e:
         print(f"[ERRO] process_faceless: {e}")
-        # Save original to temp file to be safe
-        orig_path = os.path.join(tempfile.gettempdir(), f"orig_{os.getpid()}_{os.urandom(4).hex()}.png")
-        input_image.save(orig_path, "PNG")
-        return orig_path, None
-
+        # Fallback: retorna a imagem original e uma mensagem de erro
+        return input_image, None
 
 # Interface Gradio
 demo = gr.Interface(
@@ -49,13 +36,12 @@ demo = gr.Interface(
         gr.Textbox(label="Prompt Negativo (Opcional)", value="")
     ],
     outputs=[
-        gr.Image(label="Lineart Final (Transparente)", image_mode="RGBA", format="png"),
-        gr.Image(label="Recorte do Rosto (Sem Fundo)", image_mode="RGBA", format="png")
+        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."
 )
 
-
 if __name__ == "__main__":
     demo.launch()