Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 10:35:10 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

API: Properly handle Images with RGBA color format (#5332)

Browse source
This commit is contained in:
Ercan 2024-01-22 06:08:51 -05:00 committed by GitHub
parent db1da9f98d
commit 166fdf09f3
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
extensions/openai/completions.py
View file

@ -156,8 +156,9 @@ def convert_history(history):
img = Image.open(BytesIO(my_res.content)) img = Image.open(BytesIO(my_res.content))
except Exception: except Exception:
raise 'Image cannot be loaded from the URL!' raise 'Image cannot be loaded from the URL!'
buffered = BytesIO() buffered = BytesIO()
if img.mode in ("RGBA", "P"):
img = img.convert("RGB")
img.save(buffered, format="JPEG") img.save(buffered, format="JPEG")
img_str = base64.b64encode(buffered.getvalue()).decode('utf-8') img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
content = f'<img src="data:image/jpeg;base64,{img_str}">' content = f'<img src="data:image/jpeg;base64,{img_str}">'