Update app.py

app.py

@@ -46,7 +46,7 @@ def infer(source_img, prompt, guide, steps, seed, strength):
     source_image = Image.open(source_img).convert("RGB")
     source_image = source_image.resize((512, 512), Image.Resampling.LANCZOS)
 
-    images_list = img_pipe(
+    result = img_pipe(
         [prompt],
         image=source_image,
         strength=strength,
@@ -54,8 +54,20 @@ def infer(source_img, prompt, guide, steps, seed, strength):
         num_inference_steps=steps,
         generator=generator
     )
-    print(images_list["images"])
-    return images_list["images"]
+
+    output_images = result["images"]
+    output_paths = []
+
+    for idx, img in enumerate(output_images):
+        filename = f"output_{seed}_{idx}.png"
+        save_path = os.path.join("outputs", filename)
+        os.makedirs("outputs", exist_ok=True)
+        img.save(save_path)
+        print(f"Saved image to: {save_path}")
+        output_paths.append(save_path)
+
+    # Optional: return paths or Gradio can render them too
+    return output_images
 
 print("Great sylvain ! Everything is working fine !")