o i{`@sDddlZddlmZmZmZmZmZmZddlm 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*d*Z%Gd+d,d,Z&Gd-d.d.Z'dS)/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 @seZdZdZdZifdeeefddfddZ dd e eeefd e d e defd d Z de ede eeefde edefddZ ede ede eeefde eee eeefffddZde eeefdefddZde edefddZde edefddZdS) SpanRendererzRender Spans as SVGs.spanoptionsreturnNcCs tt}tj}|D]}t|r|}t|ts&tt j j t |d| |q | |dit|_dd|D|_t|_t|_|dd|_|dd|_|d d |_|d }|ru|d |_|d |_|d|_dS|jdkrt|_t|_t|_dSt |_t!|_t"|_dS)zfInitialise span renderer options (dict): Visualiser-specific options (colors, spans) objcolorscSi|] \}}||qSupper.0labelcolorr<r<I/home/ubuntu/.local/lib/python3.10/site-packages/spacy/displacy/render.py Kz)SpanRenderer.__init__.. top_offset(span_label_offsettop_offset_steptemplater5slicestartrtlN)#dictDEFAULT_LABEL_COLORSr displacy_colorsget_allvaluescallable isinstance ValueErrorr E925formattypeupdategetDEFAULT_ENTITY_COLOR default_coloritemsr: DEFAULT_DIR direction DEFAULT_LANGlangrFrH offset_step span_templatespan_slice_templatespan_start_templaterrrrrrselfr6r: user_colors user_colorrLr<r<rC__init__9s8          zSpanRenderer.__init__Fparsedpageminifyc Cg}t|D]-\}}|dkr"|di}|dt|_|dt|_|||d|d|dq|rKdd d |D}t j ||j|jd } nd|} |rVt | S| S) 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. rsettingsrarctokensspanstitlecSg|]}tj|dqS)contentrrYr@docr<r<rC yz'SpanRenderer.render..ryrcdir) enumerater\r`rarbrcappend render_spansjoinrrYr rirmrnrorenderediprrdocsmarkupr<r<rCrenderfs  $ zSpanRenderer.renderrsrtrucCs>|||}||}tj||jd}|rtj|d|}|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']. ryrru)_assemble_per_token_info_render_markuprrYrar)rirsrtruper_token_inforr<r<rCrs  zSpanRenderer.render_spanscCs0g}t|ddd}|D]}d|d<q t|D]~\}}i}||d<g}g}|D]d} i} | d|kr9| dkrnnL|| dk} | d | d <| | d <| r]t|rW|d dndd | d<|| | d| d<| d d} | dd} | r}tj| | dnd| d<|| q'd| d<q'||d<||q|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. cSs |d|d|d |dfS)N start_token end_tokenrAr<)sr<r<rCsz7SpanRenderer._assemble_per_token_info..keyr render_slottextrrrAis_startr kb_idrvkb_url#rrkb_linkentities)sortedrlenrr\rrY)rsrtrridxtoken token_markupintersecting_spansrr5ent span_startrrr<r<rCrsF            z%SpanRenderer._assemble_per_token_inforc Csd}|D]P}t|dddd}|ddk}|rJ|sJ||d}||d}|j|j|jt|d}||jj t |d|||d7}q|t |dd 7}q|S) z,Render the markup from per-token informationrvrcSs|dS)Nrr<)dr<r<rCrsz-SpanRenderer._render_markup..rrr )r span_slices span_starts total_height ) rstrip_get_span_slices_get_span_startsrFrHrdrrerYr ) rirrrr is_whitespaceslicesstartsrr<r<rCrs*  zSpanRenderer._render_markuprcCsbg}|D]'}|j|d|j}|j|j|dd}|jj||d}||qd |S)z*Get the rendered markup of all Span slicesrArr )bgrFrv) r:r\r>r^rFrdrfrYrr)rirrentityrBrF span_slicer<r<rCrs  zSpanRenderer._get_span_slicescCszg}|D]3}|j|d|j}|j|j|dd}|dr0|jj|||d|ddnd}||qd |S)z0Get the rendered markup of all Span start tokensrArr rr)rrFrArrv) r:r\r>r^rFrdrgrYrr)rirrrrBrFrr<r<rCr s"  zSpanRenderer._get_span_startsFF)__name__ __module__ __qualname____doc__stylerstrrrlrboolrrr staticmethodrrrrr<r<r<rCr44sD.   Fr4c @sBeZdZdZdZifdeeefddfddZ d*d e eeefd e d e defd d Z de e efde eeefde eeefdefddZdededede def ddZdede de dede def ddZde d e d!e d"e def d#d$Zded%e d e de def d&d'Zde eeefdeee e efe ffd(d)ZdS)+DependencyRendererz!Render dependency parses as SVGs.depr6r7NcCs|dd|_|dd|_|d|jrdnd|_|d|jr"d nd |_|d d |_|d |jr5dnd|_|dd|_|dd|_|dd|_ |dd|_ t |_ t |_dS)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 rI arrow_width arrow_strokerdistanceoffset_x2rBz#000000rz#fffffffontArialN)r\rrrrrrrrBrrr`rarbrc)rir6r<r<rCrl(s zDependencyRenderer.__init__Frmrnroc Cstj}g}t|D]C\}}|dkr'|di}|dt|_|dt|_|d|} | | |d|d} |drIt j |dd | } | | q |rfd d d |D} tj | |j|jd } nd |} |rqt| S| S)rqrrrrarc-wordsarcsrurrvcSrwrxrz)r@svgr<r<rCr}Ur~z-DependencyRenderer.render..r)uuiduuid4hexrr\r`rarbrc render_svgrrYrrrr ) rirmrnro id_prefixrrrrr render_idrryrr<r<rCr<s*      zDependencyRenderer.renderrrrc s|_tjdd_jdjj_jt |j_ jdj _ |_ fddt|D}fddt|D}d|d|}tjj j j jjj|jjd S) aRender SVG. render_id (Union[int, str]): Unique ID, typically index of document. words (list): Individual words and their tags. arcs (list): Individual arcs and their start, end, direction and label. RETURNS (str): Rendered SVG markup. rdefaultrc s0g|]\}}|d|d|dd|qS)rtaglemmaN) render_wordr\)r@rwrir<rCr}rsz1DependencyRenderer.render_svg..c s2g|]\}}|d|d|d|d|qS)rArNendr) render_arrow)r@rarr<rCr}vs rv) idwidthheightrBrrryrrc) get_levelslevelsmaxrT highest_levelrroffset_yrrrrrrrrrrYrBrrrarc)rirrr words_svgarcs_svgryr<rrCr_s0   zDependencyRenderer.render_svgrrrrcCsf|j|j}|j||j}|jdkr|j|}t|}|dur*tj|||||dSt j||||dS)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. rON)rrrxy)rrrr) rrrrrarr rrYr)rirrrrrr html_textr<r<rCrs    zDependencyRenderer.render_wordrArNrrac Cs^|dks|dkrt||||d}ttjjd i||j|||f}|j||j|j}|j dkr8|j |}|j } |j|||j||j|j|j |d} |j dkr^|j | } |j ||jd} |j ru|j ||jd} | dkrt|jdddkr|j } |||| | } ||| | | } |j dkrd nd }tj|j||j| ||| d 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)rNrrArrOrrrrightleft)rrstrokeheadrA label_sidearcNr<)rPrWr E157rYrrrrrarrrrrrT get_arrowheadget_arcrrr)rirArNrrar error_argslevelx_startrx_endy_curve arrowheadrrr<r<rCrsF      zDependencyRenderer.render_arrowrrrrcCs d}|jrd}|j||||dS)uSRender individual arc. x_start (int): X-coordinate of arrow start point. y (int): Y-coordinate of arrow start and end point. y_curve (int): Y-corrdinate of Cubic Bézier y_curve point. x_end (int): X-coordinate of arrow end point. RETURNS (str): Definition of the arc path ('d' attribute). z!M{x},{y} C{x},{c} {e},{c} {e},{y}z M{x},{y} {x},{c} {e},{c} {e},{y})rrce)rrY)rirrrrrLr<r<rCrs zDependencyRenderer.get_arcrc Cs|dkr|||jd||jd}}}n|||jd||jd}}}d|d|dd|d||jd|d||j S)aRRender individual arrow head. direction (str): Arrow direction, 'left' or 'right'. x (int): X-coordinate of arrow start point. y (int): Y-coordinate of arrow start and end point. end (int): X-coordinate of arrow end point. RETURNS (str): Definition of the arrow head path ('d' attribute). rrM,z Lr)r)rirarrrp1p2p3r<r<rCrs &$8z DependencyRenderer.get_arrowheadcCsdddd|DD}tdd|Ddd}dg|}i}t|dd d D]-}t||d |d d }t|d |d D]}|||<q?|||d |d |df<q&|S)aCalculate available arc height "levels". Used to calculate arrow heights dynamically and without wasting space. args (list): Individual arcs and their start, end, direction and label. RETURNS (dict): Arc levels keyed by (start, end, label). cSsg|]}t|qSr<)rP)r@tr<r<rCr}z1DependencyRenderer.get_levels..cSsh|] }tt|qSr<)tuplerr_r@rr<r<rC sz0DependencyRenderer.get_levels..cSsg|]}|dqS)rr<rr<r<rCr}r rrcSs|d|dS)NrrNr<)rr<r<rCrsz/DependencyRenderer.get_levels..rrNrr rA)rrrange)rirlength max_levelrrrrr<r<rCrs  zDependencyRenderer.get_levelsr)rrrrrrrrrlrrrrintrrrrrrrr<r<r<rCr#sP #  ( .4rc @seZdZdZdZifdeeefddfddZ dd e eeefd e d e defd d Z dede eeefde edefddZ dS)EntityRendererzRender named entities as HTML.rr6r7NcCstt}tj}|D]}t|r|}t|ts&tt j j t |d| |q | |dit|_dd|D|_|dd|_|jdurWdd|jD|_t|_t|_|d }|ri||_dS|jd krst|_dSt|_dS) zhInitialise entity renderer. options (dict): Visualiser-specific options (colors, ents) r8r:cSr;r<r=r?r<r<rCrDrEz+EntityRenderer.__init__..entsNcSsg|]}|qSr<r=)r@rr<r<rCr}r z+EntityRenderer.__init__..rLrO)rPrQr rRrSrTrUrVrWr rXrYrZr[r\r]r^r_r:rr`rarbrc ent_templaterrrhr<r<rCrls,          zEntityRenderer.__init__Frmrnroc Crp) rqrrrrarcrrrurvcSrwrxrzr{r<r<rCr}/r~z)EntityRenderer.render..r) rr\r`rarbrcr render_entsrrrYr rr<r<rCrs  $ zEntityRenderer.renderrrtrucCsd}d}|D]}|d}|d}|d} |dd} |dd} | r)tj| | d nd} |d i} t||| }|||d }t|D]\}}|t|7}t|d krb|t|d krb|d 7}qF|jduso||jvr|j ||j }|||| d}| | ||j jdi|7}n||7}| }q||dd }t|D]\}}|t|7}t|d kr|t|d kr|d 7}qt j||jd}|rtj|d|}|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']. rvrrArNrrrrrparams r z
N)rArrrrrr<)r\rrYr splitrrrr>r:r^r[rrrar)rirrtruroffsetr5rArNrrrradditional_paramsr fragmentsrfragmentrB ent_settingsr<r<rCr7sL       zEntityRenderer.render_entsr)rrrrrrrrrlrrrrrr<r<r<rCrs.! r)(rtypingrrrrrrerrorsr utilr r r templatesrrrrrrrrrrrrrrrrrrrbr`r]rQr4rrr<r<r<rCsd  P     pV