import time
import random


def proximo_metro_linha(linha):
    print(f"Chegou à estação e deseja pegar a Linha {linha}.")
    tempo_espera = random.randint(1, 5)
    while tempo_espera > 0:
        print(f"Tempo estimado de espera para o próximo Metro na Linha {linha}: {tempo_espera} minutos.")
        time.sleep(60)
        tempo_espera -= 1
    print(f"O próximo Metro na Linha {linha} chegou!")


def main():
    print("Bem-vindo ao simulador de tempos de espera do Metro de Lisboa!")
    while True:
        print("Por favor, selecione a linha desejada:")
        linha = input("Digite o número da linha (1, 2, 3, 4): ")

        try:
            linha = int(linha)  # Converte a entrada para inteiro
            if linha < 1 or linha > 4:
                print("Linha inválida. Por favor, escolha uma linha de 1 a 4.")
                continue
        except ValueError:
            print("Entrada inválida. Por favor, insira um número entre 1 e 4.")
            continue

        proximo_metro_linha(linha)

        pergunta = input("Deseja ver o tempo de espera para outra linha? (s/n): ")
        if pergunta.lower() != 's':
            print("Obrigado por usar o simulador de tempos de espera do Metro de Lisboa!")
            break
if __name__ == "__main__":
    main()