Add text_normalize api
This commit is contained in:
@@ -42,7 +42,7 @@ from common.constants import (
|
|||||||
)
|
)
|
||||||
from common.log import logger
|
from common.log import logger
|
||||||
from common.tts_model import ModelHolder
|
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
|
from text.user_dict import apply_word, update_dict, read_dict, rewrite_word, delete_word
|
||||||
|
|
||||||
|
|
||||||
@@ -205,12 +205,12 @@ class MoraTone(BaseModel):
|
|||||||
tone: int
|
tone: int
|
||||||
|
|
||||||
|
|
||||||
class G2PRequest(BaseModel):
|
class TextRequest(BaseModel):
|
||||||
text: str
|
text: str
|
||||||
|
|
||||||
|
|
||||||
@router.post("/g2p")
|
@router.post("/g2p")
|
||||||
async def read_item(item: G2PRequest):
|
async def read_item(item: TextRequest):
|
||||||
try:
|
try:
|
||||||
kata_tone_list = g2kata_tone(item.text, ignore_unknown=True)
|
kata_tone_list = g2kata_tone(item.text, ignore_unknown=True)
|
||||||
except Exception as e:
|
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]
|
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")
|
@router.get("/models_info")
|
||||||
def models_info():
|
def models_info():
|
||||||
return model_holder.models_info()
|
return model_holder.models_info()
|
||||||
|
|||||||
Reference in New Issue
Block a user