123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167 |
- sample = """
- [1]: https://baike.baidu.com/item/%E8%B4%A8%E8%83%BD%E6%96%B9%E7%A8%8B/1884527 "质能方程(质能方程式)_百度百科"
- [2]: https://www.zhihu.com/question/348249281 "如何理解质能方程 E=mc²? - 知乎"
- [3]: https://zhuanlan.zhihu.com/p/32597385 "质能方程的推导与理解 - 知乎 - 知乎专栏"
- 你好,这是必应。质能方程是描述质量与能量之间的当量关系的方程[^1^][1]。用tex格式,质能方程可以写成$$E=mc^2$$,其中$E$是能量,$m$是质量,$c$是光速[^2^][2] [^3^][3]。
- """
- import re
- def preprocess_newbing_out(s):
- pattern = r"\^(\d+)\^" # 匹配^数字^
- pattern2 = r"\[(\d+)\]" # 匹配^数字^
- def sub(m):
- return "\\[" + m.group(1) + "\\]" # 将匹配到的数字作为替换值
- result = re.sub(pattern, sub, s) # 替换操作
- if "[1]" in result:
- result += (
- '<br/><hr style="border-top: dotted 1px #44ac5c;"><br/><small>'
- + "<br/>".join(
- [
- re.sub(pattern2, sub, r)
- for r in result.split("\n")
- if r.startswith("[")
- ]
- )
- + "</small>"
- )
- return result
- def close_up_code_segment_during_stream(gpt_reply):
- """
- 在gpt输出代码的中途(输出了前面的```,但还没输出完后面的```),补上后面的```
- Args:
- gpt_reply (str): GPT模型返回的回复字符串。
- Returns:
- str: 返回一个新的字符串,将输出代码片段的“后面的```”补上。
- """
- if "```" not in gpt_reply:
- return gpt_reply
- if gpt_reply.endswith("```"):
- return gpt_reply
- # 排除了以上两个情况,我们
- segments = gpt_reply.split("```")
- n_mark = len(segments) - 1
- if n_mark % 2 == 1:
- # print('输出代码片段中!')
- return gpt_reply + "\n```"
- else:
- return gpt_reply
- import markdown
- from latex2mathml.converter import convert as tex2mathml
- def markdown_convertion(txt):
- """
- 将Markdown格式的文本转换为HTML格式。如果包含数学公式,则先将公式转换为HTML格式。
- """
- pre = '<div class="markdown-body">'
- suf = "</div>"
- if txt.startswith(pre) and txt.endswith(suf):
- # print('警告,输入了已经经过转化的字符串,二次转化可能出问题')
- return txt # 已经被转化过,不需要再次转化
- markdown_extension_configs = {
- "mdx_math": {
- "enable_dollar_delimiter": True,
- "use_gitlab_delimiters": False,
- },
- }
- find_equation_pattern = r'<script type="math/tex(?:.*?)>(.*?)</script>'
- def tex2mathml_catch_exception(content, *args, **kwargs):
- try:
- content = tex2mathml(content, *args, **kwargs)
- except:
- content = content
- return content
- def replace_math_no_render(match):
- content = match.group(1)
- if "mode=display" in match.group(0):
- content = content.replace("\n", "</br>")
- return f'<font color="#00FF00">$$</font><font color="#FF00FF">{content}</font><font color="#00FF00">$$</font>'
- else:
- return f'<font color="#00FF00">$</font><font color="#FF00FF">{content}</font><font color="#00FF00">$</font>'
- def replace_math_render(match):
- content = match.group(1)
- if "mode=display" in match.group(0):
- if "\\begin{aligned}" in content:
- content = content.replace("\\begin{aligned}", "\\begin{array}")
- content = content.replace("\\end{aligned}", "\\end{array}")
- content = content.replace("&", " ")
- content = tex2mathml_catch_exception(content, display="block")
- return content
- else:
- return tex2mathml_catch_exception(content)
- def markdown_bug_hunt(content):
- """
- 解决一个mdx_math的bug(单$包裹begin命令时多余<script>)
- """
- content = content.replace(
- '<script type="math/tex">\n<script type="math/tex; mode=display">',
- '<script type="math/tex; mode=display">',
- )
- content = content.replace("</script>\n</script>", "</script>")
- 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", "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,
- )
- # 2. convert to rendered equation
- convert_stage_2_2, n = re.subn(
- find_equation_pattern, replace_math_render, convert_stage_1, flags=re.DOTALL
- )
- # 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", "codehilite", "tables", "sane_lists"]
- )
- + suf
- )
- sample = preprocess_newbing_out(sample)
- sample = close_up_code_segment_during_stream(sample)
- sample = markdown_convertion(sample)
- with open("tmp.html", "w", encoding="utf8") as f:
- f.write(
- """
- <head>
- <title>My Website</title>
- <link rel="stylesheet" type="text/css" href="style.css">
- </head>
- """
- )
- f.write(sample)
|