import cv2
from PIL import Image
import numpy as np
import os

def image_to_emoji(image, width):
    img = Image.fromarray(image).convert('L')
    #resize based on width
    aspect_ratio = img.height / img.width
    height = int(width * aspect_ratio)
    img = img.resize((width, height))
    pixels = np.array(img)
    #set the black/white threshold to the mean value of the pixels
    threshold = np.mean(pixels)
    ascii_image = ""
    #this is quite a bit faster than the previous double for loop method
    emoji_array = np.where(pixels > threshold, "⬜", "⬛")
    ascii_image = '\n'.join([''.join(row) for row in emoji_array])
    return ascii_image

def video_to_ascii(video_path, output_folder, width):
    os.makedirs(output_folder, exist_ok=True)
    video = cv2.VideoCapture(video_path)
    frame_count = 0
    while True:
        success, frame = video.read()
        if not success: #if frame fails to load/doesn't exist
            break
       
        emoji_frame = image_to_emoji(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB), width)
        frame_count += 1
        with open(os.path.join(output_folder, f"{frame_count}.txt"), "w", encoding="utf-8") as f:
            f.write(emoji_frame)    
        print(frame_count)
   
    video.release()
    print("done")

# Usage
video_path = input("Enter video path: ")
try:
    width = int(input("Enter desired width (100 is usually best): "))
except:
    print("Invalid width, defaulting to 100")
    width = 100
output_folder = "./frames"
video_to_ascii(video_path, output_folder, width)