Update app.py

app.py

@@ -437,6 +437,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                         with gr.Row():
                             run_button = gr.Button("Generate")
                             paste_back = gr.Checkbox(True, label="Paste back original")
+                        sam_block()
                 with gr.Row(equal_height=False):
                     input_image = gr.ImageMask(
                         type="pil", label="Input Image", layers=True
@@ -492,10 +493,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                             label="Input Image"
                         )
                         with gr.Row():
-                            with gr.Column(scale=2):
-                                prompt_input = gr.Textbox(label="Prompt (Optional)")
-                            with gr.Column(scale=1):
-                                runout_button = gr.Button("Generate")
+                            prompt_input = gr.Textbox(label="Prompt (Optional)")                                
                         with gr.Row():
                             target_ratio = gr.Radio(
                                 label="Expected Ratio",
@@ -554,8 +552,9 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                                         value=50,
                                         visible=False
                                     )
-                                with gr.Column():
-                                    preview_button = gr.Button("Preview alignment and mask")
+                            with gr.Row():
+                                preview_button = gr.Button("Preview Alignment")
+                                runout_button = gr.Button("Generate")
                     with gr.Column():
                         result_outpaint = ImageSlider(
                             interactive=False,