From a1b38a72fbf210360463dc022552abd7e87f3b4e Mon Sep 17 00:00:00 2001 From: Abdullah Alfaraj Date: Tue, 31 Jan 2023 11:34:20 +0300 Subject: [PATCH] img2img modes use base64 instead file path --- server/python_server/img2imgapi.py | 4 +++- server/python_server/serverMain.py | 6 +++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/server/python_server/img2imgapi.py b/server/python_server/img2imgapi.py index 489dd0b..9bde581 100644 --- a/server/python_server/img2imgapi.py +++ b/server/python_server/img2imgapi.py @@ -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() \ No newline at end of file diff --git a/server/python_server/serverMain.py b/server/python_server/serverMain.py index 725fc27..3ccde22 100644 --- a/server/python_server/serverMain.py +++ b/server/python_server/serverMain.py @@ -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}