\jH dZddlmZddlmZddlmZmZddlm Z m Z m Z m Z dZ dZdZd ZGd d e Zd Zed k(reyy)a?Invokes the Java tsurgeon on a list of trees Included with CoreNLP is a mechanism for modifying trees based on existing patterns within a tree. The patterns are found using tregex: https://nlp.stanford.edu/nlp/javadoc/javanlp/edu/stanford/nlp/trees/tregex/TregexPattern.html The modifications are then performed using tsurgeon: https://nlp.stanford.edu/nlp/javadoc/javanlp/edu/stanford/nlp/trees/tregex/tsurgeon/Tsurgeon.html This module accepts Tree objects as produced by the conparser and returns the modified trees that result from one or more tsurgeon operations. ) tree_reader)Tree)TsurgeonRequestTsurgeonResponse) send_request build_tree from_treeJavaProtobufContextz=edu.stanford.nlp.trees.tregex.tsurgeon.ProcessTsurgeonRequestc,t|ttSN)rr TSURGEON_JAVA)requests `/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/stanza/server/tsurgeon.pysend_tsurgeon_requestrs !1= AAct|tr|f}t}|D]'}|jj t |d)t d|Dr|f}|D]d}t|dk(r td|jj}|d|_ |ddD]}|jj |f|S)z Build the TsurgeonRequest object trees: a list of trees operations: a list of (tregex, tsurgeon, tsurgeon, ...) gc3<K|]}t|tywr ) isinstancestr).0xs r z build_request..)s 2!:a  2sz6Expected [tregex, tsurgeon, ...] but just got a tregexrN) rrrtreesappendralllen ValueError operationsaddtregextsurgeon)rrrtree operationoperation_requestr"s r build_requestr&s%G4 Zc234 2z 22 ] 8 y>Q UV V#..224#,Q< !!"  8H  & & - -h 7 8 8 Nrct||}t|}|jDcgc]}t|dc}Scc}w)z Returns the result of processing the given tsurgeon operations on the given trees Returns a list of modified trees, eg, the result is already processed r)r&rrr )rrrresultts r process_treesr*5s: E:.G "7 +F%+\\ 2IaLO 22 2s>c*eZdZdZdfd ZdZxZS)Tsurgeonz Tsurgeon context window This is a context window which keeps a process open. Should allow for multiple requests without launching new java processes each time. c@tt| |ttyr )superr,__init__rr )self classpath __class__s rr/zTsurgeon.__init__Hs h&y2BMRrct||}|j|}|jDcgc]}t|dc}Scc}w)Nr)r&process_requestrr )r0rrrr(r)s rprocesszTsurgeon.processKs>z2%%g.)/6A ! Q666sAr )__name__ __module__ __qualname____doc__r/r5 __classcell__)r2s@rr,r,As S7rr,cld}tj|}d}d}t|||f}t|y)z7 A small demonstration of a tsurgeon operation z[( (SBARQ (WHNP (WP Who)) (SQ (VP (VBZ sits) (PP (IN in) (NP (DT this) (NN seat))))) (. ?)))zWP=wpzrelabel wp WWWPPPN)r read_treesr*print)textrr!r"r(s rmainr?Qs< gD  " "4 (E F"H 568"4 5F &Mr__main__N)r9stanza.models.constituencyr%stanza.models.constituency.parse_treerstanza.protobufrr$stanza.server.java_protobuf_requestsrrr r r rr&r*r,r?r6rrrFsV 36=iiO B2 3 7" 7  zFr