Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use image.thumbnail

Browse Source
This commit is contained in:
oobabooga 2023-03-22 16:03:05 -03:00 committed by GitHub
parent 104212529f
commit 0abff499e2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/sd_api_pictures/script.py
View File

@ -89,15 +89,7 @@ def get_SD_pictures(description):
image.save(output_file.as_posix()) image.save(output_file.as_posix())
pic_id += 1 pic_id += 1
# lower the resolution of received images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history # lower the resolution of received images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history
width, height = image.size image.thumbnail((300, 300))
if (width > 300):
height = int(height * (300 / width))
width = 300
elif (height > 300):
width = int(width * (300 / height))
height = 300
newsize = (width, height)
image = image.resize(newsize, Image.LANCZOS)
buffered = io.BytesIO() buffered = io.BytesIO()
image.save(buffered, format="JPEG") image.save(buffered, format="JPEG")
buffered.seek(0) buffered.seek(0)

10
extensions/send_pictures/script.py
View File

@ -26,15 +26,7 @@ def caption_image(raw_image):
def generate_chat_picture(picture, name1, name2): def generate_chat_picture(picture, name1, name2):
text = f'*{name1} sends {name2} a picture that contains the following: "{caption_image(picture)}"*' text = f'*{name1} sends {name2} a picture that contains the following: "{caption_image(picture)}"*'
# lower the resolution of sent images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history # lower the resolution of sent images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history
width, height = picture.size image.thumbnail((300, 300))
if (width > 300):
height = int(height * (300 / width))
width = 300
elif (height > 300):
width = int(width * (300 / height))
height = 300
newsize = (width, height)
picture = picture.resize(newsize, Image.LANCZOS)
buffer = BytesIO() buffer = BytesIO()
picture.save(buffer, format="JPEG") picture.save(buffer, format="JPEG")
img_str = base64.b64encode(buffer.getvalue()).decode('utf-8') img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')