[Projeto] Alterar quantidade de estoque

estoque = { 
    "Caderno universitário": 50, 

    "Caneta azul": 120, 

    "Borracha branca": 30 
} 


def solicitar_nome_produto() -> str:
    while True:
        produto = input('Digite o nome do produto a ser atualizado: ').strip()
        if produto:
            return produto
        print('O nome do produto não pode ser vazio. Tente novamente.')


def solicitar_quantidade_produto() -> int:
    while True:
        try:
            quantidade = int(input('Digite a nova quantidade: '))
            if quantidade >= 0:
                return quantidade
            print('A quantidade não pode ser negativa.')
        except ValueError:
            print('Digite apenas números válidos (EX: 200). Tente Novamente.')


def alterar_quantidade_produto(estoque: dict, produto: str) -> None:
    nova_quantidade = solicitar_quantidade_produto()
    if produto in estoque:
        estoque[produto] = nova_quantidade
        print(f"Quantidade de '{produto}' atualizada para {nova_quantidade}.")
    else:
        print('Produto não encontrado no estoque.')


def exibir_novo_estoque(estoque: dict) -> None:
    print('\nEstoque atualizado:')
    for produto, quantidade in estoque.items():
        print(f'{produto}: {quantidade}')  


def main():
    produto = solicitar_nome_produto()
    alterar_quantidade_produto(estoque, produto)
    exibir_novo_estoque(estoque)


if __name__ == '__main__':
    main()