from datetime import datetime
escolha=True
while escolha:
    print("\n")
    print("      Tempo de Espera")
    print("""
    1. Tempo de Espera 
    0. Exit/Quit/Saída
    """)
    escolha= input("Escolha uma opção:  ")
    if escolha=="1":
        data_inicio_str = input("Digite a data de início (AAAA-MM-DD): ")
        data_inicio = datetime.strptime(data_inicio_str, "%Y-%m-%d")
        data_final_str = input("Digite a data final (AAAA-MM-DD): ")
        data_final = datetime.strptime(data_final_str, "%Y-%m-%d")
        diferenca = data_final - data_inicio
        dias_de_espera = diferenca.days
        print(f"Total de dias de espera: {dias_de_espera} dias")
    elif escolha=="0":
      print("\n Adeus")
      escolha = None
    else:
       print("\n Escolha não válida.\n Tente outra vez.")