hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 11:12:35 +00:00
Code Issues Actions Packages Projects Releases 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