请在此基础上添加菜单,运用switch函数
时间: 2024-05-04 08:22:08 浏览: 17
好的,为了让ChitGPT更加交互式,我可以为你添加一个菜单功能。以下是一个简单的示例代码:
```python
print("欢迎使用ChitGPT!请选择以下操作:")
print("1. 闲聊模式")
print("2. 退出程序")
while True:
choice = input("请输入操作编号:")
if choice == "1":
# 进入闲聊模式
print("进入闲聊模式,输入 exit 可以退出该模式。")
while True:
text = input("你:")
if text == "exit":
break
response = chat_model.generate(text)
print("ChitGPT:", response)
elif choice == "2":
# 退出程序
print("谢谢使用ChitGPT!")
break
else:
print("输入有误,请重新输入。")
```
在这个例子中,我们使用了一个 `while` 循环来不断打印菜单并等待用户输入选项。用户输入后,我们使用 `if-elif-else` 语句来判断用户选择的操作,并执行相应的代码块。其中,我们使用了 `switch` 函数的替代方案,即使用 `if-elif-else` 语句来执行不同的操作。在闲聊模式中,我们使用了另一个 `while` 循环来等待用户输入闲聊内容,并使用 `input` 函数获取用户输入,然后使用 `generate` 函数生成回复,并将回复输出到屏幕上。如果用户输入了 "exit",则退出闲聊模式。如果用户选择了退出程序,则结束程序运行。
相关推荐
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)