import time
import random
escolha = True

while escolha:
    print("=" * 60)
    print("\t\t\t\tTeste do Tempo de Reação")
    print("=" * 60)
    print("\t\t\t\t\t1 - Teste\n\t\t\t\t\t2 - Sair")
    opcao = str(input("\nDigite a opção: "))
    if opcao == "1":
        print("Prepare-se para o teste de tempo de reação!")
        print("Quando você ver 'Agora!', pressione 'Enter' o mais rápido possível.")
        tempo_espera = random.randint(2, 5)
        time.sleep(tempo_espera)
        print("Agora!")
        inicio = time.time()
        input("Pressione 'Enter' para registrar o tempo de reação.")
        fim = time.time()
        tempo_reacao = fim - inicio
        print(f"Seu tempo de reação foi de {tempo_reacao:.4f} segundos.")
        if tempo_reacao == 5:
            print("Parabéns, você levou 5 segundos! Aqui está seu bolo grátis!")
        else:
            print("Continue treinando.")
    elif opcao == "2":
        print("Fim do Programa")
        break
    else:
        print("Só pode marcar 1 ou 2.")