magic-pdf 0.5.8__py3-none-any.whl → 0.5.10__py3-none-any.whl

magic_pdf/cli/magicpdf.py

@@ -17,8 +17,8 @@
     
 
 效果：
-python magicpdf.py --json  s3://llm-pdf-text/scihub/xxxx.json?bytes=0,81350 
-python magicpdf.py --pdf  /home/llm/Downloads/xxxx.pdf --model /home/llm/Downloads/xxxx.json  或者 python magicpdf.py --pdf  /home/llm/Downloads/xxxx.pdf
+python magicpdf.py json-command --json  s3://llm-pdf-text/scihub/xxxx.json?bytes=0,81350
+python magicpdf.py pdf-command --pdf  /home/llm/Downloads/xxxx.pdf --model /home/llm/Downloads/xxxx.json  或者 python magicpdf.py --pdf  /home/llm/Downloads/xxxx.pdf
 """
 
 import os
@@ -45,6 +45,7 @@ from magic_pdf.rw.S3ReaderWriter import S3ReaderWriter
 from magic_pdf.rw.DiskReaderWriter import DiskReaderWriter
 from magic_pdf.rw.AbsReaderWriter import AbsReaderWriter
 import csv
+import copy
 
 parse_pdf_methods = click.Choice(["ocr", "txt", "auto"])
 
@@ -81,10 +82,11 @@ def do_parse(
     f_dump_orig_pdf=True,
     f_dump_content_list=True,
 ):
+    orig_model_list = copy.deepcopy(model_list)
 
     local_image_dir, local_md_dir = prepare_env(pdf_file_name, parse_method)
     image_writer, md_writer = DiskReaderWriter(local_image_dir), DiskReaderWriter(local_md_dir)
-    image_dir = (os.path.basename(local_image_dir),)
+    image_dir = str(os.path.basename(local_image_dir))
 
     if parse_method == "auto":
         jso_useful_key = {"_pdf_type": "", "model_list": model_list}
@@ -110,10 +112,7 @@ def do_parse(
     if f_draw_span_bbox:
         draw_span_bbox(pdf_info, pdf_bytes, local_md_dir)
 
-    # write_to_csv(r"D:\project\20231108code-clean\linshixuqiu\pdf_dev\新模型\新建文件夹\luanma.csv",
-    #              [pdf_file_name, pipe.pdf_mid_data['not_common_character_rate'], pipe.pdf_mid_data['not_printable_rate']])
-
-    md_content = pipe.pipe_mk_markdown(str(image_dir), drop_mode=DropMode.NONE)
+    md_content = pipe.pipe_mk_markdown(image_dir, drop_mode=DropMode.NONE)
     if f_dump_md:
         """写markdown"""
         md_writer.write(
@@ -133,7 +132,7 @@ def do_parse(
     if f_dump_model_json:
         """写model_json"""
         md_writer.write(
-            content=json_parse.dumps(pipe.model_list, ensure_ascii=False, indent=4),
+            content=json_parse.dumps(orig_model_list, ensure_ascii=False, indent=4),
             path=f"{pdf_file_name}_model.json",
             mode=AbsReaderWriter.MODE_TXT,
         )
@@ -146,7 +145,7 @@ def do_parse(
             mode=AbsReaderWriter.MODE_BIN,
         )
 
-    content_list = pipe.pipe_mk_uni_format(str(image_dir), drop_mode=DropMode.NONE)
+    content_list = pipe.pipe_mk_uni_format(image_dir, drop_mode=DropMode.NONE)
     if f_dump_content_list:
         """写content_list"""
         md_writer.write(
@@ -281,7 +280,7 @@ def pdf_command(pdf, model, method):
             model_path = pdf.replace(".pdf", ".json")
             if not os.path.exists(model_path):
                 logger.warning(
-                    f"not found json {model_path} existed, use paddle analyze"
+                    f"not found json {model_path} existed"
                 )
                 # 本地无模型数据则调用内置paddle分析，先传空list，在内部识别到空list再调用paddle
                 model_json = "[]"

magic_pdf/libs/version.py

@@ -1 +1 @@
-__version__ = "0.5.8"
+__version__ = "0.5.10"

magic_pdf/para/para_split_v2.py

@@ -11,6 +11,7 @@ LINE_STOP_FLAG = ['.', '!', '?', '。', '！', '？', "：", ":", ")", "）", ";
 INLINE_EQUATION = ContentType.InlineEquation
 INTERLINE_EQUATION = ContentType.InterlineEquation
 TEXT = ContentType.Text
+debug_able = False
 
 
 def __get_span_text(span):
@@ -22,6 +23,7 @@ def __get_span_text(span):
 
 
 def __detect_list_lines(lines, new_layout_bboxes, lang):
+    global debug_able
     """
     探测是否包含了列表，并且把列表的行分开.
     这样的段落特点是，顶格字母大写/数字，紧跟着几行缩进的。缩进的行首字母含小写的。
