Getão da Fruteira

fruteira = {}

def menu():
    print("====================== Menu ======================")
    print("\t1. Acrescentar nova Fruta")
    print("\t2. Adicionar quantidade a uma Fruta existente")
    print("\t3. Remover quantidade Fruta já existente")
    print("\t4. Ver a Frutaria")
    print("\t5. Sair")

while True:
    menu()
    escolha = input("\nDigite o número da opção desejada: ")

    if escolha == "1":
        nome_fruta = input("Nome da Fruta: ")
        quantidade_inicial = int(input("Digite a quantidade inicial: "))
        fruteira[nome_fruta] = quantidade_inicial

    elif escolha == "2":
        nome_fruta = input("Nome da Fruta: ")
        if nome_fruta in fruteira:
            quantidade = int(input("Quantidade a ser adicionada: "))
            fruteira[nome_fruta] += quantidade
        else:
            print("Fruta não encontrada na fruteira.")

    elif escolha == "3":
        nome_fruta = input("Nome da Fruta: ")
        quantidade_eliminar = int(input("Digite a quantidade a retirar: "))
        if nome_fruta in fruteira:
            quantidade_atual = fruteira[nome_fruta]
            if quantidade_eliminar <= quantidade_atual:
                fruteira[nome_fruta] -= quantidade_eliminar
                print(f"Quantidade {quantidade_eliminar} de {nome_fruta} removida da fruteira.")
            else:
                print("A quantidade a ser retirada é maior do que a quantidade disponível na fruteira.")
        else:
            print("Fruta não encontrada na fruteira.")

    elif escolha == "4":
        if fruteira:
            for fruta, quantidade in fruteira.items():
                print(f"Nome da Fruta: {fruta}, Quantidade: {quantidade}")
        else:
            print("A fruteira está vazia.")

    elif escolha == "5":
        print("Saindo do programa...")
        break

    else:
        print("Opção inválida. Por favor, escolha um número válido.")