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

UI: fix multiline LaTeX equations

Browse Source
This commit is contained in:
oobabooga 2024-07-27 15:33:30 -07:00
parent 493f8c3242
commit ffc713f72b
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
modules/html_generator.py
View File

@ -103,12 +103,27 @@ def convert_to_markdown(string):
result = '' result = ''
is_code = False is_code = False
is_latex = False
for line in string.split('\n'): for line in string.split('\n'):
if line.lstrip(' ').startswith('```'): stripped_line = line.strip()
if stripped_line.startswith('```'):
is_code = not is_code is_code = not is_code
elif stripped_line.startswith('$$'):
is_latex = not is_latex
elif stripped_line.endswith('$$'):
is_latex = False
elif stripped_line.startswith('\\\\['):
is_latex = True
elif stripped_line.startswith('\\\\]'):
is_latex = False
elif stripped_line.endswith('\\\\]'):
is_latex = False
result += line result += line
if is_code or line.startswith('|'): # Don't add an extra \n for tables or code
# Don't add an extra \n for tables, code, or LaTeX
if is_code or is_latex or line.startswith('|'):
result += '\n' result += '\n'
else: else:
result += '\n\n' result += '\n\n'