o ¡¿¯i ã@s dZddlZdd„Zdd„ZdS)z0Utility functions for the text translation task.éNcstttt|ddd…ƒƒ}‡fdd„|Dƒ}t|dƒ}d dd„|Dƒ¡}d |¡}d  |¡ d d¡}||||fS) zäParse hypothesis. :param list hyp: recognition hypothesis :param list char_list: list of characters :return: recognition text string :return: recognition token string :return: recognition tokenid string ÚyseqéNcsg|]}ˆ|‘qS©r©Ú.0Úidx©Ú char_listrúF/home/ubuntu/.local/lib/python3.10/site-packages/espnet/mt/mt_utils.pyÚ óz$parse_hypothesis..Úscoreú cSsg|]}t|ƒ‘qSr)Ústrrrrr r r Úz)ÚlistÚmapÚintÚfloatÚjoinÚreplace)Úhypr Útokenid_as_listÚ token_as_listr ÚtokenidÚtokenÚtextrrr Úparse_hypothesis s    rc CsFtƒ}d| ¡vr|d|d<g|d<t|dƒD]ˆ\}}t||ƒ\}}}} t|dƒdkr8t|dd ¡ƒ} nddi} | dd|7<|| d<|| d <|| d <| | d <|ddd | d <|ddd| d<|ddd| d<|d | ¡|dkr d |  ¡vrŽt d| d ¡t d| d¡t d| d¡q|S)zÞAdd N-best results to json. :param dict js: groundtruth utterance dict :param list nbest_hyps: list of hypothesis :param list char_list: list of characters :return: N-best results added utterance dict Úutt2spkÚoutputrrÚnamerz[%d]Úrec_textÚ rec_tokenÚ rec_tokenidr rÚtext_srcrÚ token_srcrÚ tokenid_srczgroundtruth: %szprediction : %sz source : %s) ÚdictÚkeysÚ enumeraterÚlenÚitemsÚappendÚloggingÚinfo) ÚjsÚ nbest_hypsr Únew_jsÚnrr!r"r#r Úout_dicrrr Úadd_results_to_json#s2   €r4)Ú__doc__r-rr4rrrr Ús