Fix handling images without subfolder

2025-03-15 09:26:45 +01:00
parent d3f9a76763
commit 6b362fddb6
1 changed files with 2 additions and 1 deletions
@@ -63,11 +63,12 @@ class ComfyUIClient:

        history = (await self.get_history(prompt_id))[prompt_id]
        for node_id, node_output in history["outputs"].items():
+            print(node_output)
            images_output = []
            if "images" in node_output:
                for image in node_output["images"]:
                    image_data = await self.get_image(
-                        image["filename"], image["subfolder"], image["type"]
+                        image["filename"], image.get("subfolder", ""), image["type"]
                    )
                    images_output.append(image_data)
            output_images[node_id] = images_output