Update app.py

app.py

@@ -37,15 +37,21 @@ def submit_prompt(prompt):
 # Function to add a new object with validation
 def add_object(object_class, bbox):
     try:
+        # Split and convert bbox string into integers
         x1, y1, x2, y2 = map(int, bbox.split(","))
+        
+        # Validate the coordinates
         if x2 < x1 or y2 < y1:
             return "Error: x2 cannot be less than x1 and y2 cannot be less than y1.", []
         if x1 < 0 or y1 < 0 or x2 > 512 or y2 > 512:
             return "Error: Coordinates must be between 0 and 512.", []
+
+        # If validation passes, add to the lists
         object_classes_list.append(object_class)
         object_bboxes_list.append(bbox)
         combined_list = [[cls, bbox] for cls, bbox in zip(object_classes_list, object_bboxes_list)]
         return combined_list
+
     except ValueError:
         return "Error: Invalid input format. Use x1,y1,x2,y2.", []
 
@@ -80,29 +86,23 @@ 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 custom CSS for orange buttons
-css = """
-button {
-    background-color: orange !important;
-    color: white !important;
-    border: none !important;
-    font-weight: bold;
-}
-"""
-
-with gr.Blocks(css=css) as demo:
+# Gradio UI
+with gr.Blocks() as demo:
     gr.Markdown("# Text-to-Image Generator with Object Addition")
 
-    # Put prompt and submit button in the same row and adjust sizes
-    with gr.Row():
-        prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here").style(width=500)
-        submit_button = gr.Button("Submit Prompt").style(width=100)
+    # 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 = 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
+            )
+            # Replace the button with the simplified "Run" button
+            submit_button = gr.Button("Submit Prompt", scale=0)  # Add scale for button size
 
     # Always visible DataFrame
     objects_display = gr.Dataframe(
@@ -115,7 +115,6 @@ with gr.Blocks(css=css) 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):
@@ -145,7 +144,7 @@ with gr.Blocks(css=css) as demo:
     submit_button.click(
         fn=submit_prompt,
         inputs=prompt,
-        outputs=[objects_display, prompt]
+        outputs=[objects_display, prompt]  # Update both the display and prompt input
     )
 
     # Add object and update display
@@ -155,13 +154,6 @@ with gr.Blocks(css=css) 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,
@@ -170,4 +162,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()