上面的代码可以简化一些吗
时间: 2023-10-21 22:07:11 浏览: 27
可以的,可以把重复使用的代码放到函数中,这样代码就会更简洁。比如说,可以把获取输入和生成回答的代码放到一个名为`chat`的函数中。下面是一个简化后的示例代码:
```python
import openai
import time
# 设置OpenAI API密钥
openai.api_key = "YOUR_API_KEY"
# 打印欢迎信息
print("欢迎使用AI聊天室!")
# 定义函数chat
def chat(prompt):
# 将输入拼接到模板中
completion = openai.Completion.create(
engine="davinci",
prompt=('Conversation with an AI:\n'
'Human: ' + prompt + '\n'
'AI:'),
max_tokens=1024,
n=1,
stop=None,
temperature=0.7,
)
# 获取AI的回答
message = completion.choices[0].text.strip()
# 打印AI的回答
print("AI:", message)
# 进入聊天循环
while True:
# 获取用户输入
prompt = input("Human:")
# 生成AI回答
chat(prompt)
# 等待一秒钟
time.sleep(1)
```
这样代码看起来更整洁,也方便后续的维护和修改。