o 0j!@s\ddlZddlmZddlZddlmZmZddlm Z m Z m Z m Z Gddde e e Z dS)N)Dict)Image ImageDraw) BaseCVResult HtmlMixin JsonMixin XlsxMixincseZdZdZdfdd ZdeeejffddZ deeeffdd Z deeeffd d Z deeeffd d Z deeeffddZ ZS)LayoutParsingResultzLayout Parsing ResultreturnNcs$t|t|t|dS)z@Initializes a new instance of the class with the specified data.N)super__init__rr )selfdata __class__r/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/layout_parsing/result.pyr s  zLayoutParsingResult.__init__c Csi}|d}|dr|jdi|dj|djd|d<|djd|d<|drt|d d krtt|dd ddddddd f}t|}d }t t|d D](}|d |}|d}|D]} dd| D\} } } } |j | | | | g|ddqhqZ||d<|drt|dd krt t|dD]}|d|}|d}|j}d|}|d||<q|drt|dd krt t|dD]}|d|}|d}|j}d|}|d||<q|S)Nmodel_settingsuse_doc_preprocessordoc_preprocessor_reslayout_det_resresoverall_ocr_resZ ocr_res_imguse_table_recognitiontable_res_listrZ output_img)rr cell_box_listcSsg|]}t|qSr)int).0posrrr 4sz/LayoutParsingResult._to_img..)outlinewidthtable_cell_imguse_seal_recognition seal_res_listseal_region_idZseal_res_regionuse_formula_recognitionformula_res_listformula_region_idZformula_res_regionr) updateZimglenrZ fromarraycopydeepcopyrZDrawrangeZ rectangle)rZ res_img_dictrr&Z table_drawZrectangle_colorsno table_resrboxx1y1Zx2y2seal_resr)Zsub_seal_res_dictkey formula_resr,Zsub_formula_res_dictrrr_to_img!sN(      zLayoutParsingResult._to_imgc Oi}|d|d<|d|d<|d}||d<|d|d<|ddr+|djd|d<|djd|d<|d jd|d <|d rht|d d krhg|d <tt|d D]}|d |}|d |jdqU|d rt|dd krg|d<tt|dD]}|d|}|d|jdq|drt|dd krg|d<tt|dD]}|d|}|d|jdqtj|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_indexrparsing_res_listrrrrrrrrr'r(r*r+)strr.r1appendr_to_str rargskwargsrrr2r3r8r:rrrrBOs8       zLayoutParsingResult._to_strc Or<)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=r>rr?rrrrrrrrr'r(r*r+)jsonr.r1rAr_to_jsonrCrrrrGxs8       zLayoutParsingResult._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. rrrrtable_region_idtable_pred)r.r1html)rrZ res_html_dictr2r3rIr9rrr_to_html  zLayoutParsingResult._to_htmlcCrH)zConverts the prediction HTML to an XLSX file path. Returns: Dict[str, str]: The str type XLSX representation result. rrrrrIrJrK)r.r1Zxlsx)rrZ res_xlsx_dictr2r3rIr9rrr_to_xlsxrNzLayoutParsingResult._to_xlsx)r N)__name__ __module__ __qualname____doc__r rr@npZndarrayr;rBrGrMrO __classcell__rrrrr s.))r )r/typingrnumpyrTZPILrrZ common.resultrrrr r rrrrs