From ae9497ac2c5685b654c7f74a8bf56e3eaaf3a960 Mon Sep 17 00:00:00 2001
From: Marc Roig Vilamala <marc.roig.vilamala@gmail.com>
Date: Fri, 28 Feb 2025 11:06:54 +0000
Subject: [PATCH] Added API endpoint for stance function
---
api/api_functions.py | 11 ++++++++++-
main.py | 14 +++++++++++++-
2 files changed, 23 insertions(+), 2 deletions(-)
diff --git a/api/api_functions.py b/api/api_functions.py
index f26c783..5914c46 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 9837e3b..36cbaf6 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
--
GitLab