o rZh) @sddlmZddlmZ dgZzddlmZddlmZddZ Wne y:Z z e fddZ WYd Z [ nd Z [ wwd d Z d d Z GdddeZddZd S))ParserI)Tree BllipParser)RerankingParserget_unified_model_parameterscCsdSNr r r ?/var/www/auris/lib/python3.10/site-packages/nltk/parse/bllip.py_ensure_bllip_import_or_errorZsr cCs td|)Nz&Couldn't import bllipparser module: %s) ImportError)ier r r r _s Nc CsTzt|D] \}}|dqWdSty)}z td|d|d|d}~ww)NasciizToken z (zH) is non-ASCII. BLLIP Parser currently doesn't support non-ASCII inputs.) enumerateencodeUnicodeEncodeError ValueError)wordsiworder r r _ensure_asciics rcCstt|jSr)rZ fromstringstrZ ptb_parse) scored_parser r r _scored_parse_to_nltk_treensrc@sDeZdZdZ     d ddZddZddZe d d d ZdS) rz Interface for parsing with BLLIP Parser. BllipParser objects can be constructed with the ``BllipParser.from_unified_model_dir`` class method or manually using the ``BllipParser`` constructor. NcCs^t|pi}|p i}t|_|jj|fi||r+|r-|jjd||d|dSdSdS)a Load a BLLIP Parser model from scratch. You'll typically want to use the ``from_unified_model_dir()`` class method to construct this object. :param parser_model: Path to parser model directory :type parser_model: str :param reranker_features: Path the reranker model's features file :type reranker_features: str :param reranker_weights: Path the reranker model's weights file :type reranker_weights: str :param parser_options: optional dictionary of parser options, see ``bllipparser.RerankingParser.RerankingParser.load_parser_options()`` for more information. :type parser_options: dict(str) :param reranker_options: optional dictionary of reranker options, see ``bllipparser.RerankingParser.RerankingParser.load_reranker_model()`` for more information. :type reranker_options: dict(str) )Zfeatures_filenameZweights_filenameNr )r rrrpZload_parser_modelZload_reranker_model)selfZ parser_modelZreranker_featuresZreranker_weightsparser_optionsreranker_optionsr r r __init__ys! zBllipParser.__init__ccs.t||j|}|D]}t|Vq dS)a Use BLLIP Parser to parse a sentence. Takes a sentence as a list of words; it will be automatically tagged with this BLLIP Parser instance's tagger. :return: An iterator that generates parse trees for the sentence from most likely to least likely. :param sentence: The sentence to be parsed :type sentence: list(str) :rtype: iter(Tree) N)rrparser)rsentence nbest_listrr r r r s   zBllipParser.parsec cshg}i}t|D]\}\}}|||dur|||<q t||j||}|D]}t|Vq*dS)a Use BLLIP to parse a sentence. Takes a sentence as a list of (word, tag) tuples; the sentence must have already been tokenized and tagged. BLLIP will attempt to use the tags provided but may use others if it can't come up with a complete parse subject to those constraints. You may also specify a tag as ``None`` to leave a token's tag unconstrained. :return: An iterator that generates parse trees for the sentence from most likely to least likely. :param sentence: Input sentence to parse as (word, tag) pairs :type sentence: list(tuple(str, str)) :rtype: iter(Tree) N)rappendrrZ parse_taggedr) rZword_and_tag_pairsrZtag_maprrtagr"rr r r tagged_parses  zBllipParser.tagged_parsecCst|\}}}||||||S)a( Create a ``BllipParser`` object from a unified parsing model directory. Unified parsing model directories are a standardized way of storing BLLIP parser and reranker models together on disk. See ``bllipparser.RerankingParser.get_unified_model_parameters()`` for more information about unified model directories. :return: A ``BllipParser`` object using the parser and reranker models in the model directory. :param model_dir: Path to the unified model directory. :type model_dir: str :param parser_options: optional dictionary of parser options, see ``bllipparser.RerankingParser.RerankingParser.load_parser_options()`` for more information. :type parser_options: dict(str) :param reranker_options: optional dictionary of reranker options, see ``bllipparser.RerankingParser.RerankingParser.load_reranker_model()`` for more information. :type reranker_options: dict(str) :rtype: BllipParser r)cls model_dirrrZparser_model_dirZreranker_features_filenameZreranker_weights_filenamer r r from_unified_model_dirsz"BllipParser.from_unified_model_dir)NNNNN)NN) __name__ __module__ __qualname____doc__rr r% classmethodr(r r r r rrs /c Csddlm}|dj}tdt|}tdd}d}d}|||fD]%}td d |z t| |}t|Wq)t yNtd Yq)wt | |D] \}} td || fqVtd t| ddgtdt| ddgtdt| ddgdS)z8This assumes the Python module bllipparser is installed.r)findzmodels/bllip_wsj_no_auxzLoading BLLIP Parsing models...zDone.z#British left waffles on Falklands .z"I saw the man with the telescope .z # ! ? : -z Sentence: %r z(parse failed)z parse %d: %szforcing 'tree' to be 'NN':)AN)treeZNNz.forcing 'A' to be 'DT' and 'tree' to be 'NNP':)r0ZDT)r1NNPzforcing 'A' to be 'NNP':)r0r2)r1NN) Z nltk.datar.pathprintrr(splitjoinnextr StopIterationrr%) r.r'ZbllipZ sentence1Z sentence2Zfail1r!r1rr r r r demos<      r9)Znltk.parse.apirZ nltk.treer__all__Z bllipparserrZbllipparser.RerankingParserrr r r rrrr9r r r r s"  F