Update app.py

app.py

@@ -121,7 +121,6 @@ with gr.Blocks() as demo:
 
     add_button = gr.Button("Add Object")
 
-
     # Advanced settings in a collapsible accordion
     with gr.Accordion("Advanced Settings", open=False):
         seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
@@ -144,10 +143,10 @@ with gr.Blocks() as demo:
             )
 
     generate_button = gr.Button("Generate Image")
-    
     result = gr.Image(label="Generated Image")
-    
-    refresh_button = gr.Button("Refresh")  # New Refresh button
+
+    # Refresh button to clear arrays and reset inputs (moved below the result)
+    refresh_button = gr.Button("Refresh")
 
     # Submit the prompt and update the display
     submit_button.click(
@@ -163,13 +162,6 @@ with gr.Blocks() as demo:
         outputs=[objects_display]
     )
 
-    # Refresh button to clear arrays and reset inputs
-    refresh_button.click(
-        fn=clear_arrays,
-        inputs=None,
-        outputs=[objects_display, prompt]
-    )
-
     # Generate image based on added objects
     generate_button.click(
         fn=generate_image,
@@ -177,5 +169,12 @@ with gr.Blocks() as demo:
         outputs=[result, seed]
     )
 
+    # Refresh button to clear arrays and reset inputs
+    refresh_button.click(
+        fn=clear_arrays,
+        inputs=None,
+        outputs=[objects_display, prompt]
+    )
+
 if __name__ == "__main__":
     demo.launch()