{-j!jddlZddlmZddlZddlmZmZddlm Z m Z m Z m Z Gdde e e Z dS)N)Dict)Image ImageDraw) BaseCVResult HtmlMixin JsonMixin XlsxMixinceZdZdZd fd ZdeeejffdZ deeeffdZ deeeffdZ deeeffdZ deeeffd Z xZS) LayoutParsingResultzLayout Parsing ResultreturnNct|tj|tj|dS)z@Initializes a new instance of the class with the specified data.N)super__init__rr )selfdata __class__s q/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/pipelines/layout_parsing/result.pyrzLayoutParsingResult.__init__sC 4   4     c Ti}|d}|dr|jdi|dj|djd|d<|djd|d<|drt|d d krtjt j|dd ddddddd f}tj|}d }tt|d D]J}|d |}|d}|D]/} d| D\} } } } | | | | | g|d0K||d<|drkt|dd krRtt|dD]/}|d|}|d}|j}d|}|d||<0|drkt|dd krRtt|dD]/}|d|}|d}|j}d|}|d||<0|S)Nmodel_settingsuse_doc_preprocessordoc_preprocessor_reslayout_det_resresoverall_ocr_res ocr_res_imguse_table_recognitiontable_res_listr output_img)rr cell_box_listc,g|]}t|S)int).0poss r z/LayoutParsingResult._to_img..4s%>%>%>3c#hh%>%>%>r)outlinewidthtable_cell_imguse_seal_recognition seal_res_listseal_region_idseal_res_regionuse_formula_recognitionformula_res_listformula_region_idformula_res_regionr%) updateimglenr fromarraycopydeepcopyrDrawrange rectangle)r res_img_dictrr- table_drawrectangle_colorsno table_resr#boxx1y1x2y2seal_resr0sub_seal_res_dictkey formula_resr4sub_formula_res_dicts r_to_imgzLayoutParsingResult._to_img!s ./ 0 1 D L  C C$'=">"B C C C)-.>)?)CE)J %&*./@*A*Em*T &' 1 2 %>#%>%>%>NBB((RR(/) .+<>>$8$? S!!rc&i}|d|d<|d|d<|d}||d<|d|d<|ddr|djd|d<|djd|d<|d jd|d <|d rwt|d d kr^g|d <tt|d D]6}|d |}|d |jd7|d rwt|dd kr^g|d<tt|dD]6}|d|}|d|jd7|drwt|dd kr^g|d<tt|dD]6}|d|}|d|jd7t j|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/r2r3)strr8r=appendr _to_str rargskwargsrrrBrCrIrLs rrUzLayoutParsingResult._to_strOst!,/\!,/\./!/ #'(:#;   !"8 9 S+/0F+G+KE+RD' (!%&6!7!;E!B "&'8"9"=e"D  1 2 Ds4@P;Q7R7RUV7V7V%'D! "S&6!78899 D D !1237 %&--imE.BCCCC 0 1 Bc$:O6P6PST6T6T$&D !So!67788 B B05_%,,X\%-@AAAA 4 5 HD+,--11')D# $S&8!9::;; H H"#56s; '(// 0FGGGG 7777777rc&i}|d|d<|d|d<|d}||d<|d|d<|ddr|djd|d<|djd|d<|d jd|d <|d rwt|d d kr^g|d <tt|d D]6}|d |}|d |jd7|d rwt|dd kr^g|d<tt|dD]6}|d|}|d|jd7|drwt|dd kr^g|d<tt|dD]6}|d|}|d|jd7t 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. rPrQrrRrrrrrrrrr.r/r2r3)jsonr8r=rTr _to_jsonrVs rr[zLayoutParsingResult._to_jsonxsu!,/\!,/\./!/ #'(:#;   !"8 9 T+/0F+G+LU+SD' (!%&6!7!u"E  1 2 Es4@P;Q7R7RUV7V7V%'D! "S&6!78899 E E !1237 %&--inU.CDDDD 0 1 Cc$:O6P6PST6T6T$&D !So!67788 C C05_%,,X]5-ABBBB 4 5 ID+,--11')D# $S&8!9::;; I I"#56s; '(// 0@0GHHHH!$8888888rc|d}i}|drit|ddkrPtt|dD]-}|d|}|d}d|}|jd||<.|S)zConverts the prediction to its corresponding HTML representation. Returns: Dict[str, str]: The str type HTML representation result. rrrrtable_region_idtable_pred)r8r=html)rr res_html_dictrBrCr]rKs r_to_htmlzLayoutParsingResult._to_html ./ 1 2 '9'9'9'9R$sCx. $sCx.rr )r:typingrnumpyrlPILrr common.resultrrr r r r%rrrss  JJJJJJJJJJJJggggg, 9gggggr