Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-23 21:18:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

Include the bot's image as base64

Browse Source 
This is needed for Colab.
This commit is contained in:
oobabooga 2023-02-17 00:24:27 -03:00
parent cb226247e8
commit c4e87c109e
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/html_generator.py
View File

@ -4,6 +4,7 @@ This is a library for formatting GPT-4chan and chat outputs as nice HTML.
'''
import base64
import copy
import re
from pathlib import Path
@ -260,7 +261,12 @@ def generate_chat_html(history, name1, name2, character):
]:
if Path(i).exists():
img = f'<img src="file/{i}">'
with open(i, "rb") as image_file:
encoded_string = base64.b64encode(image_file.read())
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
img_me = ''