example.py

from huffman.huffmancoding import HuffmanCoding
from huffman.huffmandecoding import HuffmanDecoding


def main() -> None:
    sample_text = "Hello World!"

    # Initialize encoder
    encoder = HuffmanCoding()
    encoded_bits = encoder.encode(sample_text)

    # Initialize decoder
    decoder = HuffmanDecoding()
    decoded_text = decoder.decode(encoded_bits, encoder.get_tree())

    # Display results
    print("Original text: ", sample_text)
    print("Encoded bits:  ", encoded_bits)
    print("Decoded text:  ", decoded_text)
    print("Code table:    ", encoder.get_table())
    print("Summary:       ", encoder.get_summary())


if __name__ == "__main__":
    main()