From 49c4c85b7caccaaa82c75a2c511479de30ec5442 Mon Sep 17 00:00:00 2001 From: Tom Edwards <edwardstj1@cardiff.ac.uk> Date: Wed, 12 Jun 2024 09:15:20 +0100 Subject: [PATCH] Work towards USAS --- api/api_functions.py | 13 ++++++++++--- func/usas/usas.py | 12 ++++++++++-- main.py | 9 +++++++-- requirements.txt | 3 ++- 4 files changed, 29 insertions(+), 8 deletions(-) diff --git a/api/api_functions.py b/api/api_functions.py index 92ab989..7f65efb 100644 --- a/api/api_functions.py +++ b/api/api_functions.py @@ -16,6 +16,13 @@ def get_ner_for_data(page): return make_response(jsonify(result), 400) -def get_usas_for_data(): - result = run_usas_on_text() - return result +# Perform USAS analysis on a file +# TAKES XML text page +# Returns NER results +def get_usas_for_data(page): + result = run_usas_on_text(page) + + if result["code"] == "SUCCESS": + return make_response(jsonify(result), 201) + + return make_response(jsonify(result), 400) diff --git a/func/usas/usas.py b/func/usas/usas.py index a7bc640..562fe62 100644 --- a/func/usas/usas.py +++ b/func/usas/usas.py @@ -1,3 +1,11 @@ +import spacy -def run_usas_on_text(): - return "hello usas" \ No newline at end of file + + +# Perform USAS on Text +def run_usas_on_text(page): + + + + result = {'output': "Hello USAS",'message': 'Done', 'code': 'SUCCESS'} + return result diff --git a/main.py b/main.py index 3fec6eb..e18df13 100644 --- a/main.py +++ b/main.py @@ -1,9 +1,12 @@ from flask import Flask from flask import request +from flask_cors import CORS, cross_origin from api.api_functions import * app = Flask(__name__) +cors = CORS(app) +app.config['CORS_HEADERS'] = 'Content-Type' @app.route("/") @@ -18,7 +21,7 @@ def test(): @app.route('/ner', methods=['POST']) def ner(): - request_data = request.form.to_dict() + request_data = request.get_json() page = request_data['page'] result = get_ner_for_data(page) return result @@ -26,5 +29,7 @@ def ner(): @app.route("/usas") def usas(): - result = get_usas_for_data() + request_data = request.get_json() + page = request_data['page'] + result = get_usas_for_data(page) return result diff --git a/requirements.txt b/requirements.txt index b9f546d..aacc856 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,5 @@ flask~=3.0.3 torch~=2.3.1 transformers~=4.41.2 ckip-transformers~=0.3.4 - +flask-cors~=4.0.1 +spacy~=3.7.4 -- GitLab