python数字炸弹代码相关的问题

请问如果想在游戏结束之后输入一个数然后重新开始游戏，这个应该怎么设置啊😭😭像图里这样子运行不了

while 1:
    main()
    flag=input('继续游戏?(y/n)')
    if flag=='n':
        break