Update app.py

app.py

@@ -86,6 +86,12 @@ def generate_image(prompt, guidance_scale, num_inference_steps, randomize_seed,
 
     return image, seed
 
+# Function to clear all arrays and reset the UI
+def clear_arrays():
+    object_classes_list.clear()
+    object_bboxes_list.clear()
+    return [], gr.update(value="", interactive=True)  # Clear the objects and reset the prompt
+
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Text-to-Image Generator with Object Addition")
@@ -93,16 +99,15 @@ with gr.Blocks() as demo:
     # Put prompt and submit button in the same row
     with gr.Group():
         with gr.Row():
-            # Replace gr.Textbox with gr.Text for a single-line input field
+            # Prompt input and submit button
             prompt = gr.Text(
-                label="Prompt",  # Label for the input field
-                show_label=False,  # Hide the label
-                max_lines=1,  # Single-line input
-                placeholder="Enter your prompt here",  # Placeholder text
-                container=False  # Remove the container background
+                label="Prompt",
+                show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt here",
+                container=False
             )
-            # Replace the button with the simplified "Run" button
-            submit_button = gr.Button("Submit Prompt", scale=0)  # Add scale for button size
+            submit_button = gr.Button("Submit Prompt", scale=0)
 
     # Always visible DataFrame
     objects_display = gr.Dataframe(
@@ -115,6 +120,7 @@ with gr.Blocks() as demo:
         bbox_input = gr.Textbox(label="Bounding Box (x1,y1,x2,y2)", placeholder="Enter bounding box coordinates")
 
     add_button = gr.Button("Add Object")
+    refresh_button = gr.Button("Refresh")  # New Refresh button
 
     # Advanced settings in a collapsible accordion
     with gr.Accordion("Advanced Settings", open=False):
@@ -144,7 +150,7 @@ with gr.Blocks() as demo:
     submit_button.click(
         fn=submit_prompt,
         inputs=prompt,
-        outputs=[objects_display, prompt]  # Update both the display and prompt input
+        outputs=[objects_display, prompt]
     )
 
     # Add object and update display
@@ -154,6 +160,13 @@ 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,
@@ -162,4 +175,4 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch()