ASCII.py

import cv2 as cv
import math
import os
import sys
import numpy as np
from PIL import Image
from io import BytesIO

os.system("clear")

pixel_map = r"$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/|()1{}[]?-_+~<>i!lI;:,\"^`'. "
pixel_hash = []

cap = None
screen_shot = None

def map_color_value(old_value):
    return math.floor(((old_value - 0) / (255 - 0)) * (len(pixel_map) - 1 - 0) + 0)

for i in range(256):
    pixel_hash.append(pixel_map[map_color_value(i)])

# Define keybindings and menu instructions
menu = """
=== ASCII Art Camera ===

Instructions:
- Press W to start the webcam.
- Press Q to quit the webcam.
- Press S to take a screenshot.
- Press V to display the last screenshot.

Press Enter to start...

"""

# Print the menu and wait for user input
print(menu)
input()

os.system("clear")  # Clear the screen after user input

while True:
    key = input("Press a key: ")

    if key == 'Q' or key == 'q':
        if cap is not None:
            cap.release()
        cv.destroyAllWindows()
        break

    if key == 'W' or key == 'w':
        cap = cv.VideoCapture(0)
        cap.set(cv.CAP_PROP_FRAME_WIDTH, 35)
        cap.set(cv.CAP_PROP_FRAME_HEIGHT, 20)

        for i in range(10):
            ret, image = cap.read()
            if image is None:
                break

            image_gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)

            for i, row in enumerate(image_gray):
                for j, column in enumerate(row):
                    r, g, b = image[i][j]
                    print(f"\x1b[38;2;{r};{g};{b}m{pixel_hash[column]}\x1b[0m", end='')

                print()

            sys.stdout.write(f"\033[{0};{0}H")
            sys.stdout.flush()

    if key == 'S' or key == 's':
        if cap is not None:
            ret, image = cap.read()
            if image is not None:
                screen_shot = np.copy(image)
                print("Screenshot taken.")

    if key == 'V' or key == 'v':
        if screen_shot is not None:
            img = Image.fromarray(cv.cvtColor(screen_shot, cv.COLOR_BGR2RGB))
            img.show()
        else:
            print("No screenshot available.")