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

Make thumbnails for the profile pictures (for performance)

Browse source
This commit is contained in:
oobabooga 2023-02-17 10:58:54 -03:00
parent edc0262889
commit 5eeb3f4e54
1 changed files with 18 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
modules/html_generator.py
View file

@ -7,8 +7,11 @@ This is a library for formatting GPT-4chan and chat outputs as nice HTML.
import base64 import base64
import copy import copy
import re import re
from io import BytesIO
from pathlib import Path from pathlib import Path
from PIL import Image
def generate_basic_html(s): def generate_basic_html(s):
css = """ css = """
.container { .container {
@ -178,6 +181,13 @@ def generate_4chan_html(f):
return output return output
def image_to_base64(path):
img = Image.open(path)
img.thumbnail((100, 100))
img_buffer = BytesIO()
img.convert('RGB').save(img_buffer, format='PNG')
return base64.b64encode(img_buffer.getvalue()).decode("utf-8")
def generate_chat_html(history, name1, name2, character): def generate_chat_html(history, name1, name2, character):
css = """ css = """
.chat { .chat {
@ -251,6 +261,7 @@ def generate_chat_html(history, name1, name2, character):
output = '' output = ''
output += f'<style>{css}</style><div class="chat" id="chat">' output += f'<style>{css}</style><div class="chat" id="chat">'
img = '' img = ''
for i in [ for i in [
f"characters/{character}.png", f"characters/{character}.png",
f"characters/{character}.jpg", f"characters/{character}.jpg",
@ -259,20 +270,17 @@ def generate_chat_html(history, name1, name2, character):
"img_bot.jpg", "img_bot.jpg",
"img_bot.jpeg" "img_bot.jpeg"
]: ]:
if Path(i).exists(): path = Path(i)
with open(i, "rb") as image_file: if path.exists():
encoded_string = base64.b64encode(image_file.read()) img = f'<img src="data:image/png;base64,{image_to_base64(path)}">'
if i.endswith('png'):
img = f'<img src="data:image/png;base64,{encoded_string.decode("utf-8")}">'
elif i.endswith('jpg') or i.endswith('jpeg'):
img = f'<img src="data:image/jpg;base64,{encoded_string.decode("utf-8")}">'
break break
img_me = '' img_me = ''
for i in ["img_me.png", "img_me.jpg", "img_me.jpeg"]: for i in ["img_me.png", "img_me.jpg", "img_me.jpeg"]:
if Path(i).exists(): path = Path(i)
img_me = f'<img src="file/{i}">' if path.exists():
img_me = f'<img src="data:image/png;base64,{image_to_base64(path)}">'
break break
for i,_row in enumerate(history[::-1]): for i,_row in enumerate(history[::-1]):