81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
import json
|
|
from jsonschema import validate
|
|
|
|
from .static_variables import (
|
|
empty_prompt,
|
|
invalid_prompt,
|
|
jsonprompt_schemafile,
|
|
promptTableHeaders,
|
|
default_total_outpaints,
|
|
)
|
|
prompts_keys = (default_total_outpaints, default_total_outpaints)
|
|
|
|
def process_keys(data):
|
|
#data = json.loads(txt)['data']
|
|
keys = [int(sublist[0]) for sublist in data]
|
|
max_key = max(keys)
|
|
num_keys = len(keys)
|
|
return (max_key, num_keys)
|
|
|
|
def completeOptionals(j):
|
|
if isinstance(j, dict):
|
|
# Remove header information, user dont pimp our ui
|
|
if "prompts" in j:
|
|
if "headers" in j["prompts"]:
|
|
del j["prompts"]["headers"]
|
|
j["prompts"]["headers"]=promptTableHeaders[0]
|
|
|
|
if "negPrompt" not in j:
|
|
j["negPrompt"]=""
|
|
|
|
if "prePrompt" not in j:
|
|
if "commonPromptPrefix" in j:
|
|
j["prePrompt"]=j["commonPromptPrefix"]
|
|
else:
|
|
j["prePrompt"]=""
|
|
|
|
if "postPrompt" not in j:
|
|
if "commonPromptSuffix" in j:
|
|
j["postPrompt"]=j["commonPromptSuffix"]
|
|
else:
|
|
j["postPrompt"]=""
|
|
|
|
if "audioFileName" not in j:
|
|
j["audioFileName"]= None
|
|
|
|
if "seed" not in j:
|
|
j["seed"]= -1
|
|
return j
|
|
|
|
def validatePromptJson_throws(data):
|
|
with open(jsonprompt_schemafile, "r") as s:
|
|
schema = json.load(s)
|
|
try:
|
|
validate(instance=data, schema=schema)
|
|
|
|
except Exception:
|
|
raise Exception("Your prompts are not schema valid.")
|
|
|
|
return completeOptionals(data)
|
|
|
|
|
|
def readJsonPrompt(txt, returnFailPrompt=False):
|
|
if not txt:
|
|
return empty_prompt
|
|
|
|
try:
|
|
jpr = json.loads(txt)
|
|
except Exception:
|
|
if returnFailPrompt:
|
|
print (f"Infinite Zoom: Corrupted Json structure: {txt[:24]} ...")
|
|
return invalid_prompt
|
|
raise (f"Infinite Zoom: Corrupted Json structure: {txt[:24]} ...")
|
|
|
|
try:
|
|
return validatePromptJson_throws(jpr)
|
|
except Exception:
|
|
if returnFailPrompt:
|
|
return invalid_prompt
|
|
pass
|
|
|