Add spell and util

app.py

@@ -1,7 +1,7 @@
 from fastapi import Depends, FastAPI, Header, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import RedirectResponse
-from routers import tokenize, soundex
+from routers import tokenize, soundex, util, spell
 import pythainlp
 
 
@@ -50,4 +50,6 @@ def version():
     return {"version": pythainlp.__version__}
 
 app.include_router(tokenize.router, prefix="/tokenize", tags=["Tokenize"])
-app.include_router(soundex.router, prefix="/soundex", tags=["Soundex"])
+app.include_router(soundex.router, prefix="/soundex", tags=["Soundex"])
+app.include_router(spell.router, prefix="/spell", tags=["Spell"])
+app.include_router(util.router, prefix="/util", tags=["Util"])

requirements.txt

@@ -1,6 +1,6 @@
-fastapi
-uvicorn[standard]
+fastapi==0.115.6
+uvicorn[standard]==0.34.0
 pythainlp==5.0.5
-python-crfsuite
-ssg
-tltk
+python-crfsuite==0.9.11
+ssg==0.0.8
+tltk==1.9.1

routers/soundex.py

@@ -18,6 +18,6 @@ class SoundexEngine(str, Enum):
 @router.post('/soundex')
 def soundex(text: str, engine: SoundexEngine = "udom83"):
     """
-    Word tokenize
+    This api converts Thai text into phonetic code.
     """
     return {"soundex": py_soundex(text=text, engine=engine)}

routers/spell.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+from fastapi import APIRouter
+from pythainlp.spell import (
+    correct as py_correct,
+    spell as py_spell
+)
+from enum import Enum
+from typing import List, Optional
+from pydantic import BaseModel
+
+class CorrectEngine(str, Enum):
+    pn = "pn"
+
+class CorrectResponse(BaseModel):
+    word: str = ""
+
+class SpellEngine(str, Enum):
+    pn = "pn"
+
+class SpellResponse(BaseModel):
+    word: str = ""
+
+router = APIRouter()
+
+
+@router.post('/correct', response_model=CorrectResponse)
+def correct(word: float, engine: CorrectEngine = "pn"):
+    return {"word": py_correct(word, engine=engine)}
+
+@router.post('/spell', response_model=SpellResponse)
+def spell(word: float, engine: SpellEngine = "pn"):
+    return {"word": py_spell(word, engine=engine)}

routers/util.py

@@ -0,0 +1,20 @@
+# -*- coding: utf-8 -*-
+from fastapi import APIRouter
+from pythainlp.util import (
+    bahttext as py_bahttext,
+    normalize as py_normalize,
+    tone_detector as py_tone_detector
+)
+router = APIRouter()
+
+@router.post('/bahttext')
+def bahttext(number: float):
+    return {"bahttext": py_bahttext(number)}
+
+@router.post('/normalize')
+def normalize(text: str):
+    return {"text": py_normalize(text)}
+
+@router.post('/tone_detector')
+def tone_detector(syllable: str):
+    return {"tone": py_tone_detector(syllable)}