#Escolha seu inicial
print("Escolha seu inicial:")
print("0 - Água")
print("1 - Fogo")
print("2 - Terra")

escolha = input("Digite o número da sua escolha: ")

if escolha == "0":
    print("Você escolheu Água! ")
elif escolha == "1":
    print("Você escolheu Fogo! ")
elif escolha == "2":
    print("Você escolheu Terra! ")
else:
    print("Opção inválida.")