import cv2
import numpy as np

RED = [
    [160,100,100],
    [179,255,255]
]

BLUE = [
    [110,50,50],
    [130, 255,255]
]



def isolate_by_color(filename="blueball.jpg", lower_bound=[110,50,50], upper_bound=[130, 255,255], color=None):
    img = cv2.imread(filename)

    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)

    if color == None:
        lower_color = np.array(lower_bound)
        upper_color = np.array(upper_bound)

        mask = cv2.inRange(hsv, lower_color, upper_color)
    else:

        lower_color1 = np.array(color[0])
        upper_color1 = np.array(color[1])

        mask = cv2.inRange(hsv, lower_color1, upper_color1)
        #mask = cv2.inRange(mask, lower_color1, upper_color1)

    res = cv2.bitwise_and(img, img, mask=mask)

    cv2.imshow("result",res)


    k = cv2.waitKey(0)
    if k == ord('q'):
        cv2.destroyAllWindows()

if __name__ == "__main__":
    #isolate_by_color("blueball.jpg")
    #isolate_by_color("girl.jpeg", lower_bound=[0, 100, 100], upper_bound=[10,255,255])
    isolate_by_color("girl.jpeg", color=red)