michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix other apis

Browse source
This commit is contained in:
Sena 2022-11-24 13:10:40 +08:00 committed by GitHub
parent 75b67eebf2
commit fcd75bd874
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
modules/api/api.py
View file

@ -4,7 +4,7 @@ import time
import uvicorn
from threading import Lock
from io import BytesIO
from gradio.processing_utils import encode_pil_to_base64, decode_base64_to_file, decode_base64_to_image
from gradio.processing_utils import decode_base64_to_file
from fastapi import APIRouter, Depends, FastAPI, HTTPException
from fastapi.security import HTTPBasic, HTTPBasicCredentials
from secrets import compare_digest
@ -41,6 +41,10 @@ def setUpscalers(req: dict):
reqDict.pop('upscaler_2')
return reqDict
def decode_base64_to_image(encoding):
if encoding.startswith("data:image/"):
encoding = encoding.split(";")[1].split(",")[1]
return Image.open(BytesIO(base64.b64decode(encoding)))
def encode_pil_to_base64(image):
with io.BytesIO() as output_bytes:
@ -134,10 +138,7 @@ class Api:
mask = img2imgreq.mask
if mask:
if mask.startswith("data:image/"):
mask = decode_base64_to_image(mask)
else:
mask = Image.open(BytesIO(base64.b64decode(mask)))
mask = decode_base64_to_image(mask)
populate = img2imgreq.copy(update={ # Override __init__ params
"sd_model": shared.sd_model,
@ -151,10 +152,7 @@ class Api:
imgs = []
for img in init_images:
if img.startswith("data:image/"):
img = decode_base64_to_image(img)
else:
img = Image.open(BytesIO(base64.b64decode(img)))
img = decode_base64_to_image(img)
imgs = [img] * p.batch_size
p.init_images = imgs