import flask

from . import ai, prompts

def register(app: flask.Flask):
    @app.route('/api/rephrase', methods=['GET', 'POST'])
    @app.rate_limiter.limit('2 per second')
    def api_rephrase():
        text = flask.request.args.get('text') or flask.request.json.get('text')

        messages = prompts.MESSAGES
        messages.append({
            'role': 'user',
            'content': text
        })

        return flask.jsonify(
            {
                'text': ai.generate(messages)
            }
        )