api / app.py
wannaphong's picture
Add spell and util
59fb62a
from fastapi import Depends, FastAPI, Header, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import RedirectResponse
from routers import tokenize, soundex, util, spell
import pythainlp
DESC_TEXT = """# PyThaiNLP API
PyThaiNLP API
"""
app = FastAPI(
title='PyThaiNLP API',
description=DESC_TEXT,
# summary="Deadpool's favorite app. Nuff said.",
version="0.0.1",
# terms_of_service="http://example.com/terms/",
# contact={
# "name": "Deadpoolio the Amazing",
# "url": "http://x-force.example.com/contact/",
# "email": "[email protected]",
# },
license_info={
"name": "Apache 2.0",
"identifier": "MIT",
},
)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/")
def index():
response = RedirectResponse(url='/docs')
return response
@app.get("/version")
def version():
"""
Get PyThaiNLP 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(spell.router, prefix="/spell", tags=["Spell"])
app.include_router(util.router, prefix="/util", tags=["Util"])