jjgddlmZddlmZeddlmZddlmZ ddlm Z ddl Z ddl Z ddlmZddlmZdd lmZmZdd lmZmZdd lmZ dd lmZmZn #e$rYnwxYwgd ZGddeZGddeZGddeZ GddeZ!GddeZ"GddeZ#GddeZ$GddeZ%GddeZ&Gd d!e&Z'dS)")Any) defuse_stdlib) ElementTree)Element) SubElementN)requires_package) BoundingBox)resolve_enclosing_bboxresolve_enclosing_rbbox)synthesize_kie_pagesynthesize_page) NestedObject)visualize_kie_pagevisualize_page) rWordArtefactLine PredictionBlockPageKIEPageDocumentceZdZUdZgZeeed<gZeeed<de ddfdZ de ee ffdZ e d e ee ffd Zdefd ZdS) rzVImplements an abstract document element with exporting and text rendering capabilities_children_names_exported_keyskwargsreturnNc |D]>\}}||jvrt||| t|jjd|ddS)Nz+ object does not have any attribute named '')itemsrsetattrKeyError __class____name__)selfrkvs T/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/io/elements.py__init__zElement.__init__&ssLLNN l lDAqD(((a####$."9jjfgjjjkkk  l lcfdjD}jD]S}|dvr0dt|D||<6dt|D||<T|S)z,Exports the object into a nested dict formatc2i|]}|t|S)getattr).0r&r%s r( z"Element.export../s%HHHqq'$**HHHr* predictionsc.i|]\}}|d|DS)c6g|]}|Sr-export)r/items r( z-Element.export...3s 444$ 444r*r-)r/r&cs r(r0z"Element.export..2s:...9=AA44!444...r*c6g|]}|Sr-r5)r/r9s r(r8z"Element.export..6s -_-_-_Qahhjj-_-_-_r*)rrr.r )r% export_dict children_names` r(r6zElement.export-sHHHHD4GHHH !1 ` `M//..AH}A]A]AcAcAeAe... M**.`-_'$P]B^B^-_-_-_ M**r* save_dictc tNNotImplementedErrorclsr=rs r( from_dictzElement.from_dict:s!!r*ctr?r@r%s r(renderzElement.render>s!!r*)r$ __module__ __qualname____doc__rliststr__annotations__rrr)dictr6 classmethodrDrGr-r*r(rr s``!#OT#Y### "NDI"""llllll S#X    "$sCx."""[""""""""r*rc eZdZUdZgdZeeed<gZeeed<dede de e j zde d e eefd d f fd Zd efd Zd efdZede eeffdZxZS)raImplements a word element Args: value: the text string of the word confidence: the confidence associated with the text prediction geometry: bounding box of the word in format ((xmin, ymin), (xmax, ymax)) where coordinates are relative to the page's size objectness_score: the objectness score of the detection crop_orientation: the general orientation of the crop in degrees and its confidence )value confidencegeometryobjectness_scorecrop_orientationrrrQrRrSrTrUrNct||_||_||_||_||_dSr?)superr)rQrRrSrTrU)r%rQrRrSrTrUr#s r(r)z Word.__init__QsF  $  0 0r*c|jS$Renders the full text of the elementrQrFs r(rGz Word.render` zr*c(d|jd|jdS)Nvalue='', confidence=.2rQrRrFs r( extra_reprzWord.extra_reprdsGGG4?GGGGr*r=c :fd|jD}|di|S)Nc"i|] }|| Sr-r-r/r&r=s r(r0z"Word.from_dict..i>>>a!Yq\>>>r*r-rrBs ` r(rDzWord.from_dictg1>>>>3+=>>>s}}V}}r*)r$rHrIrJrrKrLrMrfloatr npndarrayrNrr)rGrbrOrD __classcell__r#s@r(rrBs-  !l k kNDIkkk!#OT#Y### 1 1 1 * 1  1 sCx. 1  1 1 1 1 1 1HCHHHH$sCx.[r*rceZdZUdZgdZeeed<gZeeed<dede de dd ffd Z defd Z defd Z ed eeeffdZxZS)ra'Implements a non-textual element Args: artefact_type: the type of artefact confidence: the confidence of the type prediction geometry: bounding box of the word in format ((xmin, ymin), (xmax, ymax)) where coordinates are relative to the page's size. )rStyperRrr artefact_typerRrSrNcrt||_||_||_dSr?)rWr)rSrorR)r%rprRrSr#s r(r)zArtefact.__init__zs2   ! $r*c<d|jdS)rZ[])roupperrFs r(rGzArtefact.renders'49??$$''''r*c(d|jd|jdS)Nztype='r_r`)rorRrFs r(rbzArtefact.extra_reprsE EEEEEEr*r=c :fd|jD}|di|S)Nc"i|] }|| Sr-r-res r(r0z&Artefact.from_dict..rfr*r-rgrBs ` r(rDzArtefact.from_dictrhr*)r$rHrIrJrrKrLrMrrir r)rGrbrOrNrrDrlrms@r(rrms!C B BNDIBBB!#OT#Y###%c%u% %X\%%%%%% (((((FCFFFF$sCx.[r*rc eZdZUdZddgZeeed<dgZeeed<gZ ee ed< d dee de e j zdzdedzddffd Zdefd Zed eeeffd ZxZS)ramImplements a line element as a collection of words Args: words: list of word elements geometry: bounding box of the word in format ((xmin, ymin), (xmax, ymax)) where coordinates are relative to the page's size. If not specified, it will be resolved by default to the smallest bounding box enclosing all words in it. rSrTrwordsrNrcH|+ttjd|D}|At|djdkrt nt }|d|D}t|||_||_ dS)Ncg|] }|j Sr-)rTr/ws r(r8z!Line.__init__..s-P-P-PQa.@-P-P-Pr*rcg|] }|j Sr-rSr}s r(r8z!Line.__init__..s)D)D)D!*)D)D)Dr*)rz) rirjmeanlenrSr r rWr)rT)r%rzrSrTbox_resolution_fnr#s r(r)z Line.__init__s  #$RW-P-P%-P-P-P%Q%QRR   ;>uQx?P;Q;QUV;V;V 7 7\r (()D)De)D)D)DEEH u%%%  0r*cJdd|jDS)rZ c3>K|]}|VdSr?rGr}s r( zLine.render..s*77q 777777r*)joinrzrFs r(rGz Line.renders%xx77DJ777777r*r=c fd|jD}|dddDi|di|S)Nc"i|] }|| Sr-r-res r(r0z"Line.from_dict..rfr*rzcBg|]}t|Sr-)rrDr/_dicts r(r8z"Line.from_dict..$LLLdnnU++LLLr*r-rupdaterBs ` r(rDzLine.from_dictsg>>>>3+=>>> LL79KLLL    s}}V}}r*NN)r$rHrIrJrrKrLrMrrzrr rjrkrir)rGrOrNrrDrlrms@r(rrs ",-? @NDI@@@")OT#Y***E4: 59)- 11Dz1 *T11 $, 1  111111&88888$sCx.[r*rc*eZdZdZdefdZdefdZdS)rzImplements a prediction elementrc|jSrYr[rFs r(rGzPrediction.renderr\r*c8d|jd|jdd|jS)Nr^r_r`z, bounding_box=)rQrRrSrFs r(rbzPrediction.extra_reprs+eee4?eeeVZVceeer*N)r$rHrIrJrLrGrbr-r*r(rrsW))fCffffffr*rc &eZdZUdZddgZeeed<ddgZeeed<gZ ee ed<gZ ee ed<ggddfdee dee de ejzdzdedzd df fd Zdd ed efd ZedeeeffdZxZS)raImplements a block element as a collection of lines and artefacts Args: lines: list of line elements artefacts: list of artefacts geometry: bounding box of the word in format ((xmin, ymin), (xmax, ymax)) where coordinates are relative to the page's size. If not specified, it will be resolved by default to the smallest bounding box enclosing all lines and artefacts in it. rSrTrlines artefactsrNrcz|+ttjd|D}|Yd|D}d|D}t|djtjrt nt}|||z}t ||||_||_ dS)Nc0g|]}|jD] }|j Sr-)rzrT)r/liner~s r(r8z"Block.__init__..s--g-g-gT\`\f-g-gWXa.@-g-g-g-gr*c0g|]}|jD] }|j Sr-)rzrS)r/rwords r(r8z"Block.__init__..s)QQQDdjQQd$-QQQQr*cg|] }|j Sr-r)r/artefacts r(r8z"Block.__init__..sJJJHh/JJJr*rrr) rirjr isinstancerSrkr r rWr)rT) r%rrrSrT line_boxesartefact_boxesrr#s r(r)zBlock.__init__s  #$RW-g-gU-g-g-g%h%hii   QQEQQQJJJ JJJN+5eAh6G+T+Tp''Zp )(n)DEEH u :::  0r* line_breakcJ|d|jDS)rZc3>K|]}|VdSr?r)r/rs r(rzBlock.render..s*DDt{{}}DDDDDDr*)rr)r%rs r(rGz Block.renders%DDDDDDDDr*r=c fd|jD}|ddDddDd|di|S)Nc"i|] }|| Sr-r-res r(r0z#Block.from_dict..rfr*cBg|]}t|Sr-)rrDrs r(r8z#Block.from_dict..rr*rcBg|]}t|Sr-)rrDrs r(r8z#Block.from_dict..s&XXX(,,U33XXXr*rrr-rrBs ` r(rDzBlock.from_dicts>>>>3+=>>> LL79KLLLXX;AWXXX     s}}V}}r*)r)r$rHrIrJrrKrLrMrrrrrr rjrkrir)rGrOrNrrDrlrms@r(rrs^",-? @NDI@@@");!7OT#Y777E4: "ItH~"""$&48)- 11Dz1>1 *T1 1  $, 1  111111.EEEEEEE$sCx.[r*rceZdZUdZgdZeeed<dgZeeed<gZ ee ed< dde j dee de d ee e fd eeefdzd eeefdzd dffd Zdded efdZd efdZddeded dfdZd e j fdZd ded eeejffdZedeeeffdZxZS)!raImplements a page element as a collection of blocks Args: page: image encoded as a numpy array in uint8 blocks: list of block elements page_idx: the index of the page in the input raw document dimensions: the page size in pixels in format (height, width) orientation: a dictionary with the value of the rotation angle in degress and confidence of the prediction language: a dictionary with the language value and confidence of the prediction page_idx dimensions orientationlanguagerblocksrNpagerrrrrc*t|||_||_||_t |t r|nt dd|_t |t r|nt dd|_dS)N)rra rWr)rrrrrNrr)r%rrrrrrr#s r(r)z Page.__init__ s '''   $*4[$*G*Gn;;TX\imMnMnMn$.x$>$>eDt`dDeDeDe r* block_breakcJ|d|jDS)rZc3>K|]}|VdSr?r)r/bs r(rzPage.render..s*@@q @@@@@@r*)rr)r%rs r(rGz Page.renders'@@DK@@@@@@r*cd|jSNz dimensions=rrFs r(rbzPage.extra_repr.T_...r*TF interactivepreserve_aspect_ratioc tddtddddlm}t||j|||jdi|dS)a.Overlay the result on a given image Args: interactive: whether the display should be interactive preserve_aspect_ratio: pass True if you passed True to the predictor **kwargs: additional keyword arguments passed to the matplotlib.pyplot.show method matplotlib4`.show()` requires matplotlib & mplcursors installed mplcursorsrNrrr-)rmatplotlib.pyplotpyplotrr6rshowr%rrrplts r(rz Page.show"sx ']^^^']^^^''''''t{{}}di[`uvvvv6r*c @t|fi|S)zSynthesize the page from the predictions Args: **kwargs: keyword arguments passed to the `synthesize_page` method Returns: synthesized page )r r6r%rs r( synthesizezPage.synthesize1s"t{{}}77777r*docTR - XML export (hOCR) file_titlec|j}d}d}d}|j\}}d|jvr|jnd}t ddt |d} t | d} |t | d _t | d d d d t | d ddtj dt | d dddt | d} t | ddd|dzd|d|dd} |j D]} t| j dkrtd| j \\}}\}}t | ddd|dtt||zdtt||zd tt||zdtt||zd}t |d!d"d#|dtt||zdtt||zd tt||zdtt||zd}|dz }| jD]}|j \\}}\}}t |d$d%d&|dtt||zdtt||zd'tt||zdtt||zd( d}|dz }|jD]}|j \\}}\}}|j}t |d$d)d*|dtt||zdtt||zd+tt||zdtt||zd,tt|d-z d}|j|_|dz }t)j| d.d/0t)j| fS)1 Export the page as XML (hOCR-format) convention: https://github.com/kba/hocr-spec/blob/master/1.2/spec.md Args: file_title: the title of the XML file Returns: a tuple of the XML byte string, and its ElementTree renhtmlhttp://www.w3.org/1999/xhtmlxmlnszxml:langattribheadtitlemeta Content-Typetext/html; charset=utf-8z http-equivcontent ocr-system python-doctr namerocr-capabilities-ocr_page ocr_carea ocr_par ocr_line ocrx_wordbodydivocr_pagepage_image; bbox 0 0 r ; ppageno 0classidrAXML export is only available for straight bounding boxes for now. ocr_careablock_bbox z pocr_parpar_spanocr_lineline_ O; baseline 0 0; x_size 0; x_descenders 0; x_ascenders 0 ocrx_wordword_z z&; x_wconf dutf-8xmlencodingmethod)rrrkeys ETElementrLrtextdoctr __version__rrrS TypeErrorintroundrrzrRrQETtostringr)r%rp_idx block_count line_count word_countheightwidthr page_hocrrrpage_divblockxminyminxmaxymax block_div paragraphr line_spanrconfword_divs r( export_as_xmlzPage.export_as_xml<s     $.$-2D2D2F2F$F$F4==Df7Ucfgocpcp-q-qrrr )V,,)3 4!!&4~Rl(m(mnnnn  (5XUEV5X5XYY      .;jkk    )V,,  #)eai))GEGGFGGG   [8 $8 $E5>""a'' cddd). <4,4"(0;00LSte|)<)<%=%=LLE$QW-DXDX@Y@YLLte|,,--LL03E$-4H4H0I0ILL   I#&...LSte|)<)<%=%=LLE$QW-DXDX@Y@YLLte|,,--LL03E$-4H4H0I0ILL   I 1 K  $ $-1]* tltT&!+2j22"OU4%<-@-@)A)A"O"OCdU[mH\H\D]D]"O"OU4%<0011"O"O47dVm8L8L4M4M"O"O"O    a  J$$D15.LT4,4?D)!%0"6*"6"6&>Ste|1D1D-E-E&>&>ERVY_R_L`L`HaHa&>&> te|!4!455&>&>8;E$-&>%(tcz):):%;%;&>&>     H%)JHM!OJJ!$ $B IFFFW`HaHabbr*r=c fd|jD}|dddDi|di|S)Nc"i|] }|| Sr-r-res r(r0z"Page.from_dict..rfr*rcBg|]}t|Sr-)rrD)r/ block_dicts r(r8z"Page.from_dict..s$!d!d!d*%//*"="=!d!d!dr*r-rrBs ` r(rDzPage.from_dicts^>>>>3+=>>> x!d!dPYZbPc!d!d!defffs}}V}}r*rrTFr)r$rHrIrJrrKrLrMrrrrjrkr tuplerNrr)rGrbboolrrbytesrrr"rOrDrlrms@r(rrs   !V U UNDIUUU"*OT#Y+++FDK.2*.ffjfU f f #s(O f #s(^d* fsCx.4'f ffffff AA#A3AAAA/C////   D _c     8bj 8 8 8 8fcfcfceTY[][iTiNjfcfcfcfcP$sCx.[r*rceZdZUdZgdZeeed<dgZeeed<iZ e eee fed< dde j de eee fded eeefd e eefdzd e eefdzd dffd Zdded efdZd efdZddeded dfdZd e j fdZd ded eeejffdZede eeffdZxZS)!ra*Implements a KIE page element as a collection of predictions Args: predictions: Dictionary with list of block elements for each detection class page: image encoded as a numpy array in uint8 page_idx: the index of the page in the input raw document dimensions: the page size in pixels in format (height, width) orientation: a dictionary with the value of the rotation angle in degress and confidence of the prediction language: a dictionary with the language value and confidence of the prediction rrr2rNrrrrrrc*t|||_||_||_t |t r|nt dd|_t |t r|nt dd|_dS)Nr1rar)r%rr2rrrrr#s r(r)zKIEPage.__init__s [111   $*4[$*G*Gn;;TX\imMnMnMn$.x$>$>eDt`dDeDeDe r*rprediction_breakcn|d|jDS)rZc3XK|]%\}}|D]}|d|V&dS)z: Nr)r/ class_namer2rs r(rz!KIEPage.render..sb% % .Ej+kv% % fgz ) )QXXZZ ) )% % % % % % % r*)rr2r )r%r/s r(rGzKIEPage.rendersE$$% % IMIYI_I_IaIa% % %    r*cd|jSrrrFs r(rbzKIEPage.extra_reprrr*TFrrc tddtddddlm}t||j|||jdi|dS)a#Overlay the result on a given image Args: interactive: whether the display should be interactive preserve_aspect_ratio: pass True if you passed True to the predictor **kwargs: keyword arguments passed to the matplotlib.pyplot.show method rrrrNrr-)rrrrr6rrrs r(rz KIEPage.shows ']^^^']^^^'''''' KKMM49+Uj     6r*c @t|fi|S)zSynthesize the page from the predictions Args: **kwargs: keyword arguments passed to the `synthesize_kie_page` method Returns: synthesized page )r r6rs r(rzKIEPage.synthesizes"#4;;==;;F;;;r*rrcT|j}d}|j\}}d|jvr|jnd}t ddt |d}t |d}|t |d _t |d d d d t |d ddtj dt |d dddt |d} t | ddd|dzd|d|dd|j D]\} } | D]} t| j dkrtd| j \\} }\}}t | dd| d|dtt!| |zdtt!||zd tt!||zdtt!||zd}t |d!d"| d#|dtt!| |zdtt!||zd tt!||zdtt!||zd}t |d$d%| d&|dtt!| |zdtt!||zd tt!||zdtt!||zd' d}t |d$d(| d)|dtt!| |zdtt!||zd tt!||zdtt!||zd*tt!| jd+z d}| j|_|dz }ߐt'j|d,d-.t'j|fS)/rrrrrrrrrrrrrrrrrrrrrrrrrrrrrr _prediction_rrrr_par_rr_line_rr_word_z"; x_wconf rrrr)rrrrrrLrrr r r2r rrSr r r rRrQrrr)r%rrprediction_countrrrrrrr2r2 predictionrrrrprediction_divpar_divrr!s r(r"zKIEPage.export_as_xmlsu  ! $.$-2D2D2F2F$F$F4==Df7Ucfgocpcp-q-qrrr )V,,)3 4!!&4~Rl(m(mnnnn  (5XUEV5X5XYY      .;jkk    )V,,  #)eai))GEGGFGGG    (,'7'='='?'?2 &2 & #J )1 &1 & z*++q00#$ghhh-7-@* tltT!+!,!+KK9IKK"PU4%<-@-@)A)A"P"PCdU[mH\H\D]D]"P"PU4%<0011"P"P47dVm8L8L4M4M"P"P " " "%"!*!+DD2BDD"PU4%<-@-@)A)A"P"PCdU[mH\H\D]D]"P"PU4%<0011"P"P47dVm8L8L4M4M"P"P   '!+!+EE3CEE"OU4%<-@-@)A)A"O"OCdU[mH\H\D]D]"O"OU4%<0011"O"O47dVm8L8L4M4M"O"O"O    &!,!+EE3CEE"KU4%<-@-@)A)A"K"KCdU[mH\H\D]D]"K"KU4%<0011"K"K47dVm8L8L4M4M"K"K!$U:+@3+F%G%G!H!H"K"K   !+ 0  A%  c1 &f{9wuEEEr~V_G`G```r*r=c fd|jD}|dddDi|di|S)Nc"i|] }|| Sr-r-res r(r0z%KIEPage.from_dict..Rrfr*r2cBg|]}t|Sr-)rrD)r/predictions_dicts r(r8z%KIEPage.from_dict..Ts(tttGWJ001ABBtttr*r-rrBs ` r(rDzKIEPage.from_dictPsi>>>>3+=>>> tt[der[sttt    s}}V}}r*rr'r(r))r$rHrIrJrrKrLrMrr2rNrrjrkr r*rr)rGrbr+rrr,rrr"rOrDrlrms@r(rrs/  !V U UNDIUUU"/OT#Y000/1Kc4 ++,111.2*.ffjf#tJ//0f f #s(O f #s(^d* fsCx.4'f ffffff   s     /C////D_c" K|]}|VdSr?r)r/rs r(rz"Document.render..ks*>>aqxxzz>>>>>>r*)rrE)r%rMs r(rGzDocument.renderis%>>4:>>>>>>r*c 4|jD]}|jdi|dS)z#Overlay the result on a given imageNr-)rEr)r%rresults r(rz Document.showms8j " "F FK ! !& ! ! ! ! " "r*c *fd|jDS)zSynthesize all pages from their predictions Args: **kwargs: keyword arguments passed to the `Page.synthesize` method Returns: list of synthesized pages c*g|]}|jdiSr-)rr/rrs r(r8z'Document.synthesize..{s+AAAd))&))AAAr*rHrs `r(rzDocument.synthesizers"BAAAdjAAAAr*c *fd|jDS)zExport the document as XML (hOCR-format) Args: **kwargs: additional keyword arguments passed to the Page.export_as_xml method Returns: list of tuple of (bytes, ElementTree) c*g|]}|jdiSrT)r"rUs r(r8z*Document.export_as_xml..s-DDD"",,V,,DDDr*rHrs `r(r"zDocument.export_as_xml}s"EDDDDDDDr*r=c fd|jD}|dddDi|di|S)Nc"i|] }|| Sr-r-res r(r0z&Document.from_dict..rfr*rEcBg|]}t|Sr-)rrD)r/ page_dicts r(r8z&Document.from_dict..s$ _ _ _y !:!: _ _ _r*r-rrBs ` r(rDzDocument.from_dicts]>>>>3+=>>> w _ _IV]L^ _ _ _`aaas}}V}}r*)rL)rN)r$rHrIrJrrKrLrMrErr)rGrrjrkrr*r,rrr"rOrNrrDrlrms@r(rrYsM #*OT#Y***E4:&Dz& &&&&&& ???c????"""" Bd2:&6 B B B B EeE2>4I.J)K E E E E$sCx.[r*rcpeZdZUdZdgZeeed<gZee ed<dee ddffd Z xZ S) KIEDocumentrDrErrNcLt|dSrGrIrJs r(r)zKIEDocument.__init__rKr*) r$rHrIrJrrKrLrMrErr)rlrms@r(r]r]s #*OT#Y***E4=&G}& &&&&&&&&&&r*r])(typingr defusedxmlr xml.etreerrxml.etree.ElementTreerrrnumpyrjr doctr.file_utilsrdoctr.utils.common_typesr doctr.utils.geometryr r doctr.utils.reconstitutionr r doctr.utils.reprrdoctr.utils.visualizationrrModuleNotFoundError__all__rrrrrrrrr]r-r*r(rls $$$$$$ ''''''666666,,,,,, ------000000PPPPPPPPKKKKKKKK)))))) LLLLLLLLL   D  h g g"""""l"""D(((((7(((Vw@+++++7+++\ffffffff22222G222jmmmmm7mmm`kkkkkgkkk\33333w333l&&&&&(&&&&&sAA%$A%