import json
import logging

from flask import Flask, render_template, jsonify, request, make_response
import config

from backend.src.chatGPT.chat_gpt import OpenAI

app = Flask(__name__)


@app.route('/')
def index():
    return render_template('view/chat.html')


@app.route('/login', methods=['POST'])
def login():
    # 在这个例子中，使用request对象的json属性来获取JSON请求的参数。在使用POST方法访问/login路径时，可以在请求体中添加以JSON格式提交的参数。
    data = request.json
    username = data.get('username')
    password = data.get('password')
    return f'Login success: {username}, {password}'


@app.route('/register', methods=['POST'])
def register():
    # 在这个例子中，使用request对象的form属性来获取POST请求的参数。在使用POST方法访问/register路径时，可以在请求体中添加username和password参数。
    username = request.form.get('username')
    password = request.form.get('password')
    return f'Register success: {username}, {password}'


@app.route('/chat')
def chat():
    # data = {'name': 'John', 'age': 30}
    # return jsonify(data)
    logging.info(request.args)
    response = make_response(jsonify(OpenAI().chat(request.args.get('content'))))
    response.headers['Access-Control-Allow-Origin'] = '*'
    return response


@app.route('/chat_completion', methods=['POST'])
def chat_completion():
    data = request.data.decode('utf-8')
    data_dicts = json.loads(data)
    query = data_dicts.get('content')
    response = make_response(jsonify(OpenAI().chat(query)))
    response.headers['Access-Control-Allow-Origin'] = '*'
    return response


if __name__ == '__main__':
    # from transformers import GPT2TokenizerFast
    #
    # tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
    # print(len(tokenizer('hello world')['input_ids']))

    config.load_config()
    app.run()