diff --git a/api/api_functions.py b/api/api_functions.py index f26c7832f28c1fccd83cd642b2adc00a8072d0ed..5914c4651e0fadd57c2117e798426b8a47a23893 100644 --- a/api/api_functions.py +++ b/api/api_functions.py @@ -1,6 +1,7 @@ from flask import make_response, jsonify from func.ner.ner import * +from func.nlp_stance.nlp_stance import run_nlp_stance_on_text from func.sentiment.sentiment import * from func.translation.translation import run_translation_on_text from func.usas.usas import * @@ -132,4 +133,12 @@ def get_dataset_ids(): if result["code"] == "SUCCESS": return make_response(jsonify(result), 200) - return make_response(jsonify(result), 400) \ No newline at end of file + return make_response(jsonify(result), 400) + +def run_nlp_stance(table, dataset_id): + result = run_nlp_stance_on_text(table, dataset_id) + + if result["code"] == "SUCCESS": + return make_response(jsonify(result), 201) + + return make_response(jsonify(result), 400) diff --git a/main.py b/main.py index 9837e3badaace26a1e9e6a6b773808b65f2a9a05..36cbaf6cc1ff00cbef755ed975f5fe80b7122a28 100644 --- a/main.py +++ b/main.py @@ -123,4 +123,16 @@ def getfiles(): page = request_data['dataset_id'] result = get_files_all(page) - return result \ No newline at end of file + return result + +@app.route("/nlpStance", methods=['POST']) +def nlp_stance(): + + request_data = request.get_json() + + result = run_nlp_stance( + table=request_data['table'], + dataset_id=request_data['dataset_id'] + ) + + return result