|
|
import json |
|
|
from os.path import exists |
|
|
|
|
|
from shared import path_manager |
|
|
|
|
|
DEFAULT_SETTINGS = { |
|
|
"advanced_mode": False, |
|
|
"image_number": 1, |
|
|
"seed_random": True, |
|
|
"seed": 0, |
|
|
"style": ["Style: sai-cinematic"], |
|
|
"prompt": "", |
|
|
"negative_prompt": "", |
|
|
"performance": "Speed", |
|
|
"resolution": "1152x896 (4:3)", |
|
|
"base_model": path_manager.default_model_names["default_base_model_name"], |
|
|
"lora_1_model": "None", |
|
|
"lora_1_weight": path_manager.default_model_names["default_lora_weight"], |
|
|
"lora_2_model": "None", |
|
|
"lora_2_weight": path_manager.default_model_names["default_lora_weight"], |
|
|
"lora_3_model": "None", |
|
|
"lora_3_weight": path_manager.default_model_names["default_lora_weight"], |
|
|
"lora_4_model": "None", |
|
|
"lora_4_weight": path_manager.default_model_names["default_lora_weight"], |
|
|
"lora_5_model": "None", |
|
|
"lora_5_weight": path_manager.default_model_names["default_lora_weight"], |
|
|
"theme": "None", |
|
|
"auto_negative_prompt": False, |
|
|
"OBP_preset": "Standard", |
|
|
"hint_chance": 25, |
|
|
"clip_g": "clip_g.safetensors", |
|
|
"clip_l": "clip_l.safetensors", |
|
|
"clip_t5": "t5-v1_1-xxl-encoder-Q3_K_S.gguf", |
|
|
} |
|
|
|
|
|
|
|
|
def load_settings(): |
|
|
if exists("settings/settings.json"): |
|
|
with open("settings/settings.json") as f: |
|
|
settings = json.load(f) |
|
|
else: |
|
|
settings = {} |
|
|
|
|
|
|
|
|
changed = False |
|
|
for key, value in DEFAULT_SETTINGS.items(): |
|
|
if key not in settings: |
|
|
settings[key] = value |
|
|
changed = True |
|
|
|
|
|
|
|
|
if not isinstance(settings["style"], list): |
|
|
settings["style"] = [] |
|
|
|
|
|
if changed: |
|
|
with open("settings/settings.json", "w") as f: |
|
|
json.dump(settings, f, indent=2) |
|
|
|
|
|
return settings |
|
|
|
|
|
|
|
|
default_settings = load_settings() |
|
|
|
