import string
escolha = True
while escolha:
    print("\n")
    print("Contar número de cada palavra")
    print("""
    1. Contar número
    0. Exit/Quit/Saída
    """)
    escolha = input("Escolha uma opção:  ")

    if escolha == "1":
        def contar_frequencia_palavras(texto):
            texto = texto.lower()
            texto = texto.translate(str.maketrans('', '', string.punctuation))
            palavras = texto.split()
            frequencia = {}

            for palavra in palavras:
                if palavra in frequencia:
                    frequencia[palavra] += 1
                else:
                    frequencia[palavra] = 1

            return frequencia

        texto_exemplo = input("Introduz Texto: ")
        frequencia = contar_frequencia_palavras(texto_exemplo)

        print("Frequência de palavras no texto:")
        for palavra, contagem in frequencia.items():
            print(f"{palavra}: {contagem}")

    elif escolha == "0":
        print("\n Adeus")
        escolha = False
    else:
        print("\n Escolha não válida.\n Tente outra vez.")