import gradio as gr
from PIL import Image
from generic import evaluate


def generate_viz(image):
    
    print(f"Image: {image.size}")

    viz= evaluate(image, 'cpu')
    viz.savefig("visualization.png")
    return Image.open("visualization.png").convert("RGB")

title = "Explain DETR 😟 for Object Detection"

iface = gr.Interface(fn=generate_viz, inputs=[
        gr.Image(type="pil", label="Input Image"),
    ],
    outputs=[gr.Image(label="XAI-Image", width=700, height=300)],
    title=title,
    allow_flagging="never",
    cache_examples=True,
    examples=[
        ["DETR/samples/cat-remote.jpg"],
        ["DETR/samples/ski.jpg"],
        ["DETR/samples/bus.jpg"],
        ["DETR/samples/piza.jpg"],
    ],
)

iface.launch(debug=True)