1580158
/
Auto-Photoshop-StableDiffusion-Plugin
mirror of https://github.com/AbdullahAlfaraj/Auto-Photoshop-StableDiffusion-Plugin.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/python_server/img2imgapi.py
View File

@ -92,6 +92,7 @@ async def img2ImgRequest(sd_url,payload):
serverHelper.createFolder(dir_fullpath) serverHelper.createFolder(dir_fullpath)
image_paths = [] image_paths = []
metadata = [] metadata = []
images_info = []
#for each image store the prompt and settings in the meta data #for each image store the prompt and settings in the meta data
for i in r['images']: for i in r['images']:
image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0]))) 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_info = response2.json().get("info")
metadata_json = metadata_to_json.convertMetadataToJson(metadata_info) metadata_json = metadata_to_json.convertMetadataToJson(metadata_info)
metadata.append(metadata_json) metadata.append(metadata_json)
images_info.append({"base64":i,"path":image_path})
print("metadata_json: ", metadata_json) print("metadata_json: ", metadata_json)
return dirName,image_paths,metadata return dirName,images_info,metadata
if __name__=="__main__": if __name__=="__main__":
img2ImgRequest() img2ImgRequest()

6
server/python_server/serverMain.py
View File

@ -69,7 +69,7 @@ async def txt2ImgRequest(payload):
metadata.append(metadata_json) metadata.append(metadata_json)
images_info.append({"base64":i,"path":image_path}) images_info.append({"base64":i,"path":image_path})
print("metadata_json: ", metadata_json) print("metadata_json: ", metadata_json)
base64_images = r['images']
return dirName,images_info,metadata return dirName,images_info,metadata
@ -167,9 +167,9 @@ async def txt2ImgHandle(request:Request):
async def img2ImgHandle(request:Request): async def img2ImgHandle(request:Request):
print("img2ImgHandle: \n") print("img2ImgHandle: \n")
payload = await request.json() 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 {"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}