import gradio as gr
import os
import re
import markdown
from typing import Dict, Any
def create_manual_tab(constant: Dict[str, Any]) -> Dict[str, Any]:
# 添加自定义CSS，增大文本大小并添加左侧导航栏样式
custom_css = """
"""
# 添加JavaScript代码，用于处理导航点击
custom_js = """
"""
# 使用Python的markdown库将Markdown转换为HTML
def markdown_to_html(markdown_content, base_path="src/web/manual"):
"""将Markdown内容转换为HTML，并将图片嵌入为base64编码"""
# 处理图片路径，使用base64编码直接嵌入图片
def embed_image(match):
alt_text = match.group(1)
img_path = match.group(2)
# 检查路径是否为外部URL
if img_path.startswith(('http://', 'https://')):
return f'
'
# 处理本地图片路径
try:
# 去掉开头的/以获取正确的路径
if img_path.startswith('/'):
img_path = img_path[1:]
# 获取绝对路径
current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.dirname(os.path.dirname(current_dir))
abs_img_path = os.path.join(project_root, img_path)
# 读取图片并转换为base64
import base64
from pathlib import Path
image_path = Path(abs_img_path)
if image_path.exists():
image_type = image_path.suffix.lstrip('.').lower()
if image_type == 'jpg':
image_type = 'jpeg'
with open(image_path, "rb") as img_file:
encoded_string = base64.b64encode(img_file.read()).decode('utf-8')
return f'
'
else:
print(f"图片文件不存在: {abs_img_path}")
return f'[图片不存在: {img_path}]'
except Exception as e:
print(f"处理图片时出错: {e}, 路径: {img_path}")
return f'[图片加载错误: {img_path}]'
# 使用正则表达式处理所有图片标记
pattern = r'!\[(.*?)\]\((.*?)\)'
processed_content = re.sub(pattern, embed_image, markdown_content)
# 使用Python的markdown库进行转换
html = markdown.markdown(
processed_content,
extensions=[
'tables',
'fenced_code',
'codehilite',
'nl2br',
'extra',
'mdx_truly_sane_lists'
],
extension_configs={
'mdx_truly_sane_lists': {
'nested_indent': 2,
'truly_sane': True
}
}
)
return html
# 从Markdown内容生成HTML导航栏和处理内容
def generate_toc_and_content(markdown_content):
"""从Markdown内容生成HTML导航栏和处理内容"""
# 提取所有标题
headers = re.findall(r'^(#{1,3})\s+(.+)$', markdown_content, re.MULTILINE)
if not headers:
return "", markdown_content
toc_html = ""
# 为每个标题创建导航项
for i, (level, title) in enumerate(headers):
level_num = len(level)
header_id = f"header-{i}"
# 根据标题级别添加类
css_class = ""
if level_num == 2:
css_class = "nav-h2"
elif level_num == 3:
css_class = "nav-h3"
toc_html += f"- {title}
"
toc_html += "
{toc_html}
{html_content}
{toc_html}
{html_content}
{toc_html}
{html_content}
{toc_html}
{html_content}
{toc_html}
{html_content}
"
# 为每个标题创建导航项
for i, (level, title) in enumerate(headers):
level_num = len(level)
header_id = f"header-{i}"
# 根据标题级别添加类
css_class = ""
if level_num == 2:
css_class = "nav-h2"
elif level_num == 3:
css_class = "nav-h3"
toc_html += f"- {title}
"
toc_html += "
{training_toc}
{training_html}
{prediction_toc}
{prediction_html}
{evaluation_toc}
{evaluation_html}
{download_toc}
{download_html}