Add text_normalize api

This commit is contained in:
litagin02
2024-02-26 01:49:06 +09:00
parent 36462c8203
commit a6ae2bd15a

View File

@@ -42,7 +42,7 @@ from common.constants import (
)
from common.log import logger
from common.tts_model import ModelHolder
from text.japanese import g2kata_tone, kata_tone2phone_tone
from text.japanese import g2kata_tone, kata_tone2phone_tone, text_normalize
from text.user_dict import apply_word, update_dict, read_dict, rewrite_word, delete_word
@@ -205,12 +205,12 @@ class MoraTone(BaseModel):
tone: int
class G2PRequest(BaseModel):
class TextRequest(BaseModel):
text: str
@router.post("/g2p")
async def read_item(item: G2PRequest):
async def read_item(item: TextRequest):
try:
kata_tone_list = g2kata_tone(item.text, ignore_unknown=True)
except Exception as e:
@@ -221,6 +221,11 @@ async def read_item(item: G2PRequest):
return [MoraTone(mora=kata, tone=tone) for kata, tone in kata_tone_list]
@router.post("/normalize")
async def normalize_text(item: TextRequest):
return text_normalize(item.text)
@router.get("/models_info")
def models_info():
return model_holder.models_info()