Function rich_transcription_postprocess

funasr/utils/postprocess_utils.py:436–477

Rich transcription postprocess. Args: s: TODO.

(s)

def rich_transcription_postprocess(s):
    """Rich transcription postprocess.
    
        Args:
            s: TODO.
        """
    def get_emo(s):
        """Get emo.
        
            Args:
                s: TODO.
            """
        return s[-1] if s[-1] in emo_set else None

    def get_event(s):
        """Get event.
        
            Args:
                s: TODO.
            """
        return s[0] if s[0] in event_set else None

    s = s.replace("<|nospeech|><|Event_UNK|>", "❓")
    for lang in lang_dict:
        s = s.replace(lang, "<|lang|>")
    s_list = [format_str_v2(s_i).strip(" ") for s_i in s.split("<|lang|>")]
    new_s = " " + s_list[0]
    cur_ent_event = get_event(new_s)
    for i in range(1, len(s_list)):
        if len(s_list[i]) == 0:
            continue
        if get_event(s_list[i]) == cur_ent_event and get_event(s_list[i]) != None:
            s_list[i] = s_list[i][1:]
        if len(s_list[i]) == 0:
            continue
        # else:
        cur_ent_event = get_event(s_list[i])
        if get_emo(s_list[i]) != None and get_emo(s_list[i]) == get_emo(new_s):
            new_s = new_s[:-1]
        new_s += s_list[i].strip().lstrip()
    new_s = new_s.replace("The.", " ")
    return new_s.strip()