在刚才的代码中增加一个Title

好的，可以这样修改：

import openai
import re

# 设置OpenAI API key
openai.api_key = "YOUR_API_KEY"

# 定义与OpenAI进行对话的函数
def chat_with_openai(prompt):
    # 调用OpenAI的GPT-3模型生成回答
    response = openai.Completion.create(
        engine="davinci",
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )

    # 从生成的回答中提取文本
    message = response.choices[0].text.strip()

    # 清除换行符和特殊字符
    message = re.sub('[^0-9a-zA-Z\u4e00-\u9fa5\.，。？！]+', '', message)

    return message

# 定义主函数
def main():
    # 设置对话标题
    title = "与AI聊天"
    print("="*30)
    print(title.center(30))
    print("="*30)
    print("你好，我是人工智能，有什么问题可以问我哦！")

    while True:
        # 接收用户输入的信息
        user_input = input("你：")

        # 构造OpenAI的请求
        prompt = f"对话：{user_input}\nAI："

        # 调用OpenAI进行对话
        ai_response = chat_with_openai(prompt)

        # 输出AI的回答
        print(f"{title}：{ai_response}")

# 调用主函数
if __name__ == "__main__":
    main()

现在，对话的开头会出现一个标题，如下所示：

==============================
        与AI聊天         
==============================
你好，我是人工智能，有什么问题可以问我哦！
你：你会什么编程语言？
与AI聊天：我精通多种编程语言，包括Python、Java、C++等。您需要我帮您做什么编程工作吗？
你：好的，可以帮我写一个Python脚本吗？
与AI聊天：当然可以，请告诉我你需要实现什么功能。