alejandrods / cover-letter-generator-gpt2 Goto Github PK

`# -*- coding: utf-8 -*-

import os
import json
import flask
import logging
from textblob import TextBlob
from generate import generate
from flask_cors import CORS, cross_origin
from flask_ngrok import run_with_ngrok
from flask import Flask, render_template, url_for
from flask_wtf import FlaskForm
from wtforms import StringField, SubmitField, TextAreaField
from wtforms.validators import DataRequired
import torch
from transformers import GPT2LMHeadModel, GPT2Tokenizer
from flask import Flask, request, render_template
import json


app = flask.Flask(__name__)
CORS(app)
app.config['PROPAGATE_EXCEPTIONS'] = True
app = Flask(__name__, root_path='/content/cover-letter-generator-gpt2')
app.config['SECRET_KEY'] = '9bad6913d4358ac1395c5c94370ed090'
run_with_ngrok(app)
print(app.root_path)

# Logs
logging.basicConfig(level=logging.INFO,
                    format='%(asctime)s :: %(levelname)s :: %(message)s')


# Load Environment Variables
ROUTE_GENERATE = os.environ.get('ROUTE_GENERATE', 'generate')
N_TOKENS = int(os.environ.get('N_TOKENS', 20))

# This create a json file with the environment variables for javascript script
data_env = {
    "APP_URL": "{}:{}".format(HOST, PORT) if 'localhost' in HOST
    else "{}".format(HOST),
    "ROUTE": "/{}".format(ROUTE_GENERATE),
}

# Save env.json file
with open(os.path.join("/content/cover-letter-generator-gpt2/src/static", "env.json"), 'w') as f:
    json.dump(data_env, f)


@app.route('/content/cover-letter-generator-gpt2/src/templates/main.html')
@cross_origin()
def index():
    """
    Function to render the html file

    :return:
    """

    return flask.render_template('main.html')


@app.route('/generate', methods=['POST'])
@cross_origin()
def prediction():
    """
    Function to receive an user query and return generated text

    :return: Generated text
    """

    income_query = json.loads(flask.request.data)['input']

    logging.info('Query received: {}'.format(income_query))

    arg_lang = TextBlob(income_query)
    lang = arg_lang.detect_language()
    logging.info('Lang detected: {}'.format(lang))

    if lang == 'en':
        res = generate(income_query, size=N_TOKENS)
        generated = res.split(income_query)[-1]
    else:
        res = "lang_det_err"
        generated = "lang_det_err"

    logging.info('Result: {}'.format(res))

    return flask.jsonify({
        'version': 'v1.0.0',
        'body': {'query': income_query,
                 'generated': generated,
                 'result': res}
        }), 200


@app.route('/health_liveness')
@cross_origin()
def health_check():
    """
    Check status code

    :return:
    """

    return flask.jsonify({'status': 'success'}), 200


if __name__ == "__main__":
    app.run()
`