|
|
import os
|
|
|
from typing import Any
|
|
|
import json
|
|
|
class ImageRequest:
|
|
|
def __init__(self):
|
|
|
self.ratio = 1
|
|
|
self.configScale = 0
|
|
|
self.guidance_scale = 7.5
|
|
|
self.seed = 52125
|
|
|
self.prompt = ""
|
|
|
self.negative_prompt = ""
|
|
|
self.neg_prompt = ""
|
|
|
self.num_images_per_prompt = 1
|
|
|
self.imagesToGenerate = 1
|
|
|
self.num_frames = 1
|
|
|
self.size = [1920, 1080]
|
|
|
self.encodeSize = [1024, 576]
|
|
|
self.steps = 6
|
|
|
self.faceFix = 1
|
|
|
self.width = 1920
|
|
|
self.height = 1080
|
|
|
self.maxSize = 1080
|
|
|
self.resolution = 256
|
|
|
self.topOffset = 0
|
|
|
self.altitude = 0
|
|
|
self.distance = 0
|
|
|
self.fov = 0
|
|
|
self.fovx = 0
|
|
|
self.fovy = 0
|
|
|
self.range = [0,1]
|
|
|
self.range2 = [0,1]
|
|
|
self.topOffset = 0
|
|
|
self.version = "1.3"
|
|
|
self.upscale = 2
|
|
|
self.bg_upsampler = "realesrgan"
|
|
|
self.chunkSize = 8192
|
|
|
self.bg_tile = 400
|
|
|
self.suffix = "None"
|
|
|
self.only_center_face = False
|
|
|
self.aligned = False
|
|
|
self.ext = "auto"
|
|
|
self.foregroundRatio = 1.0
|
|
|
self.weight = 0.75
|
|
|
self.s_scale = 0.75
|
|
|
self.scale = 0.75
|
|
|
self.encodedFps = 8
|
|
|
self.fps = 24
|
|
|
self.loops = 0
|
|
|
self.maskIndex = 0
|
|
|
self.removeBg = False
|
|
|
self.isXL = False
|
|
|
self.isXLLightning = False
|
|
|
self.maskIndex = False
|
|
|
self.watermarked = False
|
|
|
self.exportGif = False
|
|
|
self.exportTextures = False
|
|
|
self.exportMeshes = False
|
|
|
self.exportFiles = False
|
|
|
self.exportOther = False
|
|
|
self.export = False
|
|
|
self.saveFaceEmbeddings = False
|
|
|
self.overrideForm = "False"
|
|
|
|
|
|
self.submode = "default"
|
|
|
self.mode = "default"
|
|
|
self.model = "default"
|
|
|
self.modelType = "default"
|
|
|
self.ipAdapterModel = ""
|
|
|
self.onnxModel = ""
|
|
|
self.inpaintModel = ""
|
|
|
self.controlnetModel = ""
|
|
|
self.vaeRepo = ""
|
|
|
self.schedueler = ""
|
|
|
self.device = ""
|
|
|
self.tempPath = ""
|
|
|
self.paramsData = ""
|
|
|
self.exportSize = ""
|
|
|
self.faceEmbeddings = ""
|
|
|
self.input = ""
|
|
|
self.mask = ""
|
|
|
self.inputs = []
|
|
|
self.masks = []
|
|
|
self.mediaPaths = []
|
|
|
self.urls = []
|
|
|
self.styleImages = []
|
|
|
self.vertecies = []
|
|
|
self.points = []
|
|
|
self.primitives = []
|
|
|
self.meshes = []
|
|
|
self.objects = []
|
|
|
self.output = ""
|
|
|
self.customSDBinWeights = ""
|
|
|
self.textualInversionWeights = ""
|
|
|
self.refinerRepo = ""
|
|
|
self.metadata = ""
|
|
|
self.url = ""
|
|
|
self.shareuser = ""
|
|
|
self.user = ""
|
|
|
self.ipAdapters = []
|
|
|
self.loras = []
|
|
|
|
|
|
def DetectSizeFromSize(self):
|
|
|
if isinstance(self.size, str):
|
|
|
split = self.size.split(",")
|
|
|
width = int(split[0])
|
|
|
height = int(split[1])
|
|
|
self.ratio = float(width) / float(height)
|
|
|
self.size = [width, height]
|
|
|
self.width = width
|
|
|
self.height = height
|
|
|
|
|
|
def DetectEncodeSize(self):
|
|
|
if isinstance(self.encodeSize, str):
|
|
|
split = self.encodeSize.split(" ")
|
|
|
sizeSplit = split[1].lower().split("x")
|
|
|
width = int(sizeSplit[0])
|
|
|
height = int(sizeSplit[1])
|
|
|
self.encodeSize = [width, height]
|
|
|
|
|
|
def DetectSizeFromExportSize(self):
|
|
|
split = self.exportSize.split(" ")
|
|
|
ratio = split[0]
|
|
|
sizeSplit = split[1].lower().split("x")
|
|
|
width = int(sizeSplit[0])
|
|
|
height = int(sizeSplit[1])
|
|
|
self.ratio = float(width) / float(height)
|
|
|
self.size = [width, height]
|
|
|
self.width = width
|
|
|
self.height = height
|
|
|
|
|
|
def DetectXL(self):
|
|
|
self.isXL = "XL" in self.model or "xl" in self.model
|
|
|
if self.isXL:
|
|
|
self.isXLLightning = "lightning" in self.model or "Lightning" in self.model
|
|
|
|
|
|
def JSON(self):
|
|
|
return json.dumps(self.__dict__)
|
|
|
|
|
|
def SetOutputExtension(self, extension = "png"):
|
|
|
root, ext = os.path.splitext(self.output)
|
|
|
if not ext:
|
|
|
ext = f'.{extension}'
|
|
|
self.output = root + ext
|
|
|
|
|
|
def CreateResponse(self, outputFiles = [], objectType = "image", isBase64 = False):
|
|
|
|
|
|
outputDict = {"media":[]}
|
|
|
for file in outputFiles:
|
|
|
outputDict["media"].append({"media":file,"prompt": self.prompt,
|
|
|
"seed": self.seed, "objectType": objectType, "isBase64": isBase64})
|
|
|
|
|
|
outputDict["media"].append({"media":self.input,"prompt": self.prompt,
|
|
|
"seed": self.seed, "objectType": objectType, "isBase64": True})
|
|
|
return outputDict
|
|
|
|
|
|
@staticmethod
|
|
|
def FromDict(obj: Any) -> 'ImageRequest':
|
|
|
imgRequest = ImageRequest()
|
|
|
try:
|
|
|
imgRequest.chunkSize = int(obj["chunkSize"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.steps = int(obj["steps"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.weight =float( obj["weight"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.ratio = float(obj["ratio"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.s_scale = float(obj["s_scale"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.scale = float(obj["scale"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.configScale = float(obj["configScale"])
|
|
|
imgRequest.guidance_scale = float(obj["configScale"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.guidance_scale = float(obj["guidance_scale"])
|
|
|
imgRequest.configScale = float(obj["guidance_scale"])
|
|
|
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.seed = int(obj["seed"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.exportTextures = bool(obj["exportTextures"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.exportOther = bool(obj["exportOther"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.export = bool(obj["export"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.exportFiles = bool(obj["exportFiles"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.vertecies = obj["vertecies"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.points = obj["points"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
|
|
|
try:
|
|
|
imgRequest.objects = obj["objects"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
|
|
|
try:
|
|
|
imgRequest.meshes = obj["meshes"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
|
|
|
try:
|
|
|
imgRequest.primitives = obj["primitives"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.exportMeshes = bool(obj["exportMeshes"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.faceEmbeddings = obj["faceEmbeddings"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.encodeSize = obj["encodeSize"]
|
|
|
imgRequest.DetectEncodeSize()
|
|
|
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.removeBg = int(obj["removeBg"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.ipAdapterModel = obj["ipAdapterModel"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.topOffset = int(obj["topOffset"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.foregroundRatio = float(obj["foregroundRatio"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.resolution = int(obj["resolution"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.maxSize = int(obj["maxSize"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.prompt = obj["prompt"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.negative_prompt = obj["negative_prompt"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.negative_prompt = obj["neg_prompt"]
|
|
|
imgRequest.neg_prompt = obj["neg_prompt"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.negative_prompt = obj["negative_prompt"]
|
|
|
imgRequest.neg_prompt = obj["negative_prompt"]
|
|
|
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.num_images_per_prompt = int(obj["num_images_per_prompt"])
|
|
|
imgRequest.imagesToGenerate = int(obj["num_images_per_prompt"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.num_images_per_prompt = int(obj["imagesToGenerate"])
|
|
|
imgRequest.imagesToGenerate = int(obj["imagesToGenerate"])
|
|
|
imgRequest.num_frames = int(obj["imagesToGenerate"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.num_images_per_prompt = int(obj["num_frames"])
|
|
|
imgRequest.imagesToGenerate = int(obj["num_frames"])
|
|
|
imgRequest.num_frames = int(obj["num_frames"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.size = obj["size"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.faceFix = int(obj["faceFix"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.version = str(obj["version"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.weight =float( obj["facefixweight"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.upscale =int( obj["facefixupscale"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.upscale = int(obj["upscale"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.bg_upsampler = str(obj["bg_upsampler"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.bg_tile = int(obj["bg_tile"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.loops = int(obj["loops"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.suffix = int(obj["suffix"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.only_center_face = bool(obj["only_center_face"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.exportGif =bool( obj["exportGif"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.saveFaceEmbeddings =bool( obj["saveFaceEmbeddings"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.aligned =bool( obj["aligned"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.ext = str(obj["ext"])
|
|
|
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.submode = obj["submode"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.mode = obj["mode"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.model = obj["model"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.modelType = obj["modelType"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.onnxModel = obj["onnxModel"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.inpaintModel = obj["inpaintModel"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.controlnetModel = obj["controlnetModel"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.vaeRepo = obj["vaeRepo"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
|
|
|
try:
|
|
|
imgRequest.altitude = float(obj["altitude"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.distance = float(obj["distance"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.fov = float(obj["fov"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.fovc = float(obj["fovc"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.fovy = float(obj["fovy"])
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.range = obj["range"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.range2 = obj["range2"]
|
|
|
except:
|
|
|
pass
|
|
|
|
|
|
try:
|
|
|
imgRequest.schedueler = obj["schedueler"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.device = obj["device"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.tempPath = obj["tempPath"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.paramsData = obj["paramsData"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.exportSize = obj["exportSize"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.input = obj["input"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import flask
|
|
|
import base64
|
|
|
inputImage = flask.request.files['inputImage']
|
|
|
image_string = base64.b64encode(inputImage.read())
|
|
|
imgRequest.input = image_string
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import flask
|
|
|
import base64
|
|
|
inputImage = flask.request.files['input']
|
|
|
image_string = base64.b64encode(inputImage.read())
|
|
|
imgRequest.input = image_string
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import base64
|
|
|
inputs = obj["inputs"]
|
|
|
for inputImage in inputs:
|
|
|
imgRequest.inputs.append(image_string)
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import flask
|
|
|
import base64
|
|
|
inputs = flask.request.files['inputs']
|
|
|
for inputImage in inputs:
|
|
|
image_string = base64.b64encode(inputImage.read())
|
|
|
imgRequest.inputs.append(image_string)
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import flask
|
|
|
import base64
|
|
|
inputs = flask.request.files['masks']
|
|
|
for inputImage in inputs:
|
|
|
image_string = base64.b64encode(inputImage.read())
|
|
|
imgRequest.masks.append(image_string)
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import flask
|
|
|
import base64
|
|
|
inputs = flask.request.files['styleImages']
|
|
|
for inputImage in inputs:
|
|
|
image_string = base64.b64encode(inputImage.read())
|
|
|
imgRequest.styleImages.append(image_string)
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
import flask
|
|
|
import pickle
|
|
|
import base64
|
|
|
inputs = flask.request.files['faceEmbeddings']
|
|
|
for inputImage in inputs:
|
|
|
loaded = pickle.loads(inputImage.read())
|
|
|
imgRequest.faceEmbeddings.append(loaded)
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.mask = obj["mask"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.masks = obj["masks"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.mediaPaths = obj["mediaPaths"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.urls = obj["urls"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.styleImages = obj["styleImages"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.output = obj["output"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.customSDBinWeights = obj["customSDBinWeights"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.textualInversionWeights = obj["textualInversionWeights"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.refinerRepo = obj["refinerRepo"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.metadata = obj["metadata"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.url = obj["url"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.shareuser = obj["shareuser"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.user = obj["user"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.ipAdapters = obj["ipAdapters"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.loras = obj["loras"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.maskIndex = int(obj["maskIndex"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.watermarked = bool(obj["watermarked"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.overrideForm = obj["overrideForm"]
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.isXL = bool(obj["isXL"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.isXLLightning = bool(obj["isXLLightning"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.width = int(obj["width"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.height = int(obj["height"])
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.fps = float(obj["fps"])
|
|
|
|
|
|
except:
|
|
|
pass
|
|
|
try:
|
|
|
imgRequest.encodedFps = float(obj["encodedFps"])
|
|
|
|
|
|
except:
|
|
|
pass
|
|
|
if imgRequest.isXL == False:
|
|
|
imgRequest.DetectXL()
|
|
|
return imgRequest
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
