o hLd@sddlZddlZddlZejejejejeZejeddlm Z ddl m Z ddl m Z ddlZddlmZddlmZddlmZmZddlZddlmZmZmZdd lmZmZd/d ed ed edeefddZdedefddZdeeddfddZ   d0d edeedede!de!ded edeefddZ"    d1d edeedede!de!ded efd!d"Z#d#edefd$d%Z$d&edefd'd(Z%d)edefd*d+Z&d,d-Z'e(d.kre'dSdS)2N)Pipeline)Semgrex)is_right_to_left)displacy)DocdisplayHTML)ListTupleAny)find_nth round_baseFdoclanguagevisualize_xposreturnc Cs|sdnd}gg}}td}|jD]}gggggf\}} } } } t|rmt|j} t|jD]<}||j| |j | |j |rO|j rO| |j n| |j |j dkrc| | |jq/| | |j q/n@|jD]<}||j| |j | |j |r|j r| |j n| |j |j dkr| |jdqp| |j dqp|rt|j|| | | | d}n t|j|| | | | d}||q|D]}|tj|ddd d d |d dd q|S)a Returns a list of HTML strings representing the dependency visualizations of a given stanza document. One HTML string is generated per sentence of the document object. Converts the stanza document object to a spaCy doc object and generates HTML with displaCy. @param doc: a stanza document object which can be generated with an NLP pipeline. @param language: the two letter language code for the document e.g. "en" for English. @param visualize_xpos: A toggled option to use xpos tags for part-of-speech labels instead of upos. @return: a list of HTML strings which visualize the dependencies of the doc object. FTenr)wordslemmasheadsdepstags)rrrrposdepd)compact word_spacingdistance arrow_spacing fine_grained)styleoptionsjupyter)spacyblank sentencesrlenrreversedappendtextlemmadeprelxposuposheadidrvocabrrender)rrrUSE_FINE_GRAINED html_stringssentences_to_visualizenlpsentencerrrrr sentence_lenwordstanza_to_spacy_docliner?h/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/utils/visualization/semgrex_visualizer.pyget_sentences_htmlsn                rA orig_htmlc% Csi}d}d}gd}d}|d}|d|d|||dd}|jD]`}t|jD]=\} } || } d} | jD].} | j| j}}t|d|}||vr|d |}|d |}||||dd}|d d d d| d}|d|||||dd}|||<t||dd ||d|}t||dd ||d|}||||d}t||dd |||}|d||d}|||dd}|d}||dddd}||d||}d\}}| |kr| | jdkr| }| | jdkrHt | jdkrHd} | jdj|vrH|| jdj} ||| }!|!dkr>||!d} |}n |t |!d}|} d|d| d|d|dd d }"||"|}||d7<q .bolded{font-weight: bold;}  Nrzzclass="displacy-word"zclass="bolded"zfill="currentColor"zfill=""zx=")rCrCrz# z . zzrrIN<)r edit_dep_arrow) rHTML_ARROW_BEGINNINGHTML_ARROW_ENDINGHTML_ARROW_ENDING_LENarrows_start_idx words_html arrows_html final_html arrow_number start_idxend_of_class_idx arrow_sectionedited_arrow_sectionr?r?r@rZs:          r arrow_htmlcCsd}d}d}t|ddt|dd}}|d||d|}}t||||t||||}} t|dddt|dd d} } ||| ||| } } | d| d}}|| krt||d }}|dd d d t|}dt|d|dd|dd } |dd d dt||}|dd d d t||}dt|d|d|dd|d d } npt| |d }|dd d d t|}|d d d d t|}|d |d||g|dd }d|} d|}|dd d dt||}|dd d d t||}|||g|d d }d|} |d || d || || d || d S)aZ The formatting of a single displacy arrow in svg is the following: csubj We edit the 'd = ...' parts of the section to fix the arrow direction and length to round to the nearest 50 units, centering on each word's center. This is because the words start at x=50 and have spacing of 100, so each word is at an x-value that is a multiple of 50. @param arrow_html: Original SVG for a single displaCy arrow. @return: Edited SVG for the displaCy arrow, adjusting its placement 2zd="MrrC,fillrLrG)base rNz L)r rNfloatsplitrstrjoin)r WORD_SPACINGM_OFFSETARROW_PIXEL_SIZE first_d_idx second_d_idxfirst_d_cutoffsecond_d_cutoffarrow_positionarrowhead_positionfirst_fill_start_idxsecond_fill_start_idxfirst_dsecond_d first_d_splitsecond_d_splitcorrected_arrow_poscorrected_arrowhead_pos second_term third_term fourth_termterms first_termr?r?r@rs                      rr{cCs:d}t|ddd}d}|d|d||||dS)a Adds to overflow and display settings to the SVG header to visualize overflowing HTML renderings in the Semgrex streamlit app. Prevents Semgrex search tags from being cut off at the bottom of visualizations. The opening of each HTML string looks similar to this; we add to the end of the SVG header. Banning VERB Act. @param html_string: HTML of the result of running Semgrex search on a text @return: Edited HTML to visualize the dependencies even in the case of overflow. ���zdirection: ltrr���r���z!overflow: visible; display: blockNz; )r ���)r{��� BUFFER_LENediting_start_idxSVG_HEADER_ADDITIONr?���r?���r@���edit_html_overflow ��s��� r���c������������������C���s���t�ddd}�|�d}ddg}t�}|jdt|dd �|jd tt�|d d �|jd tdd d �|jdtddd �|�}|jt j d<�zt ||d}t |d��W�dS��t y\���t dw�)aK�� IMPORTANT: For the code in this module to run, you must have corenlp and Java installed on your machine. Additionally, set an environment variable CLASSPATH equal to the path of your corenlp directory. Example: CLASSPATH=C:\Users\Alex\PycharmProjects\pythonProject\stanford-corenlp-4.5.0\stanford-corenlp-4.5.0\* r���ztokenize,pos,lemma,depparser���zeBanning opal removed artifact decks from the meta. Banning tennis resulted in players banning people.z{pos:NN}=object ���s����$       L�   ? *0�5