298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed to store resolved prompt in workflow info

Browse Source
This commit is contained in:
pythongosssss 2023-03-03 18:40:50 +00:00
parent 4ef4cf913f
commit e14b79c14e
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
web/extensions/core/dynamicPrompts.js
View File

@ -6,9 +6,6 @@ import { app } from "../../scripts/app.js";
app.registerExtension({
name: "Comfy.DynamicPrompts",
nodeCreated(node) {
// TODO: Change this to replace the value and restore it after posting
if (node.widgets) {
// Locate dynamic prompt text widgets
// Include any widgets with dynamicPrompts set to true, and customtext
@ -17,7 +14,7 @@ app.registerExtension({
);
for (const widget of widgets) {
// Override the serialization of the value to resolve dynamic prompts for all widgets supporting it in this node
widget.serializeValue = () => {
widget.serializeValue = (workflowNode, widgetIndex) => {
let prompt = widget.value;
while (prompt.replace("\\{", "").includes("{") && prompt.replace("\\}", "").includes("}")) {
const startIndex = prompt.replace("\\{", "00").indexOf("{");
@ -32,6 +29,9 @@ app.registerExtension({
prompt = prompt.substring(0, startIndex) + randomOption + prompt.substring(endIndex + 1);
}
// Overwrite the value in the serialized workflow pnginfo
workflowNode.widgets_values[widgetIndex] = prompt;
return prompt;
};
}

5
web/scripts/app.js
View File

@ -589,9 +589,10 @@ class ComfyApp {
// Store all widget values
if (widgets) {
for (const widget of widgets) {
for (const i in widgets) {
const widget = widgets[i];
if (!widget.options || widget.options.serialize !== false) {
inputs[widget.name] = widget.serializeValue ? widget.serializeValue() : widget.value;
inputs[widget.name] = widget.serializeValue ? widget.serializeValue(n, i) : widget.value;
}
}
}