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

Optimize the HTML generation speed

Browse source
This commit is contained in:
oobabooga 2023-03-15 14:19:28 -03:00
parent bf812c4893
commit c5f14fb9b8
1 changed files with 10 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
modules/html_generator.py
View file

@ -14,12 +14,16 @@ from PIL import Image
# This is to store the paths to the thumbnails of the profile pictures
image_cache = {}
def generate_basic_html(s):
with open(Path(__file__).resolve().parent / '../css/html_readable_style.css', 'r') as f:
css = f.read()
with open(Path(__file__).resolve().parent / '../css/html_readable_style.css', 'r') as f:
readable_css = f.read()
with open(Path(__file__).resolve().parent / '../css/html_4chan_style.css', 'r') as css_f:
_4chan_css = css_f.read()
with open(Path(__file__).resolve().parent / '../css/html_cai_style.css', 'r') as f:
cai_css = f.read()
def generate_basic_html(s):
s = '\n'.join([f'<p>{line}</p>' for line in s.split('\n')])
s = f'<style>{css}</style><div class="container">{s}</div>'
s = f'<style>{readable_css}</style><div class="container">{s}</div>'
return s
def process_post(post, c):
@ -37,9 +41,6 @@ def process_post(post, c):
return src
def generate_4chan_html(f):
with open(Path(__file__).resolve().parent / '../css/html_4chan_style.css', 'r') as css_f:
css = css_f.read()
posts = []
post = ''
c = -2
@ -66,7 +67,7 @@ def generate_4chan_html(f):
posts[i] = f'<div class="reply">{posts[i]}</div>\n'
output = ''
output += f'<style>{css}</style><div id="parent"><div id="container">'
output += f'<style>{_4chan_css}</style><div id="parent"><div id="container">'
for post in posts:
output += post
output += '</div></div>'
@ -101,10 +102,7 @@ def load_html_image(paths):
return ''
def generate_chat_html(history, name1, name2, character):
with open(Path(__file__).resolve().parent / '../css/html_cai_style.css', 'r') as f:
css = f.read()
output = f'<style>{css}</style><div class="chat" id="chat">'
output = f'<style>{cai_css}</style><div class="chat" id="chat">'
img_bot = load_html_image([f"characters/{character}.{ext}" for ext in ['png', 'jpg', 'jpeg']] + ["img_bot.png","img_bot.jpg","img_bot.jpeg"])
img_me = load_html_image(["img_me.png", "img_me.jpg", "img_me.jpeg"])