Update app.py

app.py

@@ -5,9 +5,14 @@ import gradio as gr
 from pathlib import Path
 import subprocess, json, tempfile, sys
 
+# Chemin vers ComfyUI et le workflow JSON
 COMFY_DIR = Path("ComfyUI")
 WORKFLOW_JSON = Path("workflow_api.json")
 
+# S'assurer que le dossier de logs existe et est visible par Hugging Face
+LOGS_DIR = Path("/app/logs")
+LOGS_DIR.mkdir(parents=True, exist_ok=True)
+
 def patch_workflow(wf, pic_path, face_path, prompt):
     for node in wf.values():
         if node.get("_meta", {}).get("title") == "Picture to swap":
@@ -28,10 +33,12 @@ def run(picture, face, positive):
     with open(WORKFLOW_JSON) as f:
         wf = json.load(f)
     wf = patch_workflow(wf, pic_path, face_path, positive)
+
     tmp_wf = tmpdir / "wf.json"
     with open(tmp_wf, "w") as f:
         json.dump(wf, f)
 
+    # Lancement de ComfyUI
     result = subprocess.run([
         sys.executable, "main.py",
         "--disable-auto-launch",
@@ -40,17 +47,26 @@ def run(picture, face, positive):
         "--workflow", str(tmp_wf)
     ], cwd=COMFY_DIR, capture_output=True, text=True)
 
+    # Sauvegarder les logs dans un fichier visible
+    log_file = LOGS_DIR / "comfy.txt"
+    with open(log_file, "w") as f:
+        f.write("=== STDOUT ===\n")
+        f.write(result.stdout)
+        f.write("\n=== STDERR ===\n")
+        f.write(result.stderr)
+
     if result.returncode != 0:
-        print("=== STDOUT ===")
-        print(result.stdout)
-        print("=== STDERR ===")
-        print(result.stderr)
-        raise RuntimeError("ComfyUI failed to run")
+        raise RuntimeError("ComfyUI failed to run — voir comfy.txt dans Files.")
 
+    # Vérifie la sortie image
     out = list(tmpdir.glob("*.png"))
     if not out:
         raise RuntimeError("No output image generated.")
-    return out[0]
+
+    final_img = LOGS_DIR / "output.png"
+    out[0].replace(final_img)
+
+    return final_img, log_file
 
 demo = gr.Interface(
     fn=run,
@@ -60,9 +76,12 @@ demo = gr.Interface(
         gr.Textbox(label="Positive prompt", lines=3,
                    placeholder="portrait of a woman, detailed face...")
     ],
-    outputs=gr.Image(type="filepath", label="Swapped"),
+    outputs=[
+        gr.Image(type="filepath", label="Swapped"),
+        gr.File(label="ComfyUI logs")
+    ],
     title="ComfyUI InstantID Face Swap",
     allow_flagging="never"
 )
 
-demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=True)
+demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=True)