请给上面程序加一个菜单

好的，这里给您提供一个简单的菜单功能，您可以根据自己的需求进行修改和扩展。

import openai
import inquirer

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

# 设置模型引擎
model_engine = "text-davinci-002"

# 定义对话函数
def chat(prompt):
    response = openai.Completion.create(
        engine=model_engine,
        prompt=prompt,
        max_tokens=1024,
        n=1,
        stop=None,
        temperature=0.7,
    )
    message = response.choices[0].text.strip()
    return message

# 定义菜单选项
menu = [
    inquirer.List('option',
                  message="请选择以下操作：",
                  choices=['1. 开始对话', '2. 退出程序'],
                  ),
]

# 循环显示菜单，直到选择退出
while True:
    # 显示菜单
    answer = inquirer.prompt(menu)

    # 根据用户选择进行操作
    if answer['option'] == '1. 开始对话':
        prompt = input("请输入您的问题或者话题：")
        print(chat(prompt))
    elif answer['option'] == '2. 退出程序':
        break

这个程序会循环显示一个菜单，让用户选择要进行的操作，如果用户选择“开始对话”，则会提示用户输入问题或话题，并根据输入返回 AI 的回答。如果用户选择“退出程序”，则会退出循环，结束程序。