import os
import argparse

import cv2
from util.vision import get_page, get_words


if __name__ == "__main__":
    parser = argparse.ArgumentParser()

    parser.add_argument("--input-image", type=str, required=True)
    parser.add_argument("--output-folder", type=str, required=True, default='files/style_samples/00')

    args = parser.parse_args()

    image = cv2.imread(args.input_image)
    image = cv2.resize(image, (image.shape[1], image.shape[0]))
    result = get_page(image)
    words, _ = get_words(result)

    output_path = args.output_folder
    if not os.path.exists(output_path):
        os.mkdir(output_path)
    for i, word in enumerate(words):
        cv2.imwrite(os.path.join(output_path, f"word{i}.png"), word)