import numpy as np
import random
from PIL import Image, ImageDraw

# mode = int(input('mode:'))
image = Image.open("temp.jpg")
draw = ImageDraw.Draw(image)
width = image.width
height = image.height

# width = image.size[0]
# height = image.size[1]
pix = image.load()
k = 0


def set_rand_pict(x, k):
    for i in range(width):
        for j in range(height):
            a = x
            b = x
            c = x
            draw.point((i, j), (a, b, c))
    image.save("temp_" + str(k) + ".jpg", "JPEG")


def brightness_down(k):
    k += 1
    for i in range(width):
        for j in range(height):
            a = pix[i, j][0] - 1
            b = pix[i, j][0] - 1
            c = pix[i, j][0] - 1
            if a < 0:
                a = 0
            if b < 0:
                b = 0
            if c < 0:
                c = 0
            if a > 255:
                a = 255
            if b > 255:
                b = 255
            if c > 255:
                c = 255

            draw.point((i, j), (a, b, c))
    image.save("temp_" + str(k) + ".jpg", "JPEG")
    return (k)


def brightness_up(k):
    k += 1
    for i in range(width):
        for j in range(height):
            a = pix[i, j][0] + 1
            b = pix[i, j][0] + 1
            c = pix[i, j][0] + 1
            if a < 0:
                a = 0
            if b < 0:
                b = 0
            if c < 0:
                c = 0
            if a > 255:
                a = 255
            if b > 255:
                b = 255
            if c > 255:
                c = 255

            draw.point((i, j), (a, b, c))
    image.save("temp_" + str(k) + ".jpg", "JPEG")
    return (k)


def score(j):
    if j > 128:
        s = j - 128
        return (s)
    else:
        s = 128 - j
        return (s)


def down(i):
    print('down ', score(x))
    return (i)


def up(i):
    i += 1
    print('up ', score(x))
    return (i)


x = np.random.randint(255)
set_rand_pict(x, k)
print(x)
while x != 128:
    if x < 128:
        x = up(x)
        k = brightness_up(k)
    else:
        x = down(x)
        k = brightness_down(k)
print('x = ', x)
del draw