1580158
/
OneButtonPrompt
mirror of https://github.com/AIrjen/OneButtonPrompt
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

import/regex fix 

fix: Changed to use relative imports
* The current import method causes import failures if the path name of a custom node changes (import failures occur when a new manager is introduced)

fix: Incorrect use of `\` in regex strings
* From Python 3.12 onwards, using the `\` character without explicitly using regex strings like r'\' is treated as a syntax error
pull/212/head
Dr.Lt.Data 2024-07-06 10:31:03 +09:00
parent 421853ea2e
commit 2ed987364b
4 changed files with 27 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
OneButtonPromptNodes.py
View File

@ -10,10 +10,10 @@ onebuttonprompt_path = os.path.join(custom_nodes_path, "OneButtonPrompt")
sys.path.append(onebuttonprompt_path)
from build_dynamic_prompt import *
from csv_reader import *
from .build_dynamic_prompt import *
from .csv_reader import *
from one_button_presets import OneButtonPresets
from .one_button_presets import OneButtonPresets
OBPresets = OneButtonPresets()
allpresets = [OBPresets.RANDOM_PRESET_OBP] + list(OBPresets.opb_presets.keys())

2
__init__.py
View File

@ -6,6 +6,6 @@ custom_nodes_path = os.path.join(folder_paths.base_path, "custom_nodes")
onebuttonprompt_path = os.path.join(custom_nodes_path, "OneButtonPrompt")
sys.path.append(onebuttonprompt_path)
from OneButtonPromptNodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
from .OneButtonPromptNodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']

44
build_dynamic_prompt.py
View File

@ -1,9 +1,9 @@
import random
import re
from csv_reader import *
from random_functions import *
from one_button_presets import OneButtonPresets
from superprompter.superprompter import *
from .csv_reader import *
from .random_functions import *
from .one_button_presets import OneButtonPresets
from .superprompter.superprompter import *
OBPresets = OneButtonPresets()
@ -4947,7 +4947,7 @@ def cleanup(completeprompt, advancedprompting, insanitylevel = 5):
completeprompt = completeprompt.replace("|", " ")
# sometimes if there are not enough artist, we get left we things formed as (:1.2)
completeprompt = re.sub('\(\:\d+\.\d+\)', '', completeprompt)
completeprompt = re.sub(r'\(\:\d+\.\d+\)', '', completeprompt)
# lets also remove some wierd stuff on lower insanitylevels
if(insanitylevel < 7):
@ -4955,16 +4955,16 @@ def cleanup(completeprompt, advancedprompting, insanitylevel = 5):
completeprompt = completeprompt.replace("fluorescent", " ")
# all cleanup steps moved here
completeprompt = re.sub('\[ ', '[', completeprompt)
completeprompt = re.sub('\[,', '[', completeprompt)
completeprompt = re.sub(' \]', ']', completeprompt)
completeprompt = re.sub(' \|', '|', completeprompt)
#completeprompt = re.sub(' \"', '\"', completeprompt)
#completeprompt = re.sub('\" ', '\"', completeprompt)
completeprompt = re.sub('\( ', '(', completeprompt)
completeprompt = re.sub(' \(', '(', completeprompt)
completeprompt = re.sub('\) ', ')', completeprompt)
completeprompt = re.sub(' \)', ')', completeprompt)
completeprompt = re.sub(r'\[ ', '[', completeprompt)
completeprompt = re.sub(r'\[,', '[', completeprompt)
completeprompt = re.sub(r' \]', ']', completeprompt)
completeprompt = re.sub(r' \|', '|', completeprompt)
#completeprompt = re.sub(r' \"', '\"', completeprompt)
#completeprompt = re.sub(r'\" ', '\"', completeprompt)
completeprompt = re.sub(r'\( ', '(', completeprompt)
completeprompt = re.sub(r' \(', '(', completeprompt)
completeprompt = re.sub(r'\) ', ')', completeprompt)
completeprompt = re.sub(r' \)', ')', completeprompt)
completeprompt = re.sub(' :', ':', completeprompt)
completeprompt = re.sub(',::', '::', completeprompt)
@ -4978,7 +4978,7 @@ def cleanup(completeprompt, advancedprompting, insanitylevel = 5):
completeprompt = re.sub(' ,', ',', completeprompt)
completeprompt = re.sub(' ,', ',', completeprompt)
completeprompt = re.sub(' ,', ',', completeprompt)
completeprompt = re.sub(',\(', ', (', completeprompt)
completeprompt = re.sub(r',\(', ', (', completeprompt)
@ -5048,17 +5048,17 @@ def cleanup(completeprompt, advancedprompting, insanitylevel = 5):
completeprompt = re.sub(' mans', ' men', completeprompt)
completeprompt = re.sub(' Womans,', ' Women', completeprompt)
completeprompt = re.sub(' womans,', ' women,', completeprompt)
completeprompt = re.sub('\(Mans', '(Men,', completeprompt)
completeprompt = re.sub('\(mans', '(men', completeprompt)
completeprompt = re.sub('\(Womans', '(Women', completeprompt)
completeprompt = re.sub('\(womans', '(women', completeprompt)
completeprompt = re.sub(r'\(Mans', '(Men,', completeprompt)
completeprompt = re.sub(r'\(mans', '(men', completeprompt)
completeprompt = re.sub(r'\(Womans', '(Women', completeprompt)
completeprompt = re.sub(r'\(womans', '(women', completeprompt)
completeprompt = re.sub('-sameothersubject-', 'it', completeprompt)
completeprompt = re.sub('-samehumansubject-', 'the person', completeprompt)
completeprompt = re.sub('(?<!\()\s?\(', ' (', completeprompt)
completeprompt = re.sub('\)(?![\s)])', ') ', completeprompt)
completeprompt = re.sub(r'(?<!\()\s?\(', ' (', completeprompt)
completeprompt = re.sub(r'\)(?![\s)])', ') ', completeprompt)
# Move the extracted LoRA's to the end of completeprompt
#completeprompt += " " + " ".join(allLoRA)

2
superprompter/superprompter.py
View File

@ -3,7 +3,7 @@ import os
import random
import torch
from transformers import T5Tokenizer, T5ForConditionalGeneration
from superprompter.download_models import download_models
from .download_models import download_models
global tokenizer, model
script_dir = os.path.dirname(os.path.abspath(__file__)) # Script directory