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

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

20
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 # This is to store the paths to the thumbnails of the profile pictures
image_cache = {} image_cache = {}
def generate_basic_html(s):
with open(Path(__file__).resolve().parent / '../css/html_readable_style.css', 'r') as f: with open(Path(__file__).resolve().parent / '../css/html_readable_style.css', 'r') as f:
css = f.read() 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 = '\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 return s
def process_post(post, c): def process_post(post, c):
@ -37,9 +41,6 @@ def process_post(post, c):
return src return src
def generate_4chan_html(f): 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 = [] posts = []
post = '' post = ''
c = -2 c = -2
@ -66,7 +67,7 @@ def generate_4chan_html(f):
posts[i] = f'<div class="reply">{posts[i]}</div>\n' posts[i] = f'<div class="reply">{posts[i]}</div>\n'
output = '' 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: for post in posts:
output += post output += post
output += '</div></div>' output += '</div></div>'
@ -101,10 +102,7 @@ def load_html_image(paths):
return '' return ''
def generate_chat_html(history, name1, name2, character): def generate_chat_html(history, name1, name2, character):
with open(Path(__file__).resolve().parent / '../css/html_cai_style.css', 'r') as f: output = f'<style>{cai_css}</style><div class="chat" id="chat">'
css = f.read()
output = f'<style>{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_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"]) img_me = load_html_image(["img_me.png", "img_me.jpg", "img_me.jpeg"])