mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-06 02:30:30 -08:00
Code Issues Projects Releases Packages Wiki Activity

Fix EXIF orientation in API image loading

Browse source
This commit is contained in:
Alon Burg 2024-02-29 10:07:15 +02:00
parent c8a5322d1f
commit 67d8dafe44
2 changed files with 51 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/api/api.py
View file

@ -86,6 +86,7 @@ def decode_base64_to_image(encoding):
response = requests.get(encoding, timeout=30, headers=headers)
try:
image = Image.open(BytesIO(response.content))
image = images.apply_exif_orientation(image)
return image
except Exception as e:
raise HTTPException(status_code=500, detail="Invalid image url") from e
@ -94,6 +95,7 @@ def decode_base64_to_image(encoding):
encoding = encoding.split(";")[1].split(",")[1]
try:
image = Image.open(BytesIO(base64.b64decode(encoding)))
image = images.apply_exif_orientation(image)
return image
except Exception as e:
raise HTTPException(status_code=500, detail="Invalid encoded image") from e