import pyautogui
import numpy as np
import time
from PIL import Image

# координаты элементов
chng_color_pos = (1128, 83)
chng_color_ok_pos = (491, 508)
coord_field = (6, 145)
red_pos = (885, 438)
green_pos = (885, 461)
blue_pos = (885, 485)
color_inputs = red_pos, green_pos, blue_pos

white_color = [255, 255, 255]

# открытие и уменьшение изображения
im = Image.open("./image.bmp")
w, h = pyautogui.size()
im_w, im_h = im.size
im = im.resize((int(im_w / (im_h / 50)), 50), Image.CUBIC)

# разложение изображения на пиксели и расчет их координат
im_w, im_h = im.size
pixels = []
for i, pixel in enumerate(im.getdata()):
    y, x = i // im_w, i % im_w
    pixels.append([x, y, *pixel])


# преобразование цветов
def color_transform(pixels: list):
    for i, pixel in enumerate(pixels):
        # округляем до 10
        pixels[i] = pixel[:2] + list(map(lambda x: x - (x % 10), pixel[2:]))
    return pixels


# сортировка цветов
def sort_pixels(pixels: list):
    return sorted(pixels, key=lambda x: x[2:], reverse=False)


time.sleep(2)

min_time_for_sleep = 1000


def draw(pixels):
    time_for_sleep = min_time_for_sleep
    number_of_last_color = 0
    last_color = None
    for pixel in pixels:
        x, y, *color = pixel
        if color != white_color:
            number_of_last_color += 1
            if color != last_color:
                number_of_last_color = 0
                last_color = color
                time_for_sleep += 50
                pyautogui.leftClick(*chng_color_pos, )
                for color, coord in zip(color, color_inputs):
                    pyautogui.leftClick(*coord, _pause=False)
                    pyautogui.press('del', 3, _pause=False)
                    pyautogui.write(str(color), _pause=False)
                pyautogui.leftClick(*chng_color_ok_pos, _pause=False)
            pyautogui.leftClick(coord_field[0] + x + (x * 7),
                                coord_field[1] + y + (y * 7),
                                _pause=False)

            if color == last_color and time_for_sleep > min_time_for_sleep:
                time_for_sleep -= number_of_last_color / 10

            time.sleep(1 / time_for_sleep)


draw(sort_pixels(color_transform(pixels)))