You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.8 KiB
57 lines
1.8 KiB
#!/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)
|
|
|
|
|
|
|