add pypinyin and test

pyproject.toml

@@ -6,6 +6,8 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
     "gradio>=4.44.1",
+    "pypinyin>=0.55.0",
+    "pytest>=8.4.2",
 ]
 
 [dependency-groups]

tests/test_utils.py

@@ -0,0 +1,10 @@
+import pytest
+from utils.utils import get_pinyin
+
+@pytest.mark.parametrize("text, expected", [
+    ("举棋不定", "jǔ qí bù dìng"),
+    ("风", "fēng"),
+    ("不怕慢，就怕站", "bù pà màn ， jiù pà zhàn"),
+])
+def test_get_pinyin_accent(text, expected):
+    assert get_pinyin(text) == expected

utils/utils.py

@@ -0,0 +1,6 @@
+from pypinyin import pinyin, Style
+
+def get_pinyin(text: str):
+    """Convert Chinese characters to pinyin with tones."""
+    py_list = pinyin(text, style=Style.TONE, heteronym=False)
+    return " ".join([syllable[0] for syllable in py_list])