o i%@svdZddlZddlmZddlmZmZiddeDiddd dd d d d d d dddddddddddddddddddddddd d d d!d"Zd#e d$e fd%d&Z d#e d$ee e ffd'd(Z e d)krgd*Zed+ed,ed+eD]2Zed-ee eZed.ee eZed/red0ed1d2qed3ed4qdSdS)5u Emotion Tag Normalizer for Spark TTS Converts old Orpheus emotion tags to Spark TTS bracket format. From: → To: [emotion] Also provides backward compatibility for legacy formats. N)Dict)LEGACY_EMOTION_MAPINDIC_EMOTION_TAGScCsi|] \}}|d|qS)z<>)strip).0kvr D/home/ubuntu/veenaModal/veena3modal/processing/emotion_normalizer.py sr laughingz[laughs]laughz laughs harderz[laughs harder]z laugh harder laugh_hardersighsz[sighs]sighgigglesz[giggle]giggleangryz[angry]excitedz [excited]whispersz [whispers]whisper screamingz [screams]screamscreamssingingz[sings]z [curious])singsingscurioustextreturncCs8d}dd}t|||}d}dd}t|||}|S)u Normalize emotion tags to Spark TTS bracket format. Conversions: - → [laughs] - → [sighs] - → [giggle] - etc. Also handles natural language: - [laughing] → [laughs] - [singing] → [sings] Args: text: Text with potential emotion tags (old or new format) Returns: Text with normalized [emotion] tags for Spark TTS z <([a-z_]+)>cSs`|d}|tvrt|Sd|d}|tvrt|Std|d|dd|dS)N<>u⚠️ Unknown old emotion: - converting to [][)grouplowerrEMOTION_TO_TAGrprint)match emotion_textold_tagr r r replace_angle_emotionEs  z5normalize_emotion_tags..replace_angle_emotion \[([^\]]+)\]cSs|d}d|d}|tvr|S|tvrt|S|dr0|dd}|tvr0t|S|d}|tvr.replace_bracket_emotion)resub)r angle_patternr,bracket_patternr4normalized_textr r r normalize_emotion_tags.sr:csztd|}td|}ttfdd|D}fdd|D}t|dkdd|Ddd|D||t|dkd S) z Validate that text has proper emotion tag format for Spark TTS. Returns: Dict with validation results r-z <([^>]+)>cs(g|]}d|dvrd|dqSr$r#r rtagvalid_emotion_tagsr r (z,validate_normalized_text..cs(g|]}d|dvrd|dqSr;r r<r>r r r@rArcSg|]}d|dqS)r!r"r r<r r r r@cSrBr;r r<r r r r@rC)has_angle_bracketsangle_bracketsbracket_emotionsr?invalid_emotion_tagsis_valid)r5findallsetrlen)rsquare_bracketsrE valid_tags invalid_tagsr r>r validate_normalized_text{s      rO__main__)zDAnd of course, the so-called 'easy' hack didn't work at all [sighs].zThat's so silly [giggles].z:[angry] I cannot believe this happened for the third time.zhispers] Look over there.z-I'm so happy I could just [singing] la-la-la!z)And then he did it again [laughs harder]!z)Old format text needs conversion.z%Old format and mixed.zP================================================================================z+Emotion Tag Normalization Tests (Spark TTS)z Original: z Normalized: rHu✅ Valid (emotion tags: r?)u⚠️ Has angle brackets: rE)__doc__r5typingrveena3modal.core.constantsrritemsr'strr:anyrO__name__ test_casesr(r normalized validationr r r r s|      M