把上述的代码转化为一个函数

好的，下面是将代码封装成一个函数的示例：

import openai
openai.api_key = "YOUR_API_KEY"

def generate_response(prompt):
    completions = openai.Completion.create(
        engine="davinci",
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.5,
    )

    message = completions.choices[0].text
    return message.strip()

这个函数的参数为 `prompt`，表示用户输入的文本。函数返回值为 AI 的回复文本。在函数内部，我们使用了 OpenAI 提供的 API 客户端，通过调用 `openai.Completion.create()` 方法来获取 AI 的回复。其中需要传入引擎名称、用户输入的文本、最大生成长度、生成的个数、停用词列表、温度等参数。最后，我们从生成的文本中获取 AI 的回复，并返回。