import gradio as gr

with gr.Blocks() as demo:
    with gr.Row():
        shown_columns_1 = gr.CheckboxGroup(
            choices=["Church","Parachute","Tench", "Garbage Truck"],
            label="Undersirable Objects",
            elem_id="column-object",
            interactive=True,
        )
    with gr.Row():
        shown_columns_2 = gr.CheckboxGroup(
            choices=["Van Gogh"],
            label="Undersirable Styles",
            elem_id="column-style",
            interactive=True,
        )
    with gr.Row():
        shown_columns_3 = gr.CheckboxGroup(
            choices=["Violence","Illegal Activity","Nudity"],
            label="Undersirable Concepts (Outputs that may be offensive in nature)",
            elem_id="column-select",
            interactive=True,
        )
    with gr.Row():
        with gr.Column(scale=1, min_width=300):
            img1 = gr.Image("images/cheetah.jpg",label="Unlearning")
        with gr.Column(scale=1, min_width=300):
            img2 = gr.Image("images/cheetah.jpg",label="Attacking")

demo.launch()