o h@svdZddlmZddlZddlZddlmZddlmZddZdd Z d d Z d d Z ddZ e dkr9e dSdS)zL Functions to visualize dependency relations in texts and Stanza documents )is_right_to_leftN)displacy)Docc Csjddddddd}td}g}|jD]}gggggf\}}}} } t|rat|j} t|jD]0} || j|| j | | j | | j | j d krW|| | j q/|| | j q/n4|jD]0} || j|| j | | j | | j | j d kr|| j d qd|| j d qdt|j|||| | d } || q|D] }tj|d |d qdS)z Takes in a Document and visualizes it using displacy. The document to visualize must be from the stanza pipeline. right-to-left languages such as Arabic are displayed right-to-left based on the language code Tz#09a3d5whiteZzSource Sans Pro)compactbgcolordistancefont arrow_spacingenr)wordslemmasheadsdepsposdep)styleoptionsN)spacyblank sentencesrlenrreversedappendtextlemmadepreluposheadidrvocabrrender)doclanguagevisualization_optionsnlpsentences_to_visualizesentencerrrrtagssent_lenworddocument_resultliner1n/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/utils/visualization/dependency_visualization.py visualize_doc s>              r3cCs||}t||dS)a Takes a string and visualizes it using displacy. The string is processed using the stanza pipeline and its dependencies are formatted into a spaCy doc object for easy visualization. Accepts valid stanza (UD) pipelines as the pipeline argument. Must supply the stanza pipeline code (the two-letter abbreviation of the language, such as 'en' for English. Must also supply the stanza pipeline object as the third argument. Nr3)r pipeline_codepiper&r1r1r2 visualize_str9s r7cCs|D]}t||qdS)a Takes in a list of Stanza document objects and a language code (ex: 'en' for English) and visualizes the dependency relationships within each document. This function uses spaCy visualizations. See the visualize_doc function for more details. Nr4)docs lang_coder&r1r1r2visualize_docsHs r:cCs(tj|dd}|D]}t|||q dS)a Takes a language code (ex: 'en' for English) and a list of strings to process and visualizes the dependency relationships in each text. This function loads the Stanza pipeline for the given language and uses it to visualize all of the strings provided. ztokenize,pos,lemma,depparse) processorsN)stanzaPipeliner7)textsr9r6rr1r1r2visualize_stringsSsr?cCs8gd}ddg}dg}t|dt|dt|ddS)N)u~برلين ترفض حصول شركة اميركية على رخصة تصنيع دبابة "ليوبارد" الالمانيةu$هل بإمكاني مساعدتك؟uأراك في مابعدuلحظة من فضلكzThis is a sentence.zWBarack Obama was born in Hawaii. He was elected President of the United States in 2008.u'中国是一个很有意思的国家。arrzh)r?) ar_strings en_strings zh_stringsr1r1r2main_s  rE__main__)__doc__stanza.models.common.constantrr<rr spacy.tokensrr3r7r:r?rE__name__r1r1r1r2s   -