Update app.py

app.py

@@ -3,47 +3,41 @@ import numpy as np
 from PIL import Image
 from rembg import remove, new_session
 
-# Initialize model with NO cropping
+# Initialize session with proper settings to prevent cropping
 session = new_session("u2net")
+bg_removal_kwargs = {
+    "alpha_matting": False,  # Disable advanced features that cause cropping
+    "session": session,
+    "only_mask": False,
+    "post_process_mask": True  # Clean edges without cropping
+}
 
-def remove_background(img):
+def remove_background(input_image):
     try:
-        # Convert to PIL if needed (preserves original size)
-        if isinstance(img, np.ndarray):
-            img = Image.fromarray(img)
-        elif isinstance(img, dict):  # Handle file uploads
-            img = Image.open(img["name"])
+        # Convert any input to PIL Image
+        if isinstance(input_image, np.ndarray):
+            img = Image.fromarray(input_image)
+        elif isinstance(input_image, dict):  # Handle paste/drop
+            img = Image.open(input_image["name"])
+        else:
+            img = input_image
         
-        # Get original dimensions
-        original_size = img.size
-        
-        # Process WITHOUT cropping
-        result = remove(
-            img,
-            session=session,
-            alpha_matting=False,       # Disable auto-crop
-            only_mask=False,
-            post_process_mask=False    # Disable post-processing that may crop
-        )
-        
-        # Force result to original size
-        if result.size != original_size:
-            result = result.resize(original_size, Image.LANCZOS)
-            
+        # Preserve original size (disable auto-resizing)
+        result = remove(img, **bg_removal_kwargs)
         return result
     
     except Exception as e:
-        print(f"Error: {e}")
-        return img  # Return original if fails
+        print(f"Error: {str(e)}")
+        return input_image  # Return original if fails
 
-# Gradio interface with size preservation
+# Gradio interface with proper image handling
 with gr.Blocks() as demo:
-    gr.Markdown("## 📸 Background Remover (Original Size)")
+    gr.Markdown("## 🖼️ Background Remover (No Cropping)")
     
     with gr.Row():
         input_img = gr.Image(
-            label="Original",
-            type="filepath",  # Better for size preservation
+            label="Original", 
+            type="pil",  # Ensures we get PIL Images
             height=400
         )
         output_img = gr.Image(