diff --git a/README.md b/README.md
index 2eb81da..e37885c 100644
--- a/README.md
+++ b/README.md
@@ -69,10 +69,11 @@ huggingface免科学上网[在线体验](https://huggingface.co/spaces/qingxu98/
- 如果输出包含公式,会同时以tex形式和渲染形式显示,方便复制和阅读

diff --git a/toolbox.py b/toolbox.py
index 75cc847..341d9e7 100644
--- a/toolbox.py
+++ b/toolbox.py
@@ -224,7 +224,7 @@ def markdown_convertion(txt):
content = content.replace('\n', '')
return f"
$${content}$$"
else:
- return f"
$${content}$"
+ return f"
${content}$"
def replace_math_render(match):
content = match.group(1)
@@ -237,10 +237,21 @@ def markdown_convertion(txt):
return content
else:
return tex2mathml_catch_exception(content)
+
+ def markdown_bug_hunt(content):
+ """
+ 解决一个mdx_math的bug(单$包裹begin命令时多余\n', '')
+ return content
+
+
if ('$' in txt) and ('```' not in txt): # 有$标识的公式符号,且没有代码段```的标识
# convert everything to html format
split = markdown.markdown(text='---')
- convert_stage_1 = markdown.markdown(text=txt, extensions=['mdx_math', 'fenced_code', 'tables'], extension_configs=markdown_extension_configs)
+ convert_stage_1 = markdown.markdown(text=txt, extensions=['mdx_math', 'fenced_code', 'tables', 'sane_lists'], extension_configs=markdown_extension_configs)
+ convert_stage_1 = markdown_bug_hunt(convert_stage_1)
# re.DOTALL: Make the '.' special character match any character at all, including a newline; without this flag, '.' will match anything except a newline. Corresponds to the inline flag (?s).
# 1. convert to easy-to-copy tex (do not render math)
convert_stage_2_1, n = re.subn(find_equation_pattern, replace_math_no_render, convert_stage_1, flags=re.DOTALL)
@@ -249,7 +260,7 @@ def markdown_convertion(txt):
# cat them together
return pre + convert_stage_2_1 + f'{split}' + convert_stage_2_2 + suf
else:
- return pre + markdown.markdown(txt, extensions=['fenced_code', 'tables']) + suf
+ return pre + markdown.markdown(txt, extensions=['fenced_code', 'tables', 'sane_lists']) + suf
def close_up_code_segment_during_stream(gpt_reply):