From 6b362fddb6075810a4b6d734ef3b44fb8cf12e5c Mon Sep 17 00:00:00 2001 From: IfnotFr Date: Sat, 15 Mar 2025 09:26:45 +0100 Subject: [PATCH] Fix handling images without subfolder --- comfyui_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfyui_client.py b/comfyui_client.py index f052e7f..ad6d22a 100644 --- a/comfyui_client.py +++ b/comfyui_client.py @@ -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