From a6ae2bd15a9f7e69cbf375d44602fe0b6855d651 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Mon, 26 Feb 2024 01:49:06 +0900 Subject: [PATCH] Add text_normalize api --- server_editor.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/server_editor.py b/server_editor.py index f859315..7f5a6ef 100644 --- a/server_editor.py +++ b/server_editor.py @@ -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()