import gradio as gr
import cv2

from inference import (
    generate_grade,
    apply_grade
)


def run(img):

    cv2.imwrite(
        "input.jpg",
        cv2.cvtColor(
            img,
            cv2.COLOR_RGB2BGR
        )
    )


    grade=generate_grade(
        "input.jpg"
    )


    output=apply_grade(
        "input.jpg",
        grade
    )


    return cv2.cvtColor(
        output,
        cv2.COLOR_BGR2RGB
    )



demo=gr.Interface(

    fn=run,

    inputs=gr.Image(),

    outputs=gr.Image(),

    title="Contrastive Color Grade AI"

)


demo.launch()