import gradio as gr
from main import main


with gr.Blocks(title="Privacy Mask", css="footer{display:none !important}") as demo:
    
    with gr.Row():
        image_input = gr.Image(type="filepath", label="Invoice Image")
        image_output = gr.Image(show_download_button=True, type="pil", label="Masked Invoice Image")
    
    with gr.Row():
        image_button = gr.Button("Masking")
        image_button.click(main, inputs=image_input, outputs=image_output)
        clear_button = gr.ClearButton([image_input, image_output])

demo.launch()