import gradio as gr

from main import process_and_generate_heatmap

if __name__ == "__main__":
    iface = gr.Interface(
        fn=process_and_generate_heatmap,
        inputs=gr.Image(type="pil"),
        outputs=gr.File(label="Heatmap Image"),
        title="Heatmap Generator",
        description="Upload an image to generate a high-resolution heatmap."
    )

    # Launch the app
    iface.launch(share=True)