Upload app.py with huggingface_hub

app.py

@@ -221,16 +221,16 @@ def parse_points_json(points_json: str, width: int, height: int) -> List[Tuple[f
 # Event handlers.                                                             #
 # --------------------------------------------------------------------------- #
 def on_image_change(image: np.ndarray):
-    """Track the current image so markers stay in sync with its dimensions."""
+    """Reset annotations whenever a new image is loaded."""
     if image is None:
-        return None, "[]", None
-    return image, "[]", draw_annotations(image, [])
+        return "[]", None
+    return "[]", draw_annotations(image, [])
 
 
 def on_image_select(image: np.ndarray, points_json: str, evt: gr.SelectData):
     """Append the clicked point (original-image pixel coords) to the annotation list."""
     if image is None:
-        return "[]", None, gr.Warning("Upload an image first.")
+        raise gr.Error("Upload an image first.")
     x, y = evt.index
     try:
         points = json.loads(points_json) if points_json else []
@@ -240,29 +240,23 @@ def on_image_select(image: np.ndarray, points_json: str, evt: gr.SelectData):
     return json.dumps(points), draw_annotations(image, points)
 
 
-def on_clear(points_json: str):
+def on_clear(points_json: str, image: np.ndarray):
     """Clear all annotations."""
-    return "[]"
+    if image is None:
+        return "[]", None
+    return "[]", draw_annotations(image, [])
 
 
-def on_undo(points_json: str):
+def on_undo(points_json: str, image: np.ndarray):
     """Remove the most recent annotation."""
+    if image is None:
+        raise gr.Error("Upload an image first.")
     try:
         points = json.loads(points_json) if points_json else []
     except json.JSONDecodeError:
         points = []
-    return json.dumps(points[:-1]) if points else "[]"
-
-
-def run_rmbg(image: np.ndarray):
-    """Remove the background from the current image (GPU-light step)."""
-    if image is None:
-        return None, "[]", None, gr.Warning("Upload an image first.")
-    if len(image.shape) == 3 and image.shape[2] == 4:
-        return image, "[]", image, None  # already has alpha
-    pil = remove_background(Image.fromarray(image))
-    out = np.array(pil)  # RGBA
-    return out, "[]", out, None
+    points = points[:-1] if points else []
+    return json.dumps(points), draw_annotations(image, points)
 
 
 @spaces.GPU(duration=300)
@@ -386,12 +380,10 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css, title="FlexPart — Image-to-3
                 elem_classes=["annot-preview"],
             )
             points_state = gr.State("[]")
-            current_image = gr.State(None)
 
             with gr.Row():
                 undo_btn = gr.Button("↩ Undo last")
                 clear_btn = gr.Button("🗑 Clear all")
-            rmbg_btn = gr.Button("🧹 Remove background", variant="secondary")
 
             with gr.Accordion("Advanced settings", open=False):
                 seed = gr.Number(value=1000, label="Seed", precision=0)
@@ -439,10 +431,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css, title="FlexPart — Image-to-3
 
     # Wiring
     input_image.select(on_image_select, inputs=[input_image, points_state], outputs=[points_state, annotated_preview])
-    input_image.change(on_image_change, inputs=[input_image], outputs=[current_image, points_state, annotated_preview])
-    undo_btn.click(on_undo, inputs=[points_state], outputs=[points_state])
-    clear_btn.click(on_clear, inputs=[points_state], outputs=[points_state])
-    rmbg_btn.click(run_rmbg, inputs=[input_image], outputs=[input_image, points_state, annotated_preview, annotated_preview])
+    input_image.change(on_image_change, inputs=[input_image], outputs=[points_state, annotated_preview])
+    undo_btn.click(on_undo, inputs=[points_state, input_image], outputs=[points_state, annotated_preview])
+    clear_btn.click(on_clear, inputs=[points_state, input_image], outputs=[points_state, annotated_preview])
     gen_btn.click(
         generate,
         inputs=[input_image, points_state, seed, guidance, steps],