Add custom styles and wildcards example for styles selector

This commit is contained in:
yolain
2024-11-08 13:16:00 +08:00
parent 1be8fa596c
commit 918bd85865
2 changed files with 24 additions and 1 deletions
+23
View File
@@ -32,6 +32,12 @@ if os.path.exists(wildcards_path):
read_wildcard_dict(wildcards_path)
else:
os.mkdir(wildcards_path)
# Add custom wildcards example
example_path = os.path.join(wildcards_path, "example.txt")
if not os.path.exists(example_path):
with open(example_path, 'w') as f:
text = "blue\nred\nyellow\ngreen\nbrown\npink\npurple\norange\nblack\nwhite"
f.write(text)
#Styles
styles_path = os.path.join(os.path.dirname(__file__), "styles")
@@ -43,6 +49,23 @@ else:
os.mkdir(styles_path)
os.mkdir(samples_path)
# Add custom styles example
example_path = os.path.join(styles_path, "your_styles.json.example")
if not os.path.exists(example_path):
import json
data = [
{
"name": "Example Style",
"name_cn": "示例样式",
"prompt": "(masterpiece), (best quality), (ultra-detailed), {prompt} ",
"negative_prompt": "text, watermark, logo"
},
]
# Write to file
with open(example_path, 'w') as f:
json.dump(data, f, indent=4, ensure_ascii=False)
# Model thumbnails
from .py.libs.add_resources import add_static_resource
from .py.libs.model import easyModelManager
+1 -1
View File
@@ -143,7 +143,7 @@ class stylesPromptSelector:
styles_dir = FOOOCUS_STYLES_DIR
for file_name in os.listdir(styles_dir):
file = os.path.join(styles_dir, file_name)
if os.path.isfile(file) and file_name.endswith(".json") and "styles" in file_name.split(".")[0]:
if os.path.isfile(file) and file_name.endswith(".json"):
styles.append(file_name.split(".")[0])
return {
"required": {