import gradio as gr
import utils

with gr.Blocks() as app:
    with gr.Tab("Add Watermark"):
        with gr.Row():
            with gr.Column():
                inp_im = gr.Image(label="Input Image")
                inp_mark = gr.Textbox(label="Watermark")
                mark_btn = gr.Button()
                msg_box = gr.Textbox(label="System Message")
            with gr.Column():
                out_im = gr.Image(label="Watermarked Image")
    mark_btn.click(utils.encode,[inp_im,inp_mark],[out_im,msg_box])
app.launch()