app.py

import gradio as gr
from PIL import Image

from src.main import IMAGE_REF_PATH, epc_scorer


REFERENCE_IMAGE = Image.open(IMAGE_REF_PATH)


def epc_scorer_wrapper(test_img):
    resp = epc_scorer.callback(None, test_img)
    if resp:
        return f"""
            <p>
                <span style="color: {"forestgreen" if resp[1] == "match" else "tomato"}; font-size: 50px;">{resp[1]}</span> <i>with 
                <span style="color: cyan; font-size: 25px;">{resp[2]}</span> confidence!</i>
            </p>
        """
    else:
        return ""


with gr.Blocks() as demo:
    gr.Markdown("# EPC? Or Not?")
    with gr.Row():
        gr.Image(value=REFERENCE_IMAGE, label="Reference Image", interactive=False)
        user_input = gr.Image(
            label="Upload Sample Image", sources=["upload"], type="pil"
        )
        text_output = gr.Markdown(label="Result")
    gr.Markdown(
        "### *Please wait a while after uploading your first image as the models might get loaded in the background*"
    )
    user_input.change(epc_scorer_wrapper, inputs=user_input, outputs=text_output)


if __name__ == "__main__":
    demo.launch()