Merge pull request #2 from ppirch/dev

Change Flask to FastAPI

Author: wannaphong

.gitignore

@@ -99,3 +99,4 @@ ENV/
 
 # mypy
 .mypy_cache/
+.vscode/settings.json

README.md

@@ -1,2 +1,7 @@
 # PyThaiNLP API
-Web API for PyThaiNLP 
+
+Web API for PyThaiNLP
+
+## How to use it
+
+Enter a command, `uvicorn main:app` to run it

app.py

@@ -1,24 +1,14 @@
 # -*- coding: utf-8 -*-
-import flask
-from pythainlp.tokenize import word_tokenize
-app = flask.Flask(__name__)
+from fastapi import Depends, FastAPI, Header, HTTPException
+from routers import tag, tokenize
 
-@app.route("/")
+app = FastAPI()
+
+
+@app.get("/")
 def hello():
-    return "Hello World!"
+    return {"Hello": "World"}
 
-@app.route("/api/word_tokenize", methods=['GET','POST'])
-@app.route("/api/v1/word_tokenize", methods=['GET','POST'])
-def word_tokenize_api():
-	if flask.request.method == 'POST':
-		source = flask.request.values.get('source')
-		engine = flask.request.values.get('engine')
-	else:
-		source = flask.request.args.get('source')
-		engine = flask.request.args.get('engine')
-	if engine==None:
-		engine='newmm'
-	return '|'.join(word_tokenize(source,engine=engine))
 
-if __name__ == '__main__':
-    app.run()
+app.include_router(tag.router)
+app.include_router(tokenize.router)

requirements.txt

@@ -1,2 +1,4 @@
-flask
-pythainlp
+fastapi
+pythainlp
+uvicorn
+pytest

routers/__init__.py

@@ -0,0 +1,13 @@
+from fastapi import APIRouter
+from pythainlp import tag
+
+router = APIRouter()
+
+@router.get('/tag/pos_tag', tags=["tag"])
+def part_of_speech_tagging(q: str, engine: str = None, corpus: str = None):
+    words = pythainlp.tokenize.word_tokenize(q)
+    if engine == None:
+        engine = 'perceptron'
+    if corpus == None:
+        corpus='orchid'
+    return tag.pos_tag(words, engine, corpus)

routers/tag.py

@@ -0,0 +1,10 @@
+from fastapi import APIRouter
+from pythainlp import tokenize
+
+router = APIRouter()
+
+@router.get('/tokenize/word_tokenize', tags=["tokenize"])
+def word_tokenize(q: str, engine: str = None):
+    if engine == None:
+        engine='newmm'
+    return '|'.join(tokenize.word_tokenize(q,engine=engine))