AIrjen
fe0cb3c70d
Various list work and extensions Extended events by a lot Changed ethnicity to cultures Made improvements in prompt output and logic Fixed minor bugs in rare cases
94 lines
2.5 KiB
Python
94 lines
2.5 KiB
Python
import json
|
|
import requests
|
|
import io
|
|
import base64
|
|
import uuid
|
|
from PIL import Image, PngImagePlugin
|
|
|
|
def call_txt2img(passingprompt,ratio,upscale,debugmode):
|
|
|
|
#set the prompt!
|
|
prompt = passingprompt
|
|
|
|
#rest of prompt things
|
|
sampler_index = "DPM2 Karras"
|
|
steps = "20"
|
|
if(debugmode==1):
|
|
steps="10"
|
|
cfg_scale = "7"
|
|
|
|
#size
|
|
if(ratio=='wide'):
|
|
width = "768"
|
|
height = "512"
|
|
elif(ratio=='portrait'):
|
|
width = "512"
|
|
height = "768"
|
|
elif(ratio=='ultrawide'):
|
|
width = "1280"
|
|
height = "360"
|
|
else:
|
|
width = "512"
|
|
height = "512"
|
|
#upscaler
|
|
enable_hr = upscale
|
|
if(debugmode==1):
|
|
enable_hr="False"
|
|
denoising_strength = "0.35"
|
|
hr_scale = "2"
|
|
hr_upscaler = "4x-UltraSharp"
|
|
hr_second_pass_steps = str(round(int(steps)/2))
|
|
|
|
|
|
|
|
#params to stay the same
|
|
url = "http://127.0.0.1:7860"
|
|
outputTXT2IMGfolder = 'C:\\automated_output\\txt2img\\'
|
|
outputTXT2IMGfilename = str(uuid.uuid4())
|
|
outputTXT2IMGpng = '.png'
|
|
outputTXT2IMGFull = '{}{}{}'.format(outputTXT2IMGfolder,outputTXT2IMGfilename,outputTXT2IMGpng)
|
|
outputTXT2IMGtxtfolder = 'C:\\automated_output\\prompts\\'
|
|
outputTXT2IMGtxt = '.txt'
|
|
outputTXT2IMGtxtFull = '{}{}{}'.format(outputTXT2IMGtxtfolder,outputTXT2IMGfilename,outputTXT2IMGtxt)
|
|
|
|
|
|
|
|
#call TXT2IMG
|
|
|
|
payload = {
|
|
"prompt": prompt,
|
|
"sampler_index": sampler_index,
|
|
"steps": steps,
|
|
"cfg_scale": cfg_scale,
|
|
"width": width,
|
|
"height": height,
|
|
"enable_hr": enable_hr,
|
|
"denoising_strength": denoising_strength,
|
|
"hr_scale": hr_scale,
|
|
"hr_upscaler": hr_upscaler,
|
|
"hr_second_pass_steps": hr_second_pass_steps
|
|
|
|
}
|
|
|
|
response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload)
|
|
|
|
r = response.json()
|
|
|
|
for i in r['images']:
|
|
image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0])))
|
|
|
|
png_payload = {
|
|
"image": "data:image/png;base64," + i
|
|
}
|
|
response2 = requests.post(url=f'{url}/sdapi/v1/png-info', json=png_payload)
|
|
|
|
pnginfo = PngImagePlugin.PngInfo()
|
|
pnginfo.add_text("parameters", response2.json().get("info"))
|
|
image.save(outputTXT2IMGFull, pnginfo=pnginfo)
|
|
|
|
|
|
with open(outputTXT2IMGtxtFull,'w',encoding="utf8") as txt:
|
|
json_object = json.dumps(payload, indent = 4)
|
|
txt.write(json_object)
|
|
|
|
return outputTXT2IMGFull |