Update sv.py

sv.py

@@ -91,32 +91,31 @@ emo_set = {"😊", "😔", "😡", "😰", "🤢", "😮"}
 event_set = {"🎼", "👏", "😀", "😭", "🤧", "😷"}
 
 
-def format_text_with_emojis(s):
-    sptk_dict = {sptk: s.count(sptk) for sptk in emoji_dict}
-
-    for sptk in emoji_dict:
-        s = s.replace(sptk, "")
+def clean_and_emoji_annotate_speech(text):
+    def get_emoji(s, emoji_set):
+        return next((char for char in s if char in emoji_set), None)
 
-    emo = "<|NEUTRAL|>"
-    for e in emo_dict:
-        if sptk_dict.get(e, 0) > sptk_dict.get(emo, 0):
-            emo = e
+    def format_text_with_emojis(s):
+        sptk_dict = {sptk: s.count(sptk) for sptk in emoji_dict}
 
-    s = (
-        "".join(event_dict[e] for e in event_dict if sptk_dict.get(e, 0) > 0)
-        + s
-        + emo_dict[emo]
-    )
+        for sptk in emoji_dict:
+            s = s.replace(sptk, "")
 
-    for emoji in emo_set.union(event_set):
-        s = s.replace(f" {emoji}", emoji).replace(f"{emoji} ", emoji)
+        emo = "<|NEUTRAL|>"
+        for e in emo_dict:
+            if sptk_dict.get(e, 0) > sptk_dict.get(emo, 0):
+                emo = e
 
-    return s.strip()
+        s = (
+            "".join(event_dict[e] for e in event_dict if sptk_dict.get(e, 0) > 0)
+            + s
+            + emo_dict[emo]
+        )
 
+        for emoji in emo_set.union(event_set):
+            s = s.replace(f" {emoji}", emoji).replace(f"{emoji} ", emoji)
 
-def clean_and_emoji_annotate_speech(text):
-    def get_emoji(s, emoji_set):
-        return next((char for char in s if char in emoji_set), None)
+        return s.strip()
 
     # Replace special tags
     text = text.replace("<|nospeech|><|Event_UNK|>", "❓")
@@ -136,14 +135,11 @@ def clean_and_emoji_annotate_speech(text):
             continue
 
         current_event = get_emoji(segment, event_set)
-        current_emotion = get_emoji(
-            segment, emo_set
-        )  # Check for emotion emoji anywhere in the segment
+        current_emotion = get_emoji(segment, emo_set)
 
         if current_event is not None:
             segment = segment[1:] if segment.startswith(current_event) else segment
 
-        # Preserve emotion emoji if it's different from the previous one
         if current_emotion is not None and current_emotion != prev_emotion:
             segment = segment.replace(current_emotion, "") + current_emotion