jXjddlZddlmZddlZddlmZddlmZmZddl m Z ddgZ d a d e eefd dfd Z ddejd e eefdededededzdededed ejfdZ dde eefdededzdededed ejfdZ d de eefdededzd ejfdZdS)!N)Any)anyascii)Image ImageDraw)get_fontsynthesize_pagesynthesize_kie_pageFentryreturnczts1t|ddkrtjddadSdSdS)Ngeometryz@Polygons with larger rotations will lead to inaccurate renderingT)ROTATION_WARNINGlenloggingwarning)r s ]/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/utils/reconstitution.py_warn_rotationrsN  E*$5 6 6! ; ;Z[[[  ; ;?2responsewh draw_proba font_familysmoothing_factor min_font_size max_font_sizec t|ddkr |d\\} } \} } | | f| | f| | f| | fg} n|d} t| \}}ttt |zttt |zttt |zttt |zf\} } } } | | z }| | z }d|vr&dd|dD}n|d} t ||}t||}||dd\}}||ks||krh||krbt t||z|}t||}||dd\}}||ks||kr||kbn #t$rt||}YnwxYwtj dfd }tj |fd | Dd tj |} || | f||d dn9#t $r,|| | ft#||d dYnwxYwn'#t$$rt'jd|YnwxYw|rd|vr|dn4t+d|dDt|dz }td |z}d |z d |f}|| | f| | fg|dt|d}|d}||dd\}}||z dz} | |z dz }!t d |!}!|| | z|!f|||d|S)Nrwords c3&K|] }|dV dS)valueN).0words r z_synthesize..:s&FFtT']FFFFFFrr'rLrc g|]C\}}tt|ztt|zfDSr()intround)r)xyrrs r z_synthesize..LsG!]!]!]TQPQ3uQU||#4#4c%A,,6G6G"H!]!]!]r)fill)rrrlt)fontr4anchorzCould not render word: confidencec3&K|] }|dV dS)r8Nr()r)rs rr+z_synthesize..]s&==Q|_======r)outlinewidthz.2f)rzipr.r/minmaxjoinrgetbbox ValueErrorrnewrDrawpolygontextUnicodeEncodeErrorr Exceptionrrsum rectangle)"rr rrrrrr r!xminyminxmaxymaxrEx_coordsy_coords word_width word_height word_text font_sizer6 text_width text_heightmaskdr8pcolor prob_font prob_textprob_text_widthprob_text_height prob_x_offset prob_y_offsets" `` r _synthesizeras 5 ""%*:%6" tltT$<$d|dD\J #gHh E!c(mm# $ $%% E!c(mm# $ $%% E!c(mm# $ $%% E!c(mm# $ $%% D$d J+K%HHFFuW~FFFFF 'N  4 ]33  Y//"&,,y"9"9!A#"> KJ&&+ *C*CUbIbIbC ,< <==}MMIK33D&*ll9&=&=ac&B #J J&&+ *C*CUbIbIb 444 ]334 9S1a&! $ $D N4  !]!]!]!]!]U\!]!]!]dg hhh x  A? ^ FFD$<IdF S S S S! ^ ^ ^ FFD$<)!4!44iX\F ] ] ] ] ] ^ ???=)==>>>>>?ju$$ ,  ==eGn=====E'N@S@SS  j ! !q!Q dD\D$<0%q III["-- !'' ,5,=,=i,H,H1,M))$o5!; //!3 A}--  }$m4iiV[dhiii OsCB4G55HH9JK3K  K K  K!K43K4ffffff?pagec|d\}}tjd||fd}|dD]} t| ddkr3| dD])} t| t || ||||||| }*N| dD]4} t| | d D]} t || ||||||| }5t j|t j S) Draw a the content of the element page (OCR response) on a blank page. Args: page: exported Page object to represent draw_proba: if True, draw words in colors to represent confidence. Blue: p=1, red: p=0 font_family: family of the font smoothing_factor: factor to smooth the font size min_font_size: minimum font size max_font_size: maximum font size Returns: the synthesized page dimensionsRGBr3r3r3rZblockslinesr) rr rrrrrr r!r$dtype)rrCrrranparrayuint8) rdrrrr r!rrrblockliner*s rr r qs7,  DAqyAo>>>Hh uW~   " "g  u%%%&%) +%5"/"/    g  u%%% M  D*!)"#-$/)9&3&3   HH   8HBH - - --rc |d\}}tjd||fd}|dD]+}|D]&}t|t ||||||}',t j|t jS)rfrgrhrirj predictions)rr rrrrrm)rrCvaluesrrarorprq)rdrrrrrru predictions rr r s&  DAqyAo>>>HM*1133   %  J : & & &"! %' HH  8HBH - - --r)FNrrr)FNrbrcr)FN)rtypingrnumpyrorPILrrfontsr__all__rdictstrrr.boolfloatrandarrayr r r(rrrs)   3 4 $sCx. T    ""RRkR S>R R R  R t RRRR [RRRRn"" :.:. sCx.:.:.t:. :.  :.  :.Z:.:.:.:.~"".". sCx.".".t".Z ".".".".".".r