from flask import Flask, render_template, request
import openai
app = Flask(__name__)
# 设置OpenAI API凭证
openai.api_key = 'YOUR_API_KEY'
# 路由：处理网页根目录请求
@app.route('/', methods=['GET', 'POST'])
def index():
if request.method == 'POST':
# 获取用户输入的消息
user_message = request.form['user_message']
# 调用ChatGPT API生成回复
response = openai.Completion.create(
engine='text-davinci-003',
prompt=user_message,
max_tokens=50,
temperature=0.7
)
# 提取回复并返回给网页
chatbot_message = response.choices[0].text.strip()
return render_template('index.html', user_message=user_message, chatbot_message=chatbot_message)
else:
# 首次加载网页时，返回空的聊天记录
return render_template('index.html', user_message='', chatbot_message='')
if __name__ == '__main__':
app.run()
