from Cimpl import *
#Code written by Adam Kahraman


'''Program returns the original picture selected and also the original picture with added filter of setting the components of Red and Green to zero while not changing the components of Blue in each pixel. Assuming that the image selected is in the same folder as the program.
'''

original_image = load_image(choose_file())
new_image = copy(original_image)

def grayscale(image: Image) -> Image:
    """Return a grayscale copy of image.

    >>> image = load_image(choose_file())
    >>> gray_image = grayscale(image)
    >>> show(gray_image)
    """
    new_image = copy(image)
    for x, y, (r, g, b) in image:

        # Use the pixel's brightness as the value of RGB components for the
        # shade of gray. These means that the pixel's original colour and the
        # corresponding gray shade will have approximately the same brightness.

        brightness = (r + g + b) // 3

        # or, brightness = (r + g + b) / 3
        # create_color will convert an argument of type float to an int

        gray = create_color(brightness, brightness, brightness)
        set_color(new_image, x, y, gray)
    return new_image



def sepia(image: Image) -> Image:
    '''Returns a sepia copy of an image selected when the function calls the grayscale function that is calling new_image.
    >>>sepia(grayscale(new_image))
    '''
    sepia_image = copy(new_image)
    for x, y, (r, g, b) in image:
        if r >= 191:
            b = b * 0.93
            r = r * 1.08
        if r <= 191:
            if r >= 63:
                b = b * 0.85
                r = r * 1.15
        if r < 63:
            b = b * 0.9
            r = r * 1.1
        new_colour = create_color(r, g, b)
        set_color(sepia_image, x, y, new_colour)
    return show(sepia_image)