ascii_gen.py

import PIL.Image

ASCII_CHARS = ["0", ".", "4"]


def to_greyscale(image):
    return image.convert("L")


def resize(image, new_width=70):
    width, height = image.size
    new_height = new_width * height / width
    return image.resize((int(new_width), int(new_height)))


def pixel_to_ascii(image):
    pixels = image.getdata()
    ascii_str = ""
    for pixel in pixels:
        ascii_str += ASCII_CHARS[pixel//128]
    return ascii_str


def main():
    path = "frames/frame-0281.jpg"
    try:
        image = PIL.Image.open(path)
    except:
        print(path, "Unable to find image ")
    # resize image
    image = resize(image)

    # convert image to greyscale image
    # convert greyscale image to ascii characters
    greyscale_image = to_greyscale(image)
    ascii_str = pixel_to_ascii(greyscale_image)
    img_width = greyscale_image.width
    ascii_str_len = len(ascii_str)
    ascii_img = ""

    # Split the string based on width  of the image
    for i in range(0, ascii_str_len, img_width):
        ascii_img += ascii_str[i:i+img_width] + \
            "\n"  # save the string to a file
    print(ascii_img)
    # with open("ascii_image.txt", "w") as f:
    #     f.write(ascii_img)


main()