string split from right

This commit is contained in:
billwuhao
2025-07-07 16:14:15 +08:00
parent d635a83356
commit 1191d4d963
2 changed files with 38 additions and 1 deletions
+37
View File
@@ -365,16 +365,53 @@ class LoadImageAndPromptFromURL:
img, _ = pil2tensor(img)
return (img, prompt, imgurl)
class StringEditNodeOBP:
@classmethod
def INPUT_TYPES(cls):
return {
"required": {
"text": ("STRING", {"forceInput": True}),
},
"optional": {
"paste_edit_text": ("STRING", {"multiline": True, "default": "",}),
"split_tag": ("STRING", {"default": "",}),
"form_right": ("BOOLEAN", {"default": False}),
},
}
RETURN_TYPES = ("STRING", "STRING", "STRING", "STRING", "STRING")
RETURN_NAMES = ("text", "split_1", "split_2", "edited_text", "merged_text")
FUNCTION = "edit_text"
CATEGORY = "🎤MW/MW-Audio-Tools"
OUTPUT_NODE = False
def edit_text(self, text, split_tag="", paste_edit_text="", form_right=False):
str_1, str_2, str_3, str_4, str_5 = text, "", "", paste_edit_text, ""
str_5 = text + paste_edit_text
if split_tag != "" and text.find(split_tag) > 0:
if form_right:
parts = text.rsplit(split_tag, 1)
str_2, str_3 = parts[0].strip(), parts[1].strip()
else:
parts = text.split(split_tag, 1)
str_2, str_3 = parts[0].strip(), parts[1].strip()
return (str_1, str_2, str_3, str_4, str_5)
NODE_CLASS_MAPPINGS = {
"LoadPrompt": LoadPrompt,
"LoadImageFromURL": LoadImageFromURL,
"LoadImageAndPromptFromURL": LoadImageAndPromptFromURL,
"StringEditNodeOBP": StringEditNodeOBP,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"LoadPrompt": "Load Prompt",
"LoadImageFromURL": "Load Image From URL",
"LoadImageAndPromptFromURL": "Load Image And Prompt From URL",
"StringEditNodeOBP": "String Edit"
}
+1 -1
View File
@@ -1,7 +1,7 @@
[project]
name = "onebuttonprompt-mw"
description = "A node that assists in one click generation of prompts (for image and video generation, etc.) in Comfyui."
version = "1.0.1"
version = "1.0.2"
license = {file = "LICENSE"}
[project.urls]