dba_app.py

# Copyright (C) 2020 IDIR Lab - UT Arlington
#
#     This program is free software: you can redistribute it and/or modify
#     it under the terms of the GNU General Public License v3 as published by
#     the Free Software Foundation.
#
#     This program is distributed in the hope that it will be useful,
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
#     GNU General Public License for more details.
#
#     You should have received a copy of the GNU General Public License
#     along with this program.  If not, see <https://www.gnu.org/licenses/>.
#
# Contact Information:
#     See: https://idir.uta.edu/cli.html
#
#     Chengkai Li
#     Box 19015
#     Arlington, TX 76019
#

from urllib import parse
from sanic import Sanic
from sanic.response import json
from nltk import sent_tokenize

from adv_transformer.core.api.api_wrapper import ClaimSpotterAPI

app = Sanic(__name__)
api = ClaimSpotterAPI()


def get_user_input(r, input_text, k="input_text"):
    try:
        if r.method == "GET":
            return parse.unquote_plus(input_text.strip())
        elif r.method == "POST":
            return r.json.get(k, "")
    except Exception as e:
        print(e)

    return ""


@app.route("/score/text/<input_text:.*>", methods=["POST"])
async def score_text(request, input_text):
    """
    Returns the scores of the text provided.

    Parameters
    ----------
    input_text : string
        Input text to be scored.
    
    Returns
    -------
    <Response>
        Returns a response object with the body containing a json-encoded dictionary containing the `claim`, it's `result`, and the `scores` associated with it.

        `claim` : string
        `result` : string
        `scores` : list[float]
    """
    input_text = get_user_input(request, input_text)
    scores = api.single_sentence_query(input_text) if input_text else []

    return json([{'claim': input_text, 'score': scores[1]}])


@app.route("/score/text/sentences/<input_text:.*>", methods=["POST"])
async def score_text(request, input_text):
    """
    Returns the scores of the text provided.

    Parameters
    ----------
    input_text : string
        Input text to be scored.

    Returns
    -------
    <Response>
        Returns a response object with the body containing a json-encoded dictionary containing the `claim`, it's `result`, and the `scores` associated with it.

        `claim` : string
        `result` : string
        `scores` : list[float]
    """
    input_text = get_user_input(request, input_text)
    sentences = sent_tokenize(input_text)
    scores = api.batch_sentence_query(sentences) if input_text else []

    return json([{'claim': sentences[i], 'score': s[1]} for i, s in enumerate(scores)])

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=8009)