import re from mineru.utils.enum_class import MakeMode, BlockType, ContentType def merge_para_with_text(para_block): para_text = '' for line in para_block['lines']: for span in line['spans']: content = span['content'] content = content.strip() if content: para_text += content else: continue return para_text def mk_blocks_to_markdown(para_blocks, make_mode, img_buket_path=''): page_markdown = [] for para_block in para_blocks: para_text = '' para_type = para_block['type'] if para_type in [BlockType.TEXT, BlockType.LIST, BlockType.INDEX, BlockType.INTERLINE_EQUATION]: para_text = merge_para_with_text(para_block) elif para_type == BlockType.TITLE: title_level = get_title_level(para_block) para_text = f'{"#" * title_level} {merge_para_with_text(para_block)}' elif para_type == BlockType.IMAGE: if make_mode == MakeMode.NLP_MD: continue elif make_mode == MakeMode.MM_MD: # 检测是否存在图片脚注 has_image_footnote = any(block['type'] == BlockType.IMAGE_FOOTNOTE for block in para_block['blocks']) # 如果存在图片脚注，则将图片脚注拼接到图片正文后面 if has_image_footnote: for block in para_block['blocks']: # 1st.拼image_caption if block['type'] == BlockType.IMAGE_CAPTION: para_text += merge_para_with_text(block) + ' \n' for block in para_block['blocks']: # 2nd.拼image_body if block['type'] == BlockType.IMAGE_BODY: for line in block['lines']: for span in line['spans']: if span['type'] == ContentType.IMAGE: if span.get('image_path', ''): para_text += f"![]({img_buket_path}/{span['image_path']})" for block in para_block['blocks']: # 3rd.拼image_footnote if block['type'] == BlockType.IMAGE_FOOTNOTE: para_text += ' \n' + merge_para_with_text(block) else: for block in para_block['blocks']: # 1st.拼image_body if block['type'] == BlockType.IMAGE_BODY: for line in block['lines']: for span in line['spans']: if span['type'] == ContentType.IMAGE: if span.get('image_path', ''): para_text += f"![]({img_buket_path}/{span['image_path']})" for block in para_block['blocks']: # 2nd.拼image_caption if block['type'] == BlockType.IMAGE_CAPTION: para_text += ' \n' + merge_para_with_text(block) elif para_type == BlockType.TABLE: if make_mode == MakeMode.NLP_MD: continue elif make_mode == MakeMode.MM_MD: for block in para_block['blocks']: # 1st.拼table_caption if block['type'] == BlockType.TABLE_CAPTION: para_text += merge_para_with_text(block) + ' \n' for block in para_block['blocks']: # 2nd.拼table_body if block['type'] == BlockType.TABLE_BODY: for line in block['lines']: for span in line['spans']: if span['type'] == ContentType.TABLE: # if processed by table model if span.get('html', ''): para_text += f"\n{span['html']}\n" elif span.get('image_path', ''): para_text += f"![]({img_buket_path}/{span['image_path']})" for block in para_block['blocks']: # 3rd.拼table_footnote if block['type'] == BlockType.TABLE_FOOTNOTE: para_text += '\n' + merge_para_with_text(block) + ' ' if para_text.strip() == '': continue else: # page_markdown.append(para_text.strip() + ' ') page_markdown.append(para_text.strip()) return page_markdown def make_blocks_to_content_list(para_block, img_buket_path, page_idx): para_type = para_block['type'] para_content = {} if para_type in [BlockType.TEXT, BlockType.LIST, BlockType.INDEX]: para_content = { 'type': 'text', 'text': merge_para_with_text(para_block), } elif para_type == BlockType.TITLE: title_level = get_title_level(para_block) para_content = { 'type': 'text', 'text': merge_para_with_text(para_block), } if title_level != 0: para_content['text_level'] = title_level elif para_type == BlockType.INTERLINE_EQUATION: para_content = { 'type': 'equation', 'text': merge_para_with_text(para_block), 'text_format': 'latex', } elif para_type == BlockType.IMAGE: para_content = {'type': 'image', 'img_path': '', 'img_caption': [], 'img_footnote': []} for block in para_block['blocks']: if block['type'] == BlockType.IMAGE_BODY: for line in block['lines']: for span in line['spans']: if span['type'] == ContentType.IMAGE: if span.get('image_path', ''): para_content['img_path'] = f"{img_buket_path}/{span['image_path']}" if block['type'] == BlockType.IMAGE_CAPTION: para_content['img_caption'].append(merge_para_with_text(block)) if block['type'] == BlockType.IMAGE_FOOTNOTE: para_content['img_footnote'].append(merge_para_with_text(block)) elif para_type == BlockType.TABLE: para_content = {'type': 'table', 'img_path': '', 'table_caption': [], 'table_footnote': []} for block in para_block['blocks']: if block['type'] == BlockType.TABLE_BODY: for line in block['lines']: for span in line['spans']: if span['type'] == ContentType.TABLE: if span.get('html', ''): para_content['table_body'] = f"{span['html']}" if span.get('image_path', ''): para_content['img_path'] = f"{img_buket_path}/{span['image_path']}" if block['type'] == BlockType.TABLE_CAPTION: para_content['table_caption'].append(merge_para_with_text(block)) if block['type'] == BlockType.TABLE_FOOTNOTE: para_content['table_footnote'].append(merge_para_with_text(block)) para_content['page_idx'] = page_idx return para_content def union_make(pdf_info_dict: list, make_mode: str, img_buket_path: str = '', ): output_content = [] for page_info in pdf_info_dict: paras_of_layout = page_info.get('para_blocks') page_idx = page_info.get('page_idx') if not paras_of_layout: continue if make_mode in [MakeMode.MM_MD, MakeMode.NLP_MD]: page_markdown = mk_blocks_to_markdown(paras_of_layout, make_mode, img_buket_path) output_content.extend(page_markdown) elif make_mode == MakeMode.STANDARD_FORMAT: for para_block in paras_of_layout: para_content = make_blocks_to_content_list(para_block, img_buket_path, page_idx) output_content.append(para_content) if make_mode in [MakeMode.MM_MD, MakeMode.NLP_MD]: return '\n\n'.join(output_content) elif make_mode == MakeMode.STANDARD_FORMAT: return output_content return None def get_title_level(block): title_level = block.get('level', 1) if title_level > 4: title_level = 4 elif title_level < 1: title_level = 0 return title_level