@@ -41,12 +43,14 @@ def __detect_list_lines(lines, new_layout_bboxes, lang):
                     if lst[i] == 1:
                         ones_in_this_interval.append(i)
                     i += 1
-                if len(ones_in_this_interval) > 1 or (start < len(lst) - 1 and ones_in_this_interval and lst[start + 1] in [2, 3]):
+                if len(ones_in_this_interval) > 1 or (
+                        start < len(lst) - 1 and ones_in_this_interval and lst[start + 1] in [2, 3]):
                     indices.append((start, i - 1))
                     ones_indices.append(ones_in_this_interval)
             else:
                 i += 1
         return indices, ones_indices
+
     def find_repeating_patterns(lst):
         indices = []
         ones_indices = []
@@ -132,7 +136,8 @@ def __detect_list_lines(lines, new_layout_bboxes, lang):
 
         list_indice, list_start_idx = find_repeating_patterns2(line_fea_encode)
         if len(list_indice) > 0:
-            logger.info(f"发现了列表，列表行数：{list_indice}， {list_start_idx}")
+            if debug_able:
+                logger.info(f"发现了列表，列表行数：{list_indice}， {list_start_idx}")
 
         # TODO check一下这个特列表里缩进的行左侧是不是对齐的。
         segments = []
@@ -140,13 +145,16 @@ def __detect_list_lines(lines, new_layout_bboxes, lang):
             for i in range(start, end + 1):
                 if i > 0:
                     if line_fea_encode[i] == 4:
-                        logger.info(f"列表行的第{i}行不是顶格的")
+                        if debug_able:
+                            logger.info(f"列表行的第{i}行不是顶格的")
                         break
             else:
-                logger.info(f"列表行的第{start}到第{end}行是列表")
+                if debug_able:
+                    logger.info(f"列表行的第{start}到第{end}行是列表")
 
         return split_indices(total_lines, list_indice), list_start_idx
 
+
 def cluster_line_x(lines: list) -> dict:
     """
     对一个block内所有lines的bbox的x0聚类
@@ -170,6 +178,7 @@ def cluster_line_x(lines: list) -> dict:
             min_x0 = x0_new_val
     return x0_2_new_val, min_x0
 
+
 def if_match_reference_list(text: str) -> bool:
     pattern = re.compile(r'^\d+\..*')
     if pattern.match(text):
@@ -190,7 +199,8 @@ def __valign_lines(blocks, layout_bboxes):
     new_layout_bboxes = []
 
     for layout_box in layout_bboxes:
-        blocks_in_layoutbox = [b for b in blocks if b["type"] == BlockType.Text and is_in_layout(b['bbox'], layout_box['layout_bbox'])]
+        blocks_in_layoutbox = [b for b in blocks if
+                               b["type"] == BlockType.Text and is_in_layout(b['bbox'], layout_box['layout_bbox'])]
         if len(blocks_in_layoutbox) == 0 or len(blocks_in_layoutbox[0]["lines"]) == 0:
             new_layout_bboxes.append(layout_box['layout_bbox'])
             continue
@@ -253,7 +263,8 @@ def __align_text_in_layout(blocks, layout_bboxes):
     """
     for layout in layout_bboxes:
         lb = layout['layout_bbox']
