mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-11 02:15:17 +00:00
Add comment support to dynamic prompts nodes
This commit is contained in:
parent
29c50954ea
commit
eed4f62cc5
@ -3,6 +3,13 @@ import { app } from "../../scripts/app.js";
|
|||||||
// Allows for simple dynamic prompt replacement
|
// Allows for simple dynamic prompt replacement
|
||||||
// Inputs in the format {a|b} will have a random value of a or b chosen when the prompt is queued.
|
// Inputs in the format {a|b} will have a random value of a or b chosen when the prompt is queued.
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Strips C-style line and block comments from a string
|
||||||
|
*/
|
||||||
|
function stripComments(str) {
|
||||||
|
return str.replace(/\/\*[\s\S]*?\*\/|\/\/.*/g,'');
|
||||||
|
}
|
||||||
|
|
||||||
app.registerExtension({
|
app.registerExtension({
|
||||||
name: "Comfy.DynamicPrompts",
|
name: "Comfy.DynamicPrompts",
|
||||||
nodeCreated(node) {
|
nodeCreated(node) {
|
||||||
@ -15,7 +22,7 @@ app.registerExtension({
|
|||||||
for (const widget of widgets) {
|
for (const widget of widgets) {
|
||||||
// Override the serialization of the value to resolve dynamic prompts for all widgets supporting it in this node
|
// Override the serialization of the value to resolve dynamic prompts for all widgets supporting it in this node
|
||||||
widget.serializeValue = (workflowNode, widgetIndex) => {
|
widget.serializeValue = (workflowNode, widgetIndex) => {
|
||||||
let prompt = widget.value;
|
let prompt = stripComments(widget.value);
|
||||||
while (prompt.replace("\\{", "").includes("{") && prompt.replace("\\}", "").includes("}")) {
|
while (prompt.replace("\\{", "").includes("{") && prompt.replace("\\}", "").includes("}")) {
|
||||||
const startIndex = prompt.replace("\\{", "00").indexOf("{");
|
const startIndex = prompt.replace("\\{", "00").indexOf("{");
|
||||||
const endIndex = prompt.replace("\\}", "00").indexOf("}");
|
const endIndex = prompt.replace("\\}", "00").indexOf("}");
|
||||||
|
Loading…
Reference in New Issue
Block a user