"vscode:/vscode.git/clone" did not exist on "d527d5d27b9110325e6cc2a97e0068278e158251"
ocr_mkcontent.py 924 Bytes
Newer Older
赵小蒙's avatar
赵小蒙 committed
1
2
3
4
5
6
7
8
9
10
11
12
def mk_nlp_markdown(pdf_info_dict: dict):

    markdown = []

    for _, page_info in pdf_info_dict.items():
        blocks = page_info.get("preproc_blocks")
        if not blocks:
            continue
        for block in blocks:
            for line in block['lines']:
                line_text = ''
                for span in line['spans']:
13
14
                    if not span.get('content'):
                        continue
赵小蒙's avatar
赵小蒙 committed
15
16
17
18
19
20
21
22
23
                    content = span['content'].replace('$', '\$')  # 转义$
                    if span['type'] == 'inline_equation':
                        content = f"${content}$"
                    elif span['type'] == 'displayed_equation':
                        content = f"$$\n{content}\n$$"
                    line_text += content + ' '
                # 在行末添加两个空格以强制换行
                markdown.append(line_text.strip() + '  ')
    return '\n'.join(markdown)