o 0jD @sddlmZddlZddlmZddlmZddlZddlm Z m Z m Z ddl m Z dd lmZmZmZmZmZmZmZdd lmZdd lmZmZmZmZmZmZmZd d l m!Z!d dl"m#Z#GdddeeeeeeZ$dS)) annotationsN)deepcopy)List)Image ImageDraw ImageFont) PINGFANG_FONT) BaseCVResult HtmlMixin JsonMixin LatexMixin MarkdownMixin WordMixin XlsxMixin)MarkdownConverter)build_handle_funcs_dictformat_centered_by_htmlformat_chart2markdown_tableformat_image_plainformat_image_scaled_by_htmlformat_text_plainsimplify_table) LayoutBlock) get_seg_flagcs~eZdZdZdfdd ZdddZdd d Zd d dZdddZdddZ dddZ d!d"ddZ d"ddZ d"ddZ ZS)#LayoutParsingResultV2zLayout Parsing Result V2returnNonecsLt|t|t|t|t|t|t|dS)z@Initializes a new instance of the class with the specified data.N)super__init__r rrr rr)selfdata __class__u/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/layout_parsing/result_v2.pyr!5s      zLayoutParsingResultV2.__init__dict[str, np.ndarray]c Csddlm}i}|d}|dr |djD]\}}|||<q|djd|d<|dr6|d jd|d <|d jd |d <|d rt|d dkrtt|dddddddddf}t |}d}t t|d D](} |d | } | d} | D]} dd| D\} }}}|j | |||g|ddqqr||d<|drt|ddkrt t|dD]} |d| }|d}|j}d|}|d ||<qt|dddddddddf}t |d}t dt |jd}tjtj|dd}|d }|D]T}|j}|j}|j}||d!}|j ||d"|durV|dd|d|df}t |j|d|krKt |d|d#|d|df}|j|t||d$d%q||d&<|S)'Nr)get_show_colormodel_settingsuse_doc_preprocessordoc_preprocessor_reslayout_det_resresZuse_region_detectionZregion_det_resoverall_ocr_resZ ocr_res_imguse_table_recognitiontable_res_listr output_img)rr cell_box_listcSsg|]}t|qSr&)int).0posr&r&r' Xsz1LayoutParsingResultV2._to_img..)outlinewidthtable_cell_imguse_seal_recognition seal_res_listseal_region_idZseal_res_regionZRGBAg;On?zutf-8)encodingparsing_res_listF)fillg?red)fontrCZlayout_order_res)utilsr)imgitemslenrZ fromarraycopyrrZDrawrangeZ rectangler6r<rZtruetyper pathbbox order_indexlabeltextstr)r"r)Z res_img_dictr*keyvaluer=Z table_drawZrectangle_colorsno table_resr5boxx1y1Zx2y2seal_resr@Zsub_seal_res_dictimageZdrawZ font_sizerEZparsing_resultblockrMindexrOZ fill_colorZ text_positionr&r&r'_to_img?sl  (    *   zLayoutParsingResultV2._to_imgdict[str, str]c Osi}|d|d<|d|d<|d|d<|d|d<|d|d<|d}||d<|d}dd |D}||d<|dd rF|d jd |d <|d jd |d <|djd |d<|drt|ddkrg|d<tt|dD]}|d|}|d|jd qp|drt|ddkrg|d<tt|dD]}|d|}|d|jd q|drt|ddkrg|d<tt|dD]}|d|} |d| jd qtj|g|Ri|S)axConverts the instance's attributes to a dictionary and then to a string. Args: *args: Additional positional arguments passed to the base class method. **kwargs: Additional keyword arguments passed to the base class method. Returns: Dict[str, str]: A dictionary with the instance's attributes converted to strings. input_path page_index page_countr<heightr*rBcSs&g|]}|j|j|j|j|jdqS)Z block_label block_contentZ block_bboxZblock_idZ block_order)rOcontentrMr]rN)r7 parsing_resr&r&r'r9sz1LayoutParsingResultV2._to_str..r+r,r.r-r/r0r1rr>r?use_formula_recognitionformula_res_list)rQrIrKappendr _to_str) r"argskwargsr#r*rBrTrUrZ formula_resr&r&r'rksF         zLayoutParsingResultV2._to_strTc s|ddjd|rddfddnddt|dd d r't}n}|dd d r9fd d}n}|dd d rQ|rLfdd}ndd}n}|ddd r`dd}n}t||||dd}|ddgD]}||dqu|S)z6Build label-to-handler mapping for content formatting.r,r2rcSs tt|SN)rrr\r&r&r'sz@LayoutParsingResultV2._build_handle_funcs_dict..cstt|dS)Noriginal_image_width)rrrprrr&r'rqs cSs|jSrorfrpr&r&r'rqsr*Zuse_chart_recognitionFr>csd||gSN )joinrp)format_image_funcformat_text_funcr&r'rqsr0csd|ddS)Nrvzz
)replacerp)ryr&r'rqs cSstd|jSru)rrfrpr&r&r'rqrhcSsd|jdS)Nz$$rtrpr&r&r'rqr{T)Z text_funcZ image_funcZ chart_funcZ table_funcZ formula_funcZ seal_funcZuse_plain_header_footer_imageZmarkdown_ignore_labelsN)shapergetrrpop)r"prettyZformat_chart_funcZformat_seal_funcZformat_table_funcZformat_formula_funchandle_funcs_dictrOr&)rxryrsr'_build_handle_funcs_dicts@   z.LayoutParsingResultV2._build_handle_funcs_dictcOsJ|dddr|jdd}i}|d|d<|d|d<|d|d<|d |d <|d |d <|d}||d<|d }g}|D]1}|j|j|j|j|jd } |dddrj||jre||j|| d <n|j| d <|| q>||d <|ddr|djd|d<|djd|d<|djd|d<|drt |ddkrg|d<t t |dD]} |d| } |d| jdq|drt |ddkrg|d<t t |dD]} |d| } |d| jdq|drt |ddkrg|d<t t |dD]} |d| } |d| jdqt j |g|Ri|S)a[ Converts the object's data to a JSON dictionary. Args: *args: Positional arguments passed to the JsonMixin._to_json method. **kwargs: Keyword arguments passed to the JsonMixin._to_json method. Returns: Dict[str, str]: A dictionary containing the object's data in JSON format. r*Zformat_block_contentFTrr`rarbr<rcrBrdrer+r,r.r-r/r0r1rr>r?rhri) r}rrOrfrMr]rNrjjsonrIrKr _to_json)r"rlrmrr#r*rBZparsing_res_list_jsonrgZres_dictrTrUrZrnr&r&r'rsb             zLayoutParsingResultV2._to_jsoncCj|d}i}|dr3t|ddkr3tt|dD]}|d|}|d}d|}|jd||<q|S)zConverts the prediction to its corresponding HTML representation. Returns: Dict[str, str]: The str type HTML representation result. r*r0r1rtable_region_idtable_pred)rIrKhtml)r"r*Z res_html_dictrTrUrrRr&r&r'_to_html1  zLayoutParsingResultV2._to_htmlcCr)zConverts the prediction HTML to an XLSX file path. Returns: Dict[str, str]: The str type XLSX representation result. r*r0r1rrrr)rIrKZxlsx)r"r*Z res_xlsx_dictrTrUrrRr&r&r'_to_xlsxArzLayoutParsingResultV2._to_xlsxFdictcCsF|j|d}tj|d||dt|dd}|d|d<|d|d<|S)z Save the parsing result to a Markdown file. Args: pretty (Optional[bool]): whether to pretty markdown by HTML, default by True. Returns: Dict rrBT imgs_in_doc)rshow_formula_numberZ use_seg_flagZget_seg_flag_funcrrar`)rrconvertr)r"rrrresultr&r&r' _to_markdownQs   z"LayoutParsingResultV2._to_markdowncCsHddlm}||d|dgd\}}||ddjd|d |d S) aConvert the object's parsing result into a Word-compatible dict. Returns: dict: { "word_blocks": List[Dict], # Simplified list of content blocks "original_image_width": int, # Pixel width of the source page "input_path": str, # Original input file path "images": List[Dict] # List of {"path": str, "img": PIL.Image} } r )build_word_blocksrBr)rr,r2rr`) word_blocksrsr`images)common.result.converterrr}r|)r"rrrr&r&r'_to_wordis   zLayoutParsingResultV2._to_wordcCsg}g}|dD]U}|j}t|dd}|dvr#|jdurq|jd}n|dkr>|jdur2|jd}n |r=|dd }d }nq|t|d }|||jdur]||jd|jd d q|||ddS)a Convert the object's parsing result into a latex-compatible dict. Returns: dict: { "latex_blocks": List[Dict], # Simplified list of content blocks "input_path": str, # Original input file path "images": List[Dict] # List of {"path": str, "img": PIL.Image} } rBrf)r[ZsealNrLchart| table)typerfrG)rLrGr`) latex_blocksrr`)rOgetattrr[rzrrj)r"rr[r\rOrfZ block_dictr&r&r' _to_latexs6          zLayoutParsingResultV2._to_latex)rr)rr()rr_)T)TF)rr)__name__ __module__ __qualname____doc__r!r^rkrrrrrrr __classcell__r&r&r$r'r0s @ 7 9 B   r)% __future__rrJrtypingrnumpynpZPILrrrZ utils.fontsr Z common.resultr r r rrrrrrZ-common.result.converter.markdown_format_funcsrrrrrrrZlayout_objectsrrFrrr&r&r&r's    $ $