`j`VddlZddlmZmZmZmZmZmZddlm Z ddl m Z m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z dZ!d Z"d Z#id d d dddddddddddddddddddd d!d"d!d#d$d%d$d&d$d'd$d(d$iZ$Gd)d*Z%Gd+d,Z&Gd-d.Z'y)/N)AnyDictListOptionalTupleUnion)Errors) escape_html minify_htmlregistry) TPL_DEP_ARCS TPL_DEP_SVG TPL_DEP_WORDSTPL_DEP_WORDS_LEMMATPL_ENT TPL_ENT_RTLTPL_ENTS TPL_FIGURE TPL_KB_LINKTPL_PAGETPL_SPAN TPL_SPAN_RTLTPL_SPAN_SLICETPL_SPAN_SLICE_RTLTPL_SPAN_STARTTPL_SPAN_START_RTL TPL_SPANS TPL_TITLEenltrz#dddORGz#7aececPRODUCTz#bfeeb7GPEz#feca74LOCz#ff9561PERSONz#aa9cfcNORPz#c887fbFACz#9cc9ccEVENTz#ffeb80LAWz#ff8197LANGUAGE WORK_OF_ARTz#f0d0ffDATEz#bfe1d9TIMEMONEYz#e4e7d2QUANTITYORDINALCARDINALPERCENTc 2eZdZdZdZifdeeefddfdZ dde eeefde d e defd Z d e ed e eeefd e edefdZ ed e ed e eeefde eee eeefffdZde eeefdefdZde edefdZde edefdZy) SpanRendererzRender Spans as SVGs.spanoptionsreturnNctt}tjj }|j D]g}t |r|}t|ts2ttjjt||j|i|j|jdit|_|j#Dcic]\}}|j%|c}}|_t(|_t,|_|jdd|_|jdd|_|jdd|_|jd }|r|d |_|d |_|d |_y|j*d k(r"t<|_t>|_t@|_ytB|_tD|_tF|_ycc}}w)zfInitialise span renderer options (dict): Visualiser-specific options (colors, spans) objcolors top_offset(span_label_offsettop_offset_steptemplater7slicestartrtlN)$dictDEFAULT_LABEL_COLORSr displacy_colorsget_allvaluescallable isinstance ValueErrorr E925formattypeupdategetDEFAULT_ENTITY_COLOR default_coloritemsupperr= DEFAULT_DIR direction DEFAULT_LANGlangr>r@ offset_step span_templatespan_slice_templatespan_start_templaterrrrrr)selfr8r= user_colors user_colorlabelcolorrDs _/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/spacy/displacy/render.py__init__zSpanRenderer.__init__9s *+..668 %,,. &J #(\ j$/ !3!3Z8H!3!IJJ MM* % &  gkk(B/01@F O uu{{}e+O %  !++lB7!(-@"!E";;'8"=;;z* !)&!1D '/'8D $'/'8D $~~&%1"+=(+=(%-"+9(+9(3Ps'G+parsedpageminifyc Bg}t|D]\}}|dk(rH|jdi}|jdt|_|jdt|_|j |j|d|d|jd|r`dj|Dcgc]}tj| c}} tj| |j |j } ndj|} |r t| S| Scc}w) Render complete markup. parsed (list): Dependency parses to render. page (bool): Render parses wrapped as full HTML page. minify (bool): Minify HTML markup. RETURNS (str): Rendered SVG or HTML markup. rsettingsrZr\tokensspanstitlecontentrsr\dir) enumeraterTrYrZr[r\append render_spansjoinrrQrr rarhrirjrenderediprmdocdocsmarkups rfrenderzSpanRenderer.renderfsf% XDAqAv55R0!)k;!G$LL> OOD--ak1W:quuW~V W  X 77hOsJ--c:OPD__T t~~VFWWX&F v& & P+Drnrorpc|j||}|j|}tj||j}|rt j||z}|S)aRender span types in text. Spans are rendered per-token, this means that for each token, we check if it's part of a span slice (a member of a span type) or a span start (the starting token of a given span type). tokens (list): Individual tokens in the text spans (list): Individual entity spans and their start, end, label, kb_id and kb_url. title (str / None): Document title set in Doc.user_data['title']. rsrurp)_assemble_per_token_info_render_markuprrQrZr )rarnrorpper_token_infors rfrxzSpanRenderer.render_spanssY 66vuE$$^4!!&dnnE %%E2V;F c(g}t|d}|D]}d|d< t|D]\}}i}||d<g}g}|D]} i} | d|cxkr| dkrnn|| dk(} | d| d<| | d <| rt|r|d dndd z| d<|j| | d| d<| j d d } | j dd} | rt j | | nd | d<|j| d| d<||d<|j||S)a6Assembles token info used to generate markup in render_spans(). tokens (List[str]): Tokens in text. spans (List[Dict[str, Any]]): Spans in text. RETURNS (List[Dict[str, List[Dict, str, Any]]]): Per token info needed to render HTML markup for given tokens and spans. c,|d|d|dz |dfS)N start_token end_tokenrd)ss rfz7SpanRenderer._assemble_per_token_info..s,- K.1]#334' rkeyr render_slottextrrrdis_startrkb_idrqkb_url#rrkb_linkentities)sortedrvlenrwrTrrQ)rnrorridxtoken token_markupintersecting_spansrr7ent span_startrrs rfrz%SpanRenderer._assemble_per_token_infosAC    !A !Am  ! $F+& 0JC,.L#(L 79 H , &#A[0AA!$](;!;J#'=CL&0C O! ##56/r2=A!" /]+ '--d3)-m).s q?Orrrr)r span_slices span_starts total_height ) rstrip_get_span_slices_get_span_startsr>r@r]rr^rQr ) rarrrr is_whitespaceslicesstartsrs rfrzSpanRenderer._render_markups# ;EeJ/5OPH"&M//1R7M ..uZ/@A..uZ/@AOO,,-''3x=1+<=? $,,33$U6]3 & &!- 4+eFmc&9::+ ;, rrcDg}|D]}|jj|dj|j}|j|j |ddz zz}|j j||}|j|dj|S)z*Get the rendered markup of all Span slicesrdrr)bgr>rq) r=rTrXrVr>r]r_rQrwry)rarrentityrer> span_slices rfrzSpanRenderer._get_span_slicess  +F KKOOF7O$9$9$;T=O=OPE  F=$9A$=>J1188%9J   z * +ww{##rcbg}|D]}|jj|dj|j}|j|j |ddz zz}|dr%|j j|||d|dnd}|j|dj|S)z0Get the rendered markup of all Span start tokensrdrrrr)rr>rdrrq) r=rTrXrVr>r]r`rQrwry)rarrrrer>rs rfrzSpanRenderer._get_span_starts s  +FKKOOF7O$9$9$;T=O=OPE  F=$9A$=>J*% ((//) /"9- 0    z * + ww{##rFF)__name__ __module__ __qualname____doc__stylerstrrrgrboolrrrx staticmethodrrrrrrrfr6r64sO E13+:S#X+:+:\PU4S>*26HL 6S DcN#}  .DS D"&tCH~"6D d3T#s(^,,- .DDLT$sCx.-Ac6$d$$&$d$$rr6c PeZdZdZdZifdeeefddfdZ d!de eeefde d e defd Z d e e efd e eeefd e eeefdefdZdededeede def dZdede de dede def dZde de de de def dZdede de de def dZd e eeefdeee e efe ffd Zy)"DependencyRendererz!Render dependency parses as SVGs.depr8r9NcP|jdd|_|jdd|_|jd|jrdnd|_|jd|jrd nd |_|jd d |_|jd |jrdnd|_|jdd|_|jdd|_|jdd|_ |jdd|_ t|_ t|_y)zInitialise dependency renderer. options (dict): Visualiser-specific options (compact, word_spacing, arrow_spacing, arrow_width, arrow_stroke, distance, offset_x, color, bg, font) compactF word_spacing- arrow_spacing rA arrow_width arrow_stroker distanceoffset_x2rez#000000rz#fffffffontArialN)rTrrrrrrrrerrrYrZr[r\)rar8s rfrgzDependencyRenderer.__init__(s{{9e4 #KK;$[[ "RTU";;}4< $+Qqc*I//)QwZ6CCuuW~&&QUU7^,>>ARARR ]]SZ$--%?? mma$*;*;&;; "%( 1   QvY%!%%2F J  "$ 1   aj!G*ah%! L  '')$rwwx'88!!ww**;;**ww    s 4G+Grrrr|c(|j|jz}|j||jzz}|jdk(r|j |z }t |}|tj|||||Stj||||S)zRender individual word. text (str): Word text. tag (str): Part-of-speech tag. i (int): Unique ID, typically word index. RETURNS (str): Rendered SVG markup. rG)rrrxy)rrrr) rrrrrZrr rrQr)rarrrr|rr html_texts rfrzDependencyRenderer.render_words MMD-- - MMA - - >>U " QA%  &--CuQ ##qAFFrrdrFrrZc |dks|dkr7t||||}ttjjd i||j |||f}|j ||jzz|jz}|jdk(r|j|z }|j} |j ||z |jzz||jzz|j|j|z zdz z } |jdk(r|j| z } |j||jzdz z } |jr|j||jzdz z } | dk(r5t|j jddkDr |j } |j!||| | } |j#|| | | } |jdk(rd nd }t%j|j&||j(| ||| S) a5Render individual arrow. label (str): Dependency label. start (int): Index of start word. end (int): Index of end word. direction (str): Arrow direction, 'left' or 'right'. i (int): Unique ID, typically arrow index. RETURNS (str): Rendered SVG markup. r)rFrrdrurGr rrrightleft)rr|strokeheadrd label_sidearcr)rHrOr E157rQrrrrrZrrrrrrL get_arrowheadget_arcrrr)rardrFrrZr| error_argslevelx_startrx_endy_curve arrowheadrrs rfrzDependencyRenderer.render_arrows 19aEs%YOJV[[//=*=> > UC/0--%$--"77$:L:LL >>U "jj7*G MM MMU{dmm+ ,dmm# $  D$6$6$>?!C D  >>U "JJ&E--%$--"7!";; <<mmedmm&;a&??G a.sE S\0IrrrFrrd)tuplerrWrHrrange) rarrtlength max_levelrrr|s rfrzDependencyRenderer.get_levelssIM!M%syy{(;"*!26!HL! !F& c?& DcN#& 4S>" & & PGG#GhsmGGPSG&, , !$, +., ;>, CF, , \ Cs Cs CS C C C Ys Ys Ys Y Y YtDcN3U3S==QSV=V8Wrrc eZdZdZdZifdeeefddfdZ dde eeefde d e defd Z d ed e eeefd e edefdZ y)EntityRendererzRender named entities as HTML.rr8r9Nc|tt}tjj }|j D]g}t |r|}t|ts2ttjjt||j|i|j|jdit|_|j#Dcic]\}}|j%|c}}|_|jdd|_|j(-|j(Dcgc]}|j%c}|_t*|_t.|_|jd}|r||_y|j,dk(r t4|_yt6|_ycc}}wcc}w)zhInitialise entity renderer. options (dict): Visualiser-specific options (colors, ents) r;r=entsNrDrG)rHrIr rJrKrLrMrNrOr rPrQrRrSrTrUrVrWrXr=rrYrZr[r\ ent_templaterr) rar8r=rbrcrdrerrDs rfrgzEntityRenderer.__init__sR *+..668 %,,. &J #(\ j$/ !3!3Z8H!3!IJJ MM* % &  gkk(B/01@F O uu{{}e+O KK- 99 04 ::DI$  ;;z*  (D ~~&$/!$+!P;s 'F3<F9rhrirjc Bg}t|D]\}}|dk(rH|jdi}|jdt|_|jdt|_|j |j|d|d|jd|r`dj|Dcgc]}tj| c}} tj| |j |j } ndj|} |r t| S| Scc}w) rlrrmrZr\rrrprqrrrt) rvrTrYrZr[r\rw render_entsryrrQrr rzs rfrzEntityRenderer.rendersf% TDAqAv55R0!)k;!G$LL> OOD,,QvY& 155>R S  T 77hOsJ--c:OPD__T t~~VFWWX&F v& & Prrrorpcd}d}|D]g}|d}|d}|d} |jdd} |jdd} | rtj| | nd} |jd i} t||| }|||j d }t |D]9\}}|t|z }t |d kDs#|t |d z k7s5|d z };|j|j|jvrl|jj|j|j}|||| d}|j| ||jjdi|z }n||z }| }j||dj d }t |D]9\}}|t|z }t |d kDs#|t |d z k7s5|d z };tj||j}|rtj||z}|S)zRender entities in text. text (str): Original text. spans (list): Individual entity spans and their start, end, label, kb_id and kb_url. title (str / None): Document title set in Doc.user_data['title']. rqrrdrFrrrrrparams rz
N)rdrrrrrr)rTrrQr splitrvrrrXr=rVrSrrrZr )rarrorproffsetr7rdrFrrrradditional_paramsr fragmentsr|fragmentre ent_settingss rfr!zEntityRenderer.render_ents7s DMEMEu+CHHWb)EXXh,FHMk((uVDSUG $2 6  eC1FVE*006I(3 % 8+h//y>A%!s9~/A*Af$F %yy EKKMTYY$>  t7I7IJ""&   ##$562$++22B\BB& F5 6M''- $Y/ !KAx k(+ +F9~!a3y>A+=&=&  !T^^D %%E2V;F rr)rrrrrrrrrgrrrrr!rrrfrrs( E13,S#X,,BPU4S>*26HL 4.. $T#s(^ 4.=Ec]. .rr)(rtypingrrrrrrerrorsr utilr r r templatesrrrrrrrrrrrrrrrrrr r[rYrUrIr6rrrrrfr/sD ::55*   9 y 9 9   i   I  9 Y 9 9 I I Y  y!" #$y%,l$l$^RRjmmr