#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
@Time    : 2023/3/26 18:38
@Author  : ZhangYuPeng
@Email   : zhangyupeng@bonc.com.cn
@File    : GPT_service.py
@Software: PyCharm

'''
import openai
from flask import Flask
from flask import request, jsonify

# openai.api_key = "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"
openai.api_key = "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"

# def chat_drop():
#     openai.api_key = "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"
#     res = openai.ChatCompletion.create(
#         model="gpt-3.5-turbo",
#         messages=[
#             {"role": "user", "content": "请帮我改写这句话:在城市发展进程当中，逐渐呈现出一些综合性的大型建筑群。"},
#             {"role": "assistant", "content": "随着城市的发展，综合性大型建筑群正在逐渐出现。"},
#             {"role": "user", "content": "这句话我不满意，再改一下帮我"}
#         ],
#         temperature=0.5,
#         top_p=1,
#     )
#     print(res.choices[0].message.content)



app = Flask(__name__)


@app.route("/chatgpt", methods=['POST', 'GET'])
def index():
    print(request.remote_addr, request.remote_user)
    prompt = request.json["prompt"]

    response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "user", "content": "请帮我改写这句话:在城市发展进程当中，逐渐呈现出一些综合性的大型建筑群。"},
                {"role": "assistant", "content": "随着城市的发展，综合性大型建筑群正在逐渐出现。"},
                {"role": "user", "content": prompt}
            ],
        )
    message = response.choices[0].message.content
    return message


if __name__ == '__main__':
    app.run('0.0.0.0', port=9999, debug=True)