Update app.py

app.py

@@ -3,41 +3,47 @@ import numpy as np
 from PIL import Image
 from rembg import remove, new_session
 
-# Initialize session with proper settings to prevent cropping
+# Initialize model with NO 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(input_image):
+def remove_background(img):
     try:
-        # 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
+        # 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"])
         
-        # Preserve original size (disable auto-resizing)
-        result = remove(img, **bg_removal_kwargs)
+        # 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)
+            
         return result
     
     except Exception as e:
-        print(f"Error: {str(e)}")
-        return input_image  # Return original if fails
+        print(f"Error: {e}")
+        return img  # Return original if fails
 
-# Gradio interface with proper image handling
+# Gradio interface with size preservation
 with gr.Blocks() as demo:
-    gr.Markdown("## 🖼️ Background Remover (No Cropping)")
+    gr.Markdown("## 📸 Background Remover (Original Size)")
     
     with gr.Row():
         input_img = gr.Image(
-            label="Original", 
-            type="pil",  # Ensures we get PIL Images
+            label="Original",
+            type="filepath",  # Better for size preservation
             height=400
         )
         output_img = gr.Image(