将刚刚进行的修改整理为一份代码发我

以下是整理后的代码：

import openai
import re

# 设置OpenAI认证密钥
openai.api_key = "YOUR_API_KEY"

# 设置模型和参数
model_engine = "text-davinci-002"
prompt = "Hello, my name is John. What can I help you with today?"

# 与OpenAI进行对话
def openai_chat(prompt, model_engine, message):
    # 设定对话的上下文
    completions = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.7,
    )

    # 获取最佳回答
    message = completions.choices[0].text
    message = re.sub('[^0-9a-zA-Z\n\.\?,!]+', ' ', message)
    message = message.strip()

    return message

# 调用openai_chat函数进行对话
while True:
    user_message = input("You: ")
    prompt = "{}\nUser: {}".format(prompt, user_message)
    response = openai_chat(prompt, model_engine, user_message)
    prompt = "{}\nOpenAI: {}".format(prompt, response)
    print("OpenAI: ", response)

请注意，需要将 "YOUR_API_KEY" 替换为您的OpenAI API密钥。执行代码时，您需要在命令行中输入消息以与AI进行交互。