Update code

routers/soundex.py

@@ -6,6 +6,10 @@ from pythainlp.soundex import (
     soundex as py_soundex
 )
 from enum import Enum
+from pydantic import BaseModel
+
+class SoundexResponse(BaseModel):
+    soundex: str = ""
 
 router = APIRouter()
 
@@ -17,7 +21,7 @@ class SoundexEngine(str, Enum):
     prayut_and_somchaip = "prayut_and_somchaip"
 
 
-@router.post('/soundex')
+@router.post('/soundex', response_model=SoundexResponse)
 def soundex(word: str, engine: SoundexEngine = "udom83"):
     """
     This api converts Thai text into phonetic code.

routers/util.py

@@ -8,9 +8,26 @@ from pythainlp.util import (
     tone_detector as py_tone_detector,
     thaiword_to_num as py_thaiword_to_num
 )
+from pydantic import BaseModel
 router = APIRouter()
 
-@router.post('/bahttext')
+
+class BahttextResponse(BaseModel):
+    bahttext: str
+
+
+class NormalizeResponse(BaseModel):
+    text: str
+
+
+class ToneResponse(BaseModel):
+    tone: str
+
+class NumeralsResponse(BaseModel):
+    number: int
+
+
+@router.post('/bahttext', response_model=BahttextResponse)
 def bahttext(number: float):
     """
     This api converts a number to Thai text and adds a suffix “บาท” (Baht). 
@@ -21,7 +38,7 @@ def bahttext(number: float):
     )
 
 
-@router.post('/normalize')
+@router.post('/normalize', response_model=NormalizeResponse)
 def normalize(text: str):
     """
     Normalize and clean Thai text
@@ -32,7 +49,7 @@ def normalize(text: str):
     )
 
 
-@router.post('/tone_detector')
+@router.post('/tone_detector', response_model=ToneResponse)
 def tone_detector(syllable: str):
     """
     Thai tone detector for word.
@@ -43,7 +60,7 @@ def tone_detector(syllable: str):
     )
 
 
-@router.post("/thaiword_to_num")
+@router.post("/thaiword_to_num", response_model=NumeralsResponse)
 def thaiword_to_num(text: str):
     """
     Converts the spelled-out numerals in Thai scripts into an actual integer.