继续写出完整代码

好的，以下是完整的代码：

import openai
import re

# 设置 CSDN API 密钥
openai.api_key = "YOUR_API_KEY"

# 定义对话模型 ID
model_engine = "text-davinci-002"

# 定义对话起始语句
start_chat_log = '''Human: Hello, I'm a person. What's your name?
AI: Hi! I'm ChitGPT, a language model developed by CSDN. How can I assist you today?'''

# 定义函数：给定对话模型 ID 和对话起始语句，返回一个对话生成器
def generate_response(model_engine, start_chat_log):
    # 创建 CSDN 对话生成器
    response_generator = openai.Completion.create(
        engine=model_engine,
        prompt=start_chat_log,
        temperature=0.7,
        max_tokens=1500,
        n=1,
        stop=None,
        timeout=5,
    )

    # 从 CSDN 返回的对话生成器中获取 AI 的回复
    response = response_generator.choices[0].text

    # 清理 AI 的回复，去掉多余的空格和换行符
    response = re.sub('[^0-9a-zA-Z,.!?]+', ' ', response)
    response = response.strip()

    return response

# 开始对话
print("ChitGPT: Hi! I'm ChitGPT, a language model developed by CSDN. How can I assist you today?")

while True:
    # 获取用户输入
    user_input = input("You: ")

    # 将用户输入加入对话起始语句中，生成新的对话起始语句
    new_chat_log = f"{start_chat_log}\nHuman: {user_input}\nAI:"

    # 生成 AI 的回复
    response = generate_response(model_engine, new_chat_log)

    # 输出 AI 的回复
    print("ChitGPT:", response)

请注意，在运行这个代码之前，你需要将 `YOUR_API_KEY` 替换为你自己的 CSDN API 密钥。此外，如果你想使用不同的对话模型，可以将 `model_engine` 替换为你想使用的模型的 ID。