-        blocks_in_layoutbox = [block for block in blocks if block["type"] == BlockType.Text and is_in_layout(block['bbox'], lb)]
+        blocks_in_layoutbox = [block for block in blocks if
+                               block["type"] == BlockType.Text and is_in_layout(block['bbox'], lb)]
         if len(blocks_in_layoutbox) == 0:
             continue
 
@@ -383,7 +394,6 @@ def __split_para_lines(lines: list, text_blocks: list) -> list:
             continue
         text_lines.append(line)
 
-
     for block in text_blocks:
         block_bbox = block["bbox"]
         para = []
@@ -394,14 +404,12 @@ def __split_para_lines(lines: list, text_blocks: list) -> list:
         if len(para) > 0:
             text_paras.append(para)
     paras = other_paras.extend(text_paras)
-    paras_sorted = sorted(paras, key = lambda x: x[0]["bbox"][1])
+    paras_sorted = sorted(paras, key=lambda x: x[0]["bbox"][1])
     return paras_sorted
 
 
-
-
-
 def __connect_list_inter_layout(blocks_group, new_layout_bbox, layout_list_info, page_num, lang):
+    global debug_able
     """
     如果上个layout的最后一个段落是列表，下一个layout的第一个段落也是列表，那么将他们连接起来。 TODO 因为没有区分列表和段落，所以这个方法暂时不实现。
     根据layout_list_info判断是不是列表。，下个layout的第一个段如果不是列表，那么看他们是否有几行都有相同的缩进。
@@ -410,7 +418,7 @@ def __connect_list_inter_layout(blocks_group, new_layout_bbox, layout_list_info,
         return blocks_group, [False, False]
 
     for i in range(1, len(blocks_group)):
-        if len(blocks_group[i]) == 0 or len(blocks_group[i-1]) == 0:
+        if len(blocks_group[i]) == 0 or len(blocks_group[i - 1]) == 0:
             continue
         pre_layout_list_info = layout_list_info[i - 1]
         next_layout_list_info = layout_list_info[i]
@@ -418,8 +426,10 @@ def __connect_list_inter_layout(blocks_group, new_layout_bbox, layout_list_info,
         next_paras = blocks_group[i]
         next_first_para = next_paras[0]
 
-        if pre_layout_list_info[1] and not next_layout_list_info[0] and next_first_para["type"] == BlockType.Text:  # 前一个是列表结尾，后一个是非列表开头，此时检测是否有相同的缩进
-            logger.info(f"连接page {page_num} 内的list")
+        if pre_layout_list_info[1] and not next_layout_list_info[0] and next_first_para[
+            "type"] == BlockType.Text:  # 前一个是列表结尾，后一个是非列表开头，此时检测是否有相同的缩进
+            if debug_able:
+                logger.info(f"连接page {page_num} 内的list")
             # 向layout_paras[i] 寻找开头具有相同缩进的连续的行
             may_list_lines = []
             lines = next_first_para.get("lines", [])
@@ -450,7 +460,8 @@ def __connect_list_inter_page(pre_page_paras, next_page_paras, pre_page_layout_b
     if pre_page_paras[-1][-1]["type"] != BlockType.Text or next_page_paras[0][0]["type"] != BlockType.Text:
         return False
     if pre_page_list_info[1] and not next_page_list_info[0]:  # 前一个是列表结尾，后一个是非列表开头，此时检测是否有相同的缩进
-        logger.info(f"连接page {page_num} 内的list")
+        if debug_able:
+            logger.info(f"连接page {page_num} 内的list")
         # 向layout_paras[i] 寻找开头具有相同缩进的连续的行
         may_list_lines = []
         next_page_first_para = next_page_paras[0][0]
@@ -503,7 +514,7 @@ def __connect_para_inter_layoutbox(blocks_group, new_layout_bbox):
         try:
             if len(blocks_group[i]) == 0:
                 continue
-            if  len(blocks_group[i - 1]) == 0:  # TODO 考虑连接问题，
+            if len(blocks_group[i - 1]) == 0:  # TODO 考虑连接问题，
                 connected_layout_blocks.append(blocks_group[i])
                 continue
             # text类型的段才需要考虑layout间的合并
@@ -534,13 +545,14 @@ def __connect_para_inter_layoutbox(blocks_group, new_layout_bbox):
 
         pre_last_line_text = pre_last_line_text.strip()
         next_first_line_text = next_first_line_text.strip()
-        if pre_last_line['bbox'][2] == pre_x2_max and pre_last_line_text and pre_last_line_text[-1] not in LINE_STOP_FLAG and \
+        if pre_last_line['bbox'][2] == pre_x2_max and pre_last_line_text and pre_last_line_text[
+            -1] not in LINE_STOP_FLAG and \
                 next_first_line['bbox'][0] == next_x0_min:  # 前面一行沾满了整个行，并且没有结尾符号.下一行没有空白开头。
             """连接段落条件成立，将前一个layout的段落和后一个layout的段落连接。"""
             #connected_layout_paras[-1][-1].extend(layout_paras[i][0])
             connected_layout_blocks[-1][-1]["lines"].extend(blocks_group[i][0]["lines"])
             #layout_paras[i].pop(0)  # 删除后一个layout的第一个段落， 因为他已经被合并到前一个layout的最后一个段落了。
-            blocks_group[i][0]["lines"] = [] #删除后一个layout第一个段落中的lines，因为他已经被合并到前一个layout的最后一个段落了
+            blocks_group[i][0]["lines"] = []  #删除后一个layout第一个段落中的lines，因为他已经被合并到前一个layout的最后一个段落了
             blocks_group[i][0][LINES_DELETED] = True
             # if len(layout_paras[i]) == 0:
             #     layout_paras.pop(i)
@@ -638,7 +650,8 @@ def find_consecutive_true_regions(input_array):
     return regions
 
 
-def __connect_middle_align_text(page_paras, new_layout_bbox, page_num, lang, debug_mode):
+def __connect_middle_align_text(page_paras, new_layout_bbox, page_num, lang):
+    global debug_able
     """
     找出来中间对齐的连续单行文本，如果连续行高度相同，那么合并为一个段落。
     一个line居中的条件是：
