import gradio as gr
import numpy as np
import cv2
from PIL import Image, ImageOps

def filter_img(input_img,filter_type):
    img = input_img

    new_height = 1080

    ratio = float(new_height) / img.shape[0]
    new_width = int(img.shape[1] * ratio)
    
    img = cv2.resize(img, (new_width,new_height))
    
    filtered_img = np.zeros_like(img)
    
    if filter_type == 'Vintage':
        sepia_filter = np.array([[0.393, 0.769, 0.189],
                         [0.349, 0.686, 0.168],
                         [0.272, 0.534, 0.131]])

        filtered_img = np.dot(img, sepia_filter.T)
        filtered_img = (filtered_img / filtered_img.max() * 255).astype(np.uint8)

    elif filter_type == 'Gray-sharp':
        img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
        filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
    elif filter_type == 'Sharp':
        filtered_img = cv2.addWeighted(img,4, cv2.blur(img,(128,128)),-4,128)
    elif filter_type == 'Gray':
        filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
    elif filter_type == 'B&W':
        filtered_img = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
        _,filtered_img = cv2.threshold(filtered_img,127,255,cv2.THRESH_BINARY)
    elif filter_type == 'Magenta':
        filtered_img[:,:,0] = img[:,:,0]
        filtered_img[:,:,2] = img[:,:,2]
    elif filter_type == 'Cyan':
        filtered_img[:,:,1] = img[:,:,1]
        filtered_img[:,:,2] = img[:,:,2]
    elif filter_type == 'Yellow':
        filtered_img[:,:,0] = img[:,:,0]
        filtered_img[:,:,1] = img[:,:,1]
    elif filter_type == 'Blue':
        filtered_img[:,:,2] = img[:,:,2]
    elif filter_type == 'Green':
        filtered_img[:,:,1] = img[:,:,1]
    elif filter_type == 'Red':
        filtered_img[:,:,0] = img[:,:,0]

    return filtered_img

demo = gr.Interface(filter_img,
                        inputs = [ gr.inputs.Image(),gr.inputs.Dropdown(["Gray","B&W","Sharp","Gray-sharp","Vintage","Magenta","Yellow","Cyan","Blue","Green","Red"])], 
                        outputs= gr.components.Image(),
                        title = "Image Filterz", 
                        description="Cool image filters!"
                    )

demo.launch(inline=False)