68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
import re
|
|
|
|
class NegativeWildcardProcessor:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"positive" : ("STRING", {"forceInput":True, "default": "", "multiline": True}),
|
|
"negative" : ("STRING", {"forceInput":True, "default": "", "multiline": True})
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ('STRING','STRING',)
|
|
RETURN_NAMES = ('text_positive','text_negative',)
|
|
FUNCTION = "process_negative_wildcards"
|
|
CATEGORY = "Dynamic Prompts"
|
|
|
|
def process_negative_wildcards(self, positive, negative):
|
|
# Pattern to match <! anything !>
|
|
pattern = r"<!(.*?)!>"
|
|
|
|
# Find all matches of the pattern in the positive string
|
|
matches = re.findall(pattern, positive)
|
|
|
|
# Remove all occurrences of the pattern from the positive string
|
|
text_positive = re.sub(r"<!.*?!>", "", positive)
|
|
text_positive = re.sub(r"<lora.*?>", "", text_positive)
|
|
|
|
# Append all matches to the end of the negative string
|
|
text_negative = negative
|
|
for match in matches:
|
|
text_negative += " " + match
|
|
|
|
return (text_positive, text_negative,)
|
|
|
|
class CustomTokenPreprocessor:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"token" : ("STRING", {"default": "", "multiline": False}),
|
|
"source" : ("STRING", {"forceInput":True, "default": "", "multiline": True}),
|
|
"target" : ("STRING", {"forceInput":True, "default": "", "multiline": True})
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ('STRING','STRING',)
|
|
RETURN_NAMES = ('processed_source','processed_target',)
|
|
FUNCTION = "process_custom_token"
|
|
CATEGORY = "Dynamic Prompts"
|
|
|
|
def process_custom_token(self, token, source, target):
|
|
# Pattern to match <! anything !>
|
|
pattern = r"<" + token + "(.*?)" + token + ">"
|
|
|
|
# Find all matches of the pattern in the positive string
|
|
matches = re.findall(pattern, source)
|
|
|
|
# Remove all occurrences of the pattern from the positive string
|
|
processed_source = re.sub(r"<" + token + ".*?" + token + ">", "", source)
|
|
|
|
# Append all matches to the end of the negative string
|
|
processed_target = target
|
|
for match in matches:
|
|
processed_target += " " + match
|
|
|
|
return (processed_source, processed_target,)
|