diff --git a/be/app/api/v1/__init__.py b/be/app/api/v1/__init__.py index c07b19f..7f550dd 100644 --- a/be/app/api/v1/__init__.py +++ b/be/app/api/v1/__init__.py @@ -1,5 +1,5 @@ from flask import Blueprint -from ...api.v1 import index, project, files, anno +from ...api.v1 import index, project, files, anno, ai def create_blueprint_v1(): @@ -9,5 +9,6 @@ def create_blueprint_v1(): project.api.register(bp_v1, url_prefix='/project') anno.api.register(bp_v1, url_prefix='/anno') files.api.register(bp_v1, url_prefix='/files') + ai.api.register(bp_v1, url_prefix='/ai') return bp_v1 diff --git a/be/app/api/v1/ai.py b/be/app/api/v1/ai.py new file mode 100644 index 0000000..7bfec33 --- /dev/null +++ b/be/app/api/v1/ai.py @@ -0,0 +1,55 @@ +from app.libs.redprint import RedPrint +import time +from flask import request +import json +import os + +from ...libs.tools import read_file, write_json, read_json_file, make_dir + +import auto_anno_2 as aa + +api = RedPrint('ai') + +class ReturnInfo: + def __init__(self): + self.errCode = 0 + self.errMsg = '' + self.info = [] + +@api.route('/nlp/cls', methods=['POST']) +def nlp_cls(): + ret_info = ReturnInfo() + try: + j = request.get_json() + texts = j['texts'] + types = j['types'] + annos = [] + for text in texts: + anno = aa.cls(text, types) + annos.append(anno) + ret_info.info = annos + except Exception as e: + print(e) + ret_info.errCode = 404 + ret_info.errMsg = str(e) + + return ret_info.__dict__ + +@api.route('/nlp/ner', methods=['POST']) +def nlp_ner(): + ret_info = ReturnInfo() + try: + j = request.get_json() + texts = j['texts'] + types = j['types'] + annos = [] + for text in texts: + anno = aa.ner(text, types) + annos.append(anno) + ret_info.info = annos + except Exception as e: + print(e) + ret_info.errCode = 404 + ret_info.errMsg = str(e) + + return ret_info.__dict__ diff --git a/be/requirements.txt b/be/requirements.txt new file mode 100644 index 0000000..8c25f5f --- /dev/null +++ b/be/requirements.txt @@ -0,0 +1,2 @@ +flask +auto_anno_2