Como aparecer na tela uma mensagem e clicar em uma tecla pra reiniciar o jogo em Pygame

Fiz o jogo da cobrinha dos nokia antigo em pygame, porém quero fazer pra quando a cobra bater na parede ou a cabeça encostar em alguma parte do corpo, o jogo aparecer na tela a mensagem de game over e ter a opção de apertar uma tecla para reiniciar.