@@ -660,11 +673,12 @@ def __connect_middle_align_text(page_paras, new_layout_bbox, page_num, lang, deb
             for start, end in consecutive_single_line_indices:
                 #start += index_offset
                 #end += index_offset
-                line_hi = np.array([block["lines"][0]['bbox'][3] - block["lines"][0]['bbox'][1] for block in layout_para[start:end + 1]])
+                line_hi = np.array([block["lines"][0]['bbox'][3] - block["lines"][0]['bbox'][1] for block in
+                                    layout_para[start:end + 1]])
                 first_line_text = ''.join([__get_span_text(span) for span in layout_para[start]["lines"][0]['spans']])
                 if "Table" in first_line_text or "Figure" in first_line_text:
                     pass
-                if debug_mode:
+                if debug_able:
                     logger.info(line_hi.std())
 
                 if line_hi.std() < 2:
@@ -677,10 +691,10 @@ def __connect_middle_align_text(page_paras, new_layout_bbox, page_num, lang, deb
                             and not all([x1 == layout_box[2] for x1 in all_right_x1]):
                         merge_para = [block["lines"][0] for block in layout_para[start:end + 1]]
                         para_text = ''.join([__get_span_text(span) for line in merge_para for span in line['spans']])
-                        if debug_mode:
+                        if debug_able:
                             logger.info(para_text)
                         layout_para[start]["lines"] = merge_para
-                        for i_para in range(start+1, end+1):
+                        for i_para in range(start + 1, end + 1):
                             layout_para[i_para]["lines"] = []
                             layout_para[i_para][LINES_DELETED] = True
                         #layout_para[start:end + 1] = [merge_para]
@@ -713,14 +727,15 @@ def __do_split_page(blocks, layout_bboxes, new_layout_bbox, page_num, lang):
     blocks_group = __group_line_by_layout(blocks, layout_bboxes)  # block内分段
     layout_list_info = __split_para_in_layoutbox(blocks_group, new_layout_bbox, lang)  # layout内分段
     blocks_group, page_list_info = __connect_list_inter_layout(blocks_group, new_layout_bbox, layout_list_info,
-                                                                page_num, lang)  # layout之间连接列表段落
+                                                               page_num, lang)  # layout之间连接列表段落
     connected_layout_blocks = __connect_para_inter_layoutbox(blocks_group, new_layout_bbox)  # layout间链接段落
 
     return connected_layout_blocks, page_list_info
 
 
-
 def para_split(pdf_info_dict, debug_mode, lang="en"):
+    global debug_able
+    debug_able = debug_mode
     new_layout_of_pages = []  # 数组的数组，每个元素是一个页面的layoutS
     all_page_list_info = []  # 保存每个页面开头和结尾是否是列表
     for page_num, page in pdf_info_dict.items():
@@ -744,14 +759,14 @@ def para_split(pdf_info_dict, debug_mode, lang="en"):
 
         is_conn = __connect_para_inter_page(pre_page_paras, next_page_paras, pre_page_layout_bbox,
                                             next_page_layout_bbox, page_num, lang)
-        if debug_mode:
+        if debug_able:
             if is_conn:
                 logger.info(f"连接了第{page_num - 1}页和第{page_num}页的段落")
 
         is_list_conn = __connect_list_inter_page(pre_page_paras, next_page_paras, pre_page_layout_bbox,
                                                  next_page_layout_bbox, all_page_list_info[page_num - 1],
                                                  all_page_list_info[page_num], page_num, lang)
-        if debug_mode:
+        if debug_able:
             if is_list_conn:
                 logger.info(f"连接了第{page_num - 1}页和第{page_num}页的列表段落")
 
@@ -762,7 +777,7 @@ def para_split(pdf_info_dict, debug_mode, lang="en"):
     for page_num, page in enumerate(pdf_info_dict.values()):
         page_paras = page['para_blocks']
         new_layout_bbox = new_layout_of_pages[page_num]
-        __connect_middle_align_text(page_paras, new_layout_bbox, page_num, lang, debug_mode=debug_mode)
+        __connect_middle_align_text(page_paras, new_layout_bbox, page_num, lang)
         __merge_signle_list_text(page_paras, new_layout_bbox, page_num, lang)
 
     # layout展平

magic_pdf/pdf_parse_by_ocr.py

@@ -1,219 +1,18 @@
-import time
-from loguru import logger
-from magic_pdf.libs.commons import (
-    fitz,
-    get_delta_time,
-    get_docx_model_output,
-)
-from magic_pdf.libs.convert_utils import dict_to_list
-from magic_pdf.libs.coordinate_transform import get_scale_ratio
-from magic_pdf.libs.drop_tag import DropTag
-from magic_pdf.libs.hash_utils import compute_md5
-from magic_pdf.libs.ocr_content_type import ContentType
-from magic_pdf.para.para_split import para_split
-from magic_pdf.pre_proc.construct_page_dict import ocr_construct_page_component
-from magic_pdf.pre_proc.detect_footer_by_model import parse_footers
-from magic_pdf.pre_proc.detect_footnote import parse_footnotes_by_model
-from magic_pdf.pre_proc.detect_header import parse_headers
-from magic_pdf.pre_proc.detect_page_number import parse_pageNos
-from magic_pdf.pre_proc.cut_image import ocr_cut_image_and_table
-from magic_pdf.pre_proc.ocr_detect_layout import layout_detect
-from magic_pdf.pre_proc.ocr_dict_merge import (
-    merge_spans_to_line_by_layout, merge_lines_to_block,
-)
-from magic_pdf.pre_proc.ocr_span_list_modify import remove_spans_by_bboxes, remove_overlaps_min_spans, \
-    adjust_bbox_for_standalone_block, modify_y_axis, modify_inline_equation, get_qa_need_list, \
-    remove_spans_by_bboxes_dict
-from magic_pdf.pre_proc.remove_bbox_overlap import remove_overlap_between_bbox_for_span
-
-
-def parse_pdf_by_ocr(
-        pdf_bytes,
-        pdf_model_output,
-        imageWriter,
-        start_page_id=0,
-        end_page_id=None,
-        debug_mode=False,
-):
-    pdf_bytes_md5 = compute_md5(pdf_bytes)
-
-    pdf_docs = fitz.open("pdf", pdf_bytes)
-    # 初始化空的pdf_info_dict
-    pdf_info_dict = {}
-
-    start_time = time.time()
-
-    end_page_id = end_page_id if end_page_id else len(pdf_docs) - 1
-    for page_id in range(start_page_id, end_page_id + 1):
-
-        # 获取当前页的page对象
-        page = pdf_docs[page_id]
-        # 获取当前页的宽高
-        page_w = page.rect.width
-        page_h = page.rect.height
-
-        if debug_mode:
-            time_now = time.time()
-            logger.info(
-                f"page_id: {page_id}, last_page_cost_time: {get_delta_time(start_time)}"
-            )
-            start_time = time_now
-
-        # 获取当前页的模型数据
-        ocr_page_info = get_docx_model_output(
-            pdf_model_output, page_id
-        )
-
-        """从json中获取每页的页码、页眉、页脚的bbox"""
-        page_no_bboxes = parse_pageNos(page_id, page, ocr_page_info)
-        header_bboxes = parse_headers(page_id, page, ocr_page_info)
-        footer_bboxes = parse_footers(page_id, page, ocr_page_info)
-        footnote_bboxes = parse_footnotes_by_model(page_id, page, ocr_page_info, debug_mode=debug_mode)
-
-        # 构建需要remove的bbox字典
-        need_remove_spans_bboxes_dict = {
-            DropTag.PAGE_NUMBER: page_no_bboxes,
-            DropTag.HEADER: header_bboxes,
-            DropTag.FOOTER: footer_bboxes,
-            DropTag.FOOTNOTE: footnote_bboxes,
-        }
-
-        layout_dets = ocr_page_info["layout_dets"]
-        spans = []
-
-        # 计算模型坐标和pymu坐标的缩放比例
-        horizontal_scale_ratio, vertical_scale_ratio = get_scale_ratio(
-            ocr_page_info, page
-        )
-
-        for layout_det in layout_dets:
-            category_id = layout_det["category_id"]
-            allow_category_id_list = [1, 7, 13, 14, 15]
-            if category_id in allow_category_id_list:
-                x0, y0, _, _, x1, y1, _, _ = layout_det["poly"]
-                bbox = [
-                    int(x0 / horizontal_scale_ratio),
-                    int(y0 / vertical_scale_ratio),
-                    int(x1 / horizontal_scale_ratio),
-                    int(y1 / vertical_scale_ratio),
-                ]
-                # 删除高度或者宽度为0的spans
-                if bbox[2] - bbox[0] == 0 or bbox[3] - bbox[1] == 0:
-                    continue
-                """要删除的"""
-                #  3: 'header',      # 页眉
-                #  4: 'page number', # 页码
-                #  5: 'footnote',    # 脚注
-                #  6: 'footer',      # 页脚
-                """当成span拼接的"""
-                #  1: 'image', # 图片
-                #  7: 'table',       # 表格
-                #  13: 'inline_equation',     # 行内公式
-                #  14: 'interline_equation',      # 行间公式
-                #  15: 'text',      # ocr识别文本
-                """layout信息"""
-                #  11: 'full column',   # 单栏
-                #  12: 'sub column',    # 多栏
-                span = {
-                    "bbox": bbox,
-                }
-                if category_id == 1:
-                    span["type"] = ContentType.Image
-
-                elif category_id == 7:
-                    span["type"] = ContentType.Table
-
-                elif category_id == 13:
-                    span["content"] = layout_det["latex"]
-                    span["type"] = ContentType.InlineEquation
-                elif category_id == 14:
-                    span["content"] = layout_det["latex"]
-                    span["type"] = ContentType.InterlineEquation
-                elif category_id == 15:
-                    span["content"] = layout_det["text"]
-                    span["type"] = ContentType.Text
-                # print(span)
-                spans.append(span)
-            else:
-                continue
-
-        '''删除重叠spans中较小的那些'''
-        spans, dropped_spans_by_span_overlap = remove_overlaps_min_spans(spans)
-
-        '''
-        删除remove_span_block_bboxes中的bbox
-        并增加drop相关数据
-        '''
-        spans, dropped_spans_by_removed_bboxes = remove_spans_by_bboxes_dict(spans, need_remove_spans_bboxes_dict)
-
-        '''对image和table截图'''
-        spans = ocr_cut_image_and_table(spans, page, page_id, pdf_bytes_md5, imageWriter)
-
-        '''行内公式调整, 高度调整至与同行文字高度一致(优先左侧, 其次右侧)'''
-        displayed_list = []
-        text_inline_lines = []
-        modify_y_axis(spans, displayed_list, text_inline_lines)
-
-        '''模型识别错误的行间公式, type类型转换成行内公式'''
-        spans = modify_inline_equation(spans, displayed_list, text_inline_lines)
-
-        '''bbox去除粘连'''
-        spans = remove_overlap_between_bbox_for_span(spans)
-        '''
-        对tpye=["interline_equation", "image", "table"]进行额外处理,
-        如果左边有字的话,将该span的bbox中y0调整至不高于文字的y0
-        '''
-        spans = adjust_bbox_for_standalone_block(spans)
-
-        '''从ocr_page_info中解析layout信息(按自然阅读方向排序,并修复重叠和交错的bad case)'''
-        layout_bboxes, layout_tree = layout_detect(ocr_page_info['subfield_dets'], page, ocr_page_info)
-
-        '''将spans合并成line(在layout内,从上到下,从左到右)'''
-        lines, dropped_spans_by_layout = merge_spans_to_line_by_layout(spans, layout_bboxes)
-
-        '''将lines合并成block'''
-        blocks = merge_lines_to_block(lines)
-
-        '''获取QA需要外置的list'''
-        images, tables, interline_equations, inline_equations = get_qa_need_list(blocks)
-
-        '''drop的span_list合并'''
-        dropped_spans = []
-        dropped_spans.extend(dropped_spans_by_span_overlap)
-        dropped_spans.extend(dropped_spans_by_removed_bboxes)
-        dropped_spans.extend(dropped_spans_by_layout)
-
-        dropped_text_block = []
-        dropped_image_block = []
-        dropped_table_block = []
-        dropped_equation_block = []
-        for span in dropped_spans:
-            # drop出的spans进行分类
-            if span['type'] == ContentType.Text:
-                dropped_text_block.append(span)
-            elif span['type'] == ContentType.Image:
-                dropped_image_block.append(span)
-            elif span['type'] == ContentType.Table:
-                dropped_table_block.append(span)
-            elif span['type'] in [ContentType.InlineEquation, ContentType.InterlineEquation]:
-                dropped_equation_block.append(span)
-
-        '''构造pdf_info_dict'''
-        page_info = ocr_construct_page_component(blocks, layout_bboxes, page_id, page_w, page_h, layout_tree,
-                                                 images, tables, interline_equations, inline_equations,
-                                                 dropped_text_block, dropped_image_block, dropped_table_block,
-                                                 dropped_equation_block,
-                                                 need_remove_spans_bboxes_dict)
-        pdf_info_dict[f"page_{page_id}"] = page_info
-
-    """分段"""
-
-    para_split(pdf_info_dict, debug_mode=debug_mode)
-
-    """dict转list"""
-    pdf_info_list = dict_to_list(pdf_info_dict)
-    new_pdf_info_dict = {
-        "pdf_info": pdf_info_list,
-    }
-
-    return new_pdf_info_dict
+from magic_pdf.pdf_parse_union_core import pdf_parse_union
+
+
+def parse_pdf_by_ocr(pdf_bytes,
+                     model_list,
+                     imageWriter,
+                     start_page_id=0,
+                     end_page_id=None,
+                     debug_mode=False,
+                     ):
+    return pdf_parse_union(pdf_bytes,
+                           model_list,
+                           imageWriter,
+                           "ocr",
+                           start_page_id=start_page_id,
+                           end_page_id=end_page_id,
+                           debug_mode=debug_mode,
+                           )