3 d@sHddlZddlZddddddddd d d d d ddddZGdddZdS)Nzyou arezyou wereyouz you wouldzyou havezyou willyourzI amzI waszI havezI willmyZmineme)zi amzi wasizi'mzi'dzi'vezi'llrzyou arezyou werezyou'vezyou'llrZyoursrrc@sBeZdZifddZddZddZddZd d Zdd d ZdS)ChatcCs$dd|D|_||_|j|_dS)a Initialize the chatbot. Pairs is a list of patterns and responses. Each pattern is a regular expression matching the user's statement or question, e.g. r'I like (.*)'. For each such pattern a list of possible responses is given, e.g. ['Why do you like %1', 'Did you ever dislike %1']. Material which is matched by parenthesized sections of the patterns (e.g. .*) is mapped to the numbered positions in the responses, e.g. %1. :type pairs: list of tuple :param pairs: The patterns and responses :type reflections: dict :param reflections: A mapping between first and second person expressions :rtype: None cSs"g|]\}}tj|tj|fqS)recompile IGNORECASE).0xyrr./tmp/pip-build-v9q4h5k9/nltk/nltk/chat/util.py 3sz!Chat.__init__..N)_pairs _reflections_compile_reflections_regex)selfpairs reflectionsrrr__init__#sz Chat.__init__cCs2t|jtdd}tjdjdjttj|tj S)NT)keyreversez\b({})\b|) sortedrlenr r formatjoinmapescaper )rZ sorted_reflrrrr7szChat._compile_reflectionscsjjfdd|jS)z Substitute words in the string, according to the specified reflections, e.g. "I'm" -> "you are" :type str: str :param str: The string to be mapped :rtype: str csj|j|j|jS)N)rstringstartend)mo)rrrHsz"Chat._substitute..)rsublower)rstrr)rr _substitute=s zChat._substitutecCsj|jd}xZ|dkrdt||d|d}|d||j|j|||dd}|jd}q W|S)N%r)findintr*group)rresponsematchposnumrrr _wildcardsKs   ,zChat._wildcardscCsxz|jD]p\}}|j|}|rtj|}|j||}|dddkrT|ddd}|d ddkrt|dd d}|SqWdS) z Generate a response to the user input. :type str: str :param str: The string to be mapped :rtype: str r-Nz?..z???r8r8r8)rr2randomchoicer5)rr)patternr1r2resprrrrespondWs    z Chat.respondquitc Cstd}xj||krn|}y td}Wntk r:t|YnX|rx|ddkr\|dd}qBWt|j|qWdS)N>r,z!.rA)inputEOFErrorprintr=)rr>Z user_inputrrrconverseqs  z Chat.converseN)r>) __name__ __module__ __qualname__rrr*r5r=rErrrrr"s   r)r9r rrrrrr s$