Update core/image_generator.py

core/image_generator.py

@@ -529,10 +529,13 @@ async def generate_images(prompt_or_json, seed: int | None = None, num_images: i
     pipe = load_pipeline()
     images = []
     for i in range(num_images):
-        gen = torch.Generator(device).manual_seed(seed + i) if seed is not None else None
-        try:
-            img = pipe(prompt_or_json, num_inference_steps=30, generator=gen).images[0]
-            img_path = TMP_DIR / f"prompt_{i}.png"
-            img
-::contentReference[oaicite:9]{index=9}
+      gen = torch.Generator(device).manual_seed(seed + i) if seed is not None else None
+      try:
+        img = pipe(prompt_or_json, num_inference_steps=30, generator=gen).images[0]
+        img_path = TMP_DIR / f"prompt_{i}.png"
+        img.save(img_path)
+        images.append(pil_to_base64(img))
+      except Exception as e:
+        print(f"[ImageGen] ⚠️ Failed on image {i}: {e}")
+