[Update]Add attack section

app.py

@@ -76,10 +76,19 @@ with gr.Blocks(css=custom_css) as demo:
 
         with gr.Column():
              shown_columns_step = gr.Slider(
-                            1, 100, value=40, 
-                            step=1, label="Attack Steps", info="Choose between 1 and 100",
+                            0, 100, value=40, 
+                            step=1, label="Attack Steps", info="Choose between 0 and 100",
                             interactive=True,)
-
+    with gr.Row() as attack:
+        with gr.Column(scale=1, min_width=260):
+            text_input = gr.Textbox(label="Input Prompt")
+            img1 = gr.Image("images/cheetah.jpg",label="Image")
+        with gr.Column(scale=1, max_width=60):
+            start_button = gr.Button("UnlearnDiffAtk!")
+        with gr.Column(scale=1, min_width=260):
+            text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
+            img2 = gr.Image("images/cheetah.jpg",label="After Defense")
+            
     # with gr.Column():
     #     gr.Examples(examples=[
     #         ["CompVis/stable-diffusion-v1-4", "nudity", "text_grad"]