U -eS@s|dZddlmZmZmZmZddlmZmZm Z dgZ d eeefe e e dddZ dee ee ee e e d d d Zd S)z_ Utility method to visualize the alignment between one or more reference and hypothesis pairs. )DictListTupleUnion)CharacterOutput WordOutputAlignmentChunkvisualize_alignmentT)output show_measures skip_correctreturnc Cs|j}|j}|j}t|t}d}tt|||D]`\}\} } } |r^t| dkr^| djdkr^q0|d|dd7}|t | | | | d7}|d7}q0|rv|dt|d7}|d |j d 7}|d |j d 7}|d |j d 7}|d |j d7}|r|d|jddd7}n`|d|jddd7}|d|jddd7}|d|jddd7}|d|jddd7}n |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) referencesZ hypothesesZ alignments isinstancer enumerateziplentype_construct_comparison_stringZ substitutionsZ deletions insertionshitsZcerZmerZwilZwipZwer) r r r r hypothesisZ alignmentZis_cerZ final_stridxgthpchunksr%P/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/jiwer/alignment.pyr s<D   F) referencer opsrr csd}d}d}|D]}|jdks*|jdkr^||j|j}||j|j} |jdkrXdndn||jdkr||j|j}d d tt|D} d nF|jd kr||j|j} d d tt| D}dntd|jfdd tt|D} t|| | D]\} } } t t| t| t| }| dkr>d dg|} n| dkrXd dg|} || d|7}|| d|7}|| d|7}|r|d7}|d7}|d7}qq|r|ddd|ddd|dddS|d|d|dSdS)NzREF: zHYP: z r substitutersdeletecSsg|]}dqS*r%.0_r%r%r& sz0_construct_comparison_string..dinsertcSsg|]}dqSr,r%r.r%r%r&r1sizunparseable op name=csg|]}qSr%r%r.Zop_charr%r&r1sr-r>rr) rZ ref_start_idxZ ref_end_idxZ hyp_start_idxZ hyp_end_idxranger ValueErrorrmaxjoinupper)r'r r(rZref_strZhyp_strZop_stroprefZhypZop_charsrfr#cZstr_lenr%r5r&rsD     .rN)TT)F)__doc__typingrrrrZ jiwer.processrrr__all__boolstrr rr%r%r%r&s&  m