Update app.py

app.py

@@ -52,7 +52,7 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 520px;
+    max-width: 960px;
 }
 """
 
@@ -64,18 +64,20 @@ with gr.Blocks(css=css) as demo:
 
         input_image = gr.Image(label="Upload the image for editing", type="pil")
         with gr.Row():
+            with gr.Column():
+                prompt = gr.Text(
+                    label="Prompt",
+                    show_label=False,
+                    max_lines=1,
+                    placeholder="Enter your prompt for editing (e.g., 'Remove glasses', 'Add a hat')",
+                    container=False,
+                )
+                
+                run_button = gr.Button("Run", scale=0)
             
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt for editing (e.g., 'Remove glasses', 'Add a hat')",
-                container=False,
-            )
-            
-            run_button = gr.Button("Run", scale=0)
-        
-        result = gr.Image(label="Result", show_label=False)
+            with gr.Column():
+                result = gr.Image(label="Result", show_label=False)
+                reuse_button = gr.Button("Reuse this image", scale=0)
         
         with gr.Accordion("Advanced Settings", open=False):
             
@@ -103,5 +105,10 @@ with gr.Blocks(css=css) as demo:
         inputs = [input_image, prompt, seed, randomize_seed, guidance_scale],
         outputs = [result, seed]
     )
+    reuse_button.click(
+        fn = lambda image: image,
+        inputs = [result],
+        outputs = [input_image]
+    )
 
 demo.launch()