from markdown_it import MarkdownIt
import html
import re
def get_style_css(style_name):
"""
根据选择的样式名称获取对应的CSS样式文件
Args:
style_name (str): 样式名称，可选值为"Default"、"MBE"、"Glassmorphism"、"Apple"
Returns:
str: CSS样式内容
"""
if style_name == "Default":
return open("assets/demo.css", "r").read()
elif style_name == "1":
return open("assets/demo.1.css", "r").read()
elif style_name == "MBE":
return open("assets/demo_mbe.css", "r").read()
elif style_name == "Glassmorphism":
return open("assets/demo_glassmorphism.css", "r").read()
elif style_name == "Apple":
return open("assets/demo_apple.css", "r").read()
elif style_name == "Paper":
return open("assets/demo_paper.css", "r").read()
else:
return open("assets/demo.css", "r").read()
def decorate_writing(writing_result, style="Default"):
if not writing_result:
return writing_result
cite_pattern = r']*)?>(.*?)'
takeaway_pattern = r'[^"\']+)["\'])?>(?P[^<]*)'
citation_map = {}
def replace_cite(match):
nonlocal citation_map
urls = match.group(1).split(',')
content = match.group(2)
citation_html = []
for url in urls:
if url not in citation_map:
citation_map[url] = len(citation_map) + 1
current_index = citation_map[url]
citation_html.append((f'{current_index}', current_index))
citation_html = sorted(citation_html, key=lambda x: x[1])
citation_html = ', '.join([x[0] for x in citation_html])
cite_html = f'{content}[{citation_html}]'
return cite_html
decorated_result = re.sub(cite_pattern, replace_cite, writing_result, flags=re.S)
def replace_takeaway(match):
class_attr = match.group('class')
content = match.group('content')
if class_attr:
return f'{content}
'
else:
return f'{content}
'
decorated_result = re.sub(takeaway_pattern, replace_takeaway, decorated_result, flags=re.S)
mermaid_pattern = r'```mermaid\n(.*?)\n```'
def decorate_mermaid(match):
return f"""
{match.group(1)}
"""
decorated_result = re.sub(mermaid_pattern, decorate_mermaid, decorated_result, flags=re.S)
echarts_pattern = r'```echarts\n(.*?)\n```'
echarts_index = 0
def replace_echarts(match):
"""
将echarts代码块转换为HTML和JavaScript
Args:
match: 正则表达式匹配对象
Returns:
str: 包含HTML和JavaScript的echarts图表代码
"""
nonlocal echarts_index
echarts_code = match.group(1)
echarts_id = f'echarts-container-{echarts_index}'
echarts_index += 1
replace_code = f"""
Echarts Rendering...
"""
return replace_code
decorated_result = re.sub(echarts_pattern, replace_echarts, decorated_result, flags=re.S)
md = MarkdownIt()
body = md.render(decorated_result)
selected_css = get_style_css(style)
html_content = """
""" + body + """
"""
# 转义HTML内容以便在iframe中安全使用
# 这是必要的，因为HTML内容包含引号和其他特殊字符
escaped_html_content = html.escape(html_content)
# 定义iframe的样式属性
iframe_style = "width: 100%; height: 1024px; transform-origin: top left; border-color: lightgrey; border-width: 1px; border-radius: 10px;"
# 创建最终的iframe HTML，通过srcdoc属性注入转义后的HTML内容
# 设置loading="eager"和importance="high"以优先加载
# pointer-events="none"防止用户与iframe内容交互
iframe_content = f''
# 返回最终的iframe HTML内容
iframe_content = re.sub(r'\n\s*\n', '\n', iframe_content)
return iframe_content