img2img modes use base64 instead file path

pull/78/head
Abdullah Alfaraj 2023-01-31 11:34:20 +03:00
parent 8d4c85c047
commit a1b38a72fb
2 changed files with 6 additions and 4 deletions

View File

@ -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()

View File

@ -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}