写一个用python调用chatgpt的代码

以下是一个使用Python调用chatgpt的代码示例：

import openai
import re

# 填写OpenAI API key
openai.api_key = "YOUR_API_KEY"

# 定义聊天函数
def chat(prompt):
    # 调用OpenAI的chat模型
    response = openai.Completion.create(
        engine="davinci",
        prompt=prompt,
        temperature=0.6,
        max_tokens=1500,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0
    )
    # 获取聊天回复
    message = response.choices[0].text
    # 去除多余的空格和换行符
    message = re.sub('[\n]+', '\n', message).strip()
    return message

# 测试聊天函数
prompt = "你好"
while True:
    print("你：", prompt)
    response = chat(prompt)
    print("ChatGPT：", response)
    prompt = input("你：")

注意，在上面的代码中，您需要将YOUR_API_KEY替换为您自己的OpenAI API密钥。此外，还需要安装openai模块。您可以使用以下命令进行安装：

pip install openai

运行代码后，您可以开始与ChatGPT进行交互。在提示符下输入您的消息，ChatGPT将回复您的消息。您可以通过输入“退出”来终止聊天。