img2img modes use base64 instead file path
parent
8d4c85c047
commit
a1b38a72fb
|
|
@ -92,6 +92,7 @@ async def img2ImgRequest(sd_url,payload):
|
|||
serverHelper.createFolder(dir_fullpath)
|
||||
image_paths = []
|
||||
metadata = []
|
||||
images_info = []
|
||||
#for each image store the prompt and settings in the meta data
|
||||
for i in r['images']:
|
||||
image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0])))
|
||||
|
|
@ -110,9 +111,10 @@ async def img2ImgRequest(sd_url,payload):
|
|||
metadata_info = response2.json().get("info")
|
||||
metadata_json = metadata_to_json.convertMetadataToJson(metadata_info)
|
||||
metadata.append(metadata_json)
|
||||
images_info.append({"base64":i,"path":image_path})
|
||||
print("metadata_json: ", metadata_json)
|
||||
|
||||
return dirName,image_paths,metadata
|
||||
return dirName,images_info,metadata
|
||||
|
||||
if __name__=="__main__":
|
||||
img2ImgRequest()
|
||||
|
|
@ -69,7 +69,7 @@ async def txt2ImgRequest(payload):
|
|||
metadata.append(metadata_json)
|
||||
images_info.append({"base64":i,"path":image_path})
|
||||
print("metadata_json: ", metadata_json)
|
||||
base64_images = r['images']
|
||||
|
||||
|
||||
return dirName,images_info,metadata
|
||||
|
||||
|
|
@ -167,9 +167,9 @@ async def txt2ImgHandle(request:Request):
|
|||
async def img2ImgHandle(request:Request):
|
||||
print("img2ImgHandle: \n")
|
||||
payload = await request.json()
|
||||
dir_name,image_paths,metadata = await img2imgapi.img2ImgRequest(sd_url,payload)
|
||||
dir_name,images_info,metadata = await img2imgapi.img2ImgRequest(sd_url,payload)
|
||||
# return {"prompt":payload.prompt,"images": ""}
|
||||
return {"payload": payload,"dir_name": dir_name,"image_paths":image_paths,"metadata":metadata}
|
||||
return {"payload": payload,"dir_name": dir_name,"images_info":images_info,"metadata":metadata}
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue