Update TaiwaneseHokkien.py

TaiwaneseHokkien.py

@@ -4,6 +4,8 @@ from transformers import pipeline
 model_repo_id = "emlinking/wav2vec2-large-xls-r-300m-tsm-asr-v6" 
 util = Converter() 
 def get_initial_length(s):
+    if s[0] in ['a', 'e', 'i', 'o', 'u']: return 0 
+    if (len(s) == 3 and s[:2] == 'ng') or (len(s) == 2 and s[:1] == 'm'): return 0  
     if len(s) >= 3 and s[2] == 'h':
         return 3
     elif len(s) >= 2 and s[1] in ['s', 'h']:
@@ -29,7 +31,10 @@ class TaiwaneseHokkien(Language):
                 til = get_initial_length(target_syl)
                 uil = get_initial_length(user_syl)
                 if target_syl[:til] != user_syl[:uil]:
-                    result.append((user_syl[:uil], 'initial error'))
+                    if uil == 0:
+                        result.append((' ', 'initial error'))
+                    else:
+                        result.append((user_syl[:uil], 'initial error'))
                 else:
                     result.append((user_syl[:uil], None))
                 if target_syl[til:-1] != user_syl[uil:-1]:
@@ -60,4 +65,4 @@ class TaiwaneseHokkien(Language):
         return result 
     @property 
     def compare_colors(self):
-        return {'tone error': 'red', 'initial error': 'blue', 'rime error': 'green', 'missing syllables': 'yellow', 'extra syllables': 'stone'}
+        return {'tone error': 'red', 'initial error': 'blue', 'rime error': 'green', 'missing syllables': 'yellow', 'extra syllables': 'gray'}