Update app.py

app.py

@@ -1,22 +1,41 @@
 import gradio as gr
 from rembg import remove
-from PIL import Image
+from PIL import Image, ImageDraw
+import io
 
-def remove_bg(image_path):
-    if image_path is None:
+def remove_bg(image, points):
+    if image is None:
         return None
-    image = Image.open(image_path)  # ✅ Open the image from the file path
+    
+    # Load image
+    image = Image.open(image).convert("RGBA")
+    
+    # Convert points format
+    keep_points = [tuple(p["coordinates"]) for p in points if p["label"] == "keep"]
+    remove_points = [tuple(p["coordinates"]) for p in points if p["label"] == "remove"]
+    
+    # Apply automatic background removal
     output = remove(image)
+    
+    # Apply manual refinement (simple demonstration using circles)
+    draw = ImageDraw.Draw(output)
+    for point in keep_points:
+        draw.ellipse([point[0]-5, point[1]-5, point[0]+5, point[1]+5], fill=(255, 255, 255, 255))
+    for point in remove_points:
+        draw.ellipse([point[0]-5, point[1]-5, point[0]+5, point[1]+5], fill=(0, 0, 0, 0))
+    
     return output
 
 # Creating Gradio interface
 iface = gr.Interface(
     fn=remove_bg,
-    inputs=gr.Image(type="filepath"),  # ✅ Make sure Gradio passes the file path
+    inputs=[
+        gr.Image(type="filepath"),
+        gr.AnnotatedImage(choices=["keep", "remove"], label="Refine Background")
+    ],
     outputs=gr.Image(type="pil"),
-    title="AI Background Remover",
-    description="Upload an image and remove its background automatically!",
+    title="AI Background Remover with Refinement",
+    description="Upload an image, remove its background, and refine the result using + (keep) and - (remove) points!",
 )
 
 iface.launch()
-