from flask import Flask, request, render_template, jsonify
from revChatGPT.V1 import Chatbot
import json

app = Flask(__name__)
chatbot = Chatbot(config={
    "access_token": "你的access_token 访问https://chat.openai.com/api/auth/session 获取"
})
 

@app.route("/chat",methods=["POST"])
def chat():

    if request.get_data() is None:
        return_dict['return_code'] = '5004'
        return_dict['return_info'] = '请求参数为空'
        return json.dumps(return_dict, ensure_ascii=False)
    # 获取传入的参数

    prompt = get_Data.get('prompt')

    response = chatbot_ask(prompt)
    return jsonify({
            "message": "请求成功",
            "data":  response,
            "status": "Success",
          })

def chatbot_ask(prompt):
    response = ''
    for data in chatbot.ask(prompt):
        response = data["message"]
    return response
    

if __name__ == "__main__": 
    app.run(host="0.0.0.0",port="8000")

简单写了一个基于flask 的chatgpt 请求接口

Logo

助力广东及东莞地区开发者,代码托管、在线学习与竞赛、技术交流与分享、资源共享、职业发展,成为松山湖开发者首选的工作与学习平台

更多推荐