Update gradio_demo.py

gradio_demo.py

@@ -965,8 +965,8 @@ with block:
                         find_objects_button = gr.Button(value="(Option 1) Segment Object from text")
                         text_prompt = gr.Textbox(
                                 label="Text Prompt", 
-                                placeholder="Enter object classes separated by periods (e.g. 'car . person .')",
-                                value="couch . table ."
+                                placeholder="Enter object classes separated by periods (e.g. 'couch . table .') or leave empty to get all objects",
+                                value=""
                             )
                     extract_button = gr.Button(value="(Option 2) Remove Background")
                 with gr.Row():
@@ -1065,16 +1065,22 @@ with block:
                 # Step 1: Input and Extract
                 with gr.Row():
                     with gr.Group():
-                        gr.Markdown("### Step 1: Extract Foreground")
+                        gr.Markdown("Extract Foreground")
                         input_image = gr.Image(type="numpy", label="Input Image", height=480)
-                        # find_objects_button = gr.Button(value="Find Objects")
-                        extract_button = gr.Button(value="Remove Background")
+                        find_objects_button = gr.Button(value="(Option 1) Segment Object from text")
+                        text_prompt = gr.Textbox(
+                                label="Text Prompt", 
+                                placeholder="Enter object classes separated by periods (e.g. 'car . person .'). Leave empty to get all objects",
+                                value=""
+                            )
+                    extract_button = gr.Button(value="(Option 2) Remove Background")
+                        #find_objects_button = gr.Button(value="Find Objects")
                     extracted_fg = gr.Image(type="numpy", label="Extracted Foreground", height=480)
 
                 with gr.Row():    
                 # Step 2: Background and Position
                     with gr.Group():
-                        gr.Markdown("### Step 2: Position on Background")
+                        gr.Markdown("Position on Background")
                         input_bg = gr.Image(type="numpy", label="Background Image", height=480)
                         
                         with gr.Row():
@@ -1199,20 +1205,28 @@ with block:
             def __init__(self):
                 self.original_bg = None
 
-            def update_position(self, background, x_pos, y_pos, scale):
-                """Update composite when position changes"""
-                if background is None:
-                    return None
+            # def update_position(self, background, x_pos, y_pos, scale):
+            #     """Update composite when position changes"""
+            #     if background is None:
+            #         return None
                 
-                if self.original_bg is None:
-                    self.original_bg = background.copy()
+            #     if self.original_bg is None:
+            #         self.original_bg = background.copy()
                 
-                # Convert string values to float
-                x_pos = float(x_pos)
-                y_pos = float(y_pos)
-                scale = float(scale)
+            #     # Convert string values to float
+            #     x_pos = float(x_pos)
+            #     y_pos = float(y_pos)
+            #     scale = float(scale)
                 
-                return mask_mover.create_composite(self.original_bg, x_pos, y_pos, scale)
+            #     return mask_mover.create_composite(self.original_bg, x_pos, y_pos, scale)
+            def update_position(background, x_pos, y_pos, scale):
+                if background is None:
+                    return None
+                # Restore a fresh copy of the original background
+                fresh_bg = bg_manager.original_bg.copy()
+                # Composite the foreground once
+                return mask_mover.create_composite(fresh_bg, float(x_pos), float(y_pos), float(scale))
+
 
         # Create an instance of BackgroundManager
         bg_manager = BackgroundManager()