From d24c74e34a38d10342efc1412b67c94d36c070e4 Mon Sep 17 00:00:00 2001 From: LenAnderson Date: Mon, 20 Nov 2023 12:51:36 +0000 Subject: [PATCH] update comfy generate to use saved workflows --- public/scripts/extensions/stable-diffusion/index.js | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/public/scripts/extensions/stable-diffusion/index.js b/public/scripts/extensions/stable-diffusion/index.js index 52f897fb6..968863fcc 100644 --- a/public/scripts/extensions/stable-diffusion/index.js +++ b/public/scripts/extensions/stable-diffusion/index.js @@ -2060,7 +2060,18 @@ async function generateComfyImage(prompt) { 'height', ]; - let workflow = extension_settings.sd.comfy_workflow.replace('"%prompt%"', JSON.stringify(prompt)); + const workflowResponse = await fetch('/api/sd/comfy/workflow', { + method: 'POST', + headers: getRequestHeaders(), + body: JSON.stringify({ + file_name: extension_settings.sd.comfy_workflow, + }), + }); + if (!workflowResponse.ok) { + const text = await workflowResponse.text(); + toastr.error(`Failed to load workflow.\n\n${text}`); + } + let workflow = (await workflowResponse.json()).replace('"%prompt%"', JSON.stringify(prompt)); workflow = workflow.replace('"%seed%"', JSON.stringify(Math.round(Math.random() * Number.MAX_SAFE_INTEGER))); placeholders.forEach(ph => { workflow = workflow.replace(`"%${ph}%"`, JSON.stringify(extension_settings.sd[ph]));