请大家看一下这个代码

请大家看一下这个代码，不会写里面内容啦，第二个图片是呈现结果，请求帮忙

mylist = ["张三", "李四", "王五"]


def add(name):
    if(query(name) == False):
        mylist.append(name)
        print("添加成功")
    else:
        print(f"{name}已经存在列表中")


def query(name):
    if name in mylist:
        return True
    return False


def delete(name):

    if(query(name) == False):
        print(f"{name}患者不在列表中")
    else:
        mylist.remove(name)
        print("删除成功")


def remark(name):
    if(query(name) == True):
        new_name = input("请输入更新后患者名字:")
        idx = mylist.index(name)
        mylist[idx] = new_name
        print("备注成功")
    else:
        print(f"{name}患者不在列表中")


def show():
    for show_people in mylist:
        print(f"现有患者为{show_people}")


def main():
    print("="*20)
    print("欢迎使用患者管理系统")
    print("添加患者\t【输入1】* ")
    print("删除患者\t【输入2】")
    print("备注患者\t【输入3】")
    print("展示患者\t【输入4】")
    print("退出\t\t【输入5】")
    print("="*20)
    return input("请输入您的选择：")


while True:
    keyword_input = main()
    if keyword_input == "1":
        name = input("请输入患者名字：")
        add(name)
        continue
    elif keyword_input == "2":
        name = input("请输入患者名字: ")
        delete(name)
        continue
    elif keyword_input == "3":
        name = input("请输入患者名字:")
        remark(name)
        continue
    elif keyword_input == "4":
        show()
        continue
    else:
        print("程序退出")
        break