o !wiE @sdZddlmZmZddlmZmZmZdgZ  ddeeefde de d e fd dZ dd ee d ee deede d e f ddZ dS)z_ Utility method to visualize the alignment between one or more reference and hypothesis pairs. )ListUnion)CharacterOutput WordOutputAlignmentChunkvisualize_alignmentToutput show_measures skip_correctreturnc Cs|j}|j}|j}t|t}d}tt|||D]0\}\} } } |r/t| dkr/| djdkr/q|d|dd7}|t | | | | d7}|d7}q|r|dt|d7}|d |j d 7}|d |j d 7}|d |j d 7}|d |j d7}|r|d|jddd7}|S|d|jddd7}|d|jddd7}|d|jddd7}|d|jddd7}|S|dd}|S)a Visualize the output of [jiwer.process_words][process.process_words] and [jiwer.process_characters][process.process_characters]. The visualization shows the alignment between each processed reference and hypothesis pair. If `show_measures=True`, the output string will also contain all measures in the output. Args: output: The processed output of reference and hypothesis pair(s). show_measures: If enabled, the visualization will include measures like the WER or CER skip_correct: If enabled, the visualization will exclude correct reference and hypothesis pairs Returns: (str): The visualization as a string Example: This code snippet ```python import jiwer out = jiwer.process_words( ["short one here", "quite a bit of longer sentence"], ["shoe order one", "quite bit of an even longest sentence here"], ) print(jiwer.visualize_alignment(out)) ``` will produce this visualization: ```txt sentence 1 REF: # short one here HYP: shoe order one * I S D sentence 2 REF: quite a bit of # # longer sentence # HYP: quite * bit of an even longest sentence here D I I S I number of sentences: 2 substitutions=2 deletions=2 insertions=4 hits=5 mer=61.54% wil=74.75% wip=25.25% wer=88.89% ``` When `show_measures=False`, only the alignment will be printed: ```txt sentence 1 REF: # short one here HYP: shoe order one * I S D sentence 2 REF: quite a bit of # # longer sentence # HYP: quite * bit of an even longest sentence here D I I S I ``` requalz sentence  )include_space_seperatorznumber of sentences: zsubstitutions= z deletions=z insertions=zhits=z cer=dz.2fz% z mer=%z wil=z wip=z wer=N) references hypotheses alignments isinstancer enumerateziplentype_construct_comparison_string substitutions deletions insertionshitscermerwilwipwer) rr r r hypothesis alignmentis_cer final_stridxgthpchunksr/L/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/jiwer/alignment.pyrs:D     F referencer'opsrcsd}d}d}|D]}|jdks|jdkr.||j|j}||j|j} |jdkr+dndn>|jdkrI||j|j}d d tt|D} d n#|jd krd||j|j} d d tt| D}dntd|jfdd tt|D} t|| | D]U\} } } t t| t| t| }| dkrd dg|} n | dkrd dg|} || d|7}|| d|7}|| d|7}|r|d7}|d7}|d7}qq|r|ddd|ddd|dddS|d|d|dS)NzREF: zHYP: z r substitutersdeletecSg|]}dqS*r/.0_r/r/r0 z0_construct_comparison_string..dinsertcSr6r7r/r9r/r/r0r<r=izunparseable op name=csg|]}qSr/r/r9op_charr/r0r<r=r8r >rr) r ref_start_idx ref_end_idx hyp_start_idx hyp_end_idxranger ValueErrorrmaxjoinupper)r1r'r2rref_strhyp_strop_stroprefhypop_charsrfr-cstr_lenr/rAr0rsH  .rN)TT)F) __doc__typingrr jiwer.processrrr__all__boolstrrrr/r/r/r0s6  m