import gradio as gr
import cv2 as cv

def blur(file):      
    img = cv.imread(file.name)
    img = img[...,::-1]
    avg_blur = cv.blur(img,(5,5))
    gaus_blur = cv.GaussianBlur(img,(5,5),0)
    med_blur  = cv.medianBlur(img,5)

    return img, avg_blur, gaus_blur, med_blur

with gr.Blocks() as demo:
    with gr.Row():
        with gr.Column():
            with gr.Row():
                input_img = gr.Image(label="Original", type="file")
            with gr.Row():
                button = gr.Button("Smooth Image")
            with gr.Row():
                example = gr.Examples(["kitty.jpg"], input_img)
        with gr.Column():
            with gr.Row():
                orig_img = gr.Image(label="Original")
                avg_blur = gr.Image(label="Average Blur")
            with gr.Row():
                gaus_blur = gr.Image(label="Gaussian Blur")
                med_blur = gr.Image(label="Median Blur")
    button.click(blur, inputs=input_img, outputs=[orig_img,avg_blur,gaus_blur,med_blur])

demo.launch()