import numpy as np
from matplotlib.colors import rgb_to_hsv

from check import plot_preds
from utils import extract_colors

if __name__ == "__main__":
    rgb_tensor, _ = extract_colors()
    xkcd_rgb = rgb_tensor.numpy()
    xkcd_hsv = rgb_to_hsv(xkcd_rgb)
    plot_preds(xkcd_hsv[:, 0], fname="hsv", roll=True)
    rgb = np.eye(3)
    print("Pure RGB in Hue-Space:")
    print(rgb_to_hsv(rgb)[:, 0])