Support quality parameter (Standard/HD/Ultra) with zero PNG compression for maximum quality

app.py

@@ -58,8 +58,9 @@ def health_check():
 @app.route('/remove-background', methods=['POST'])
 def remove_background():
     """
-    Background removal using rembg with u2netp model
+    Background removal using rembg with u2net model
     NO external APIs - 100% self-hosted!
+    Supports quality options: standard, hd, ultra
     """
     try:
         # Get uploaded file
@@ -67,10 +68,31 @@ def remove_background():
             return {'error': 'No image file provided'}, 400
         
         file = request.files['image']
-        input_image = Image.open(file.stream)
+        input_image = Image.open(file.stream).convert('RGBA')
+        
+        # Get quality setting (default: ultra)
+        quality = request.form.get('quality', 'ultra').lower()
         
         print(f"🎨 Processing image: {file.filename}")
-        print(f"📐 Size: {input_image.size}")
+        print(f"📐 Original size: {input_image.size}")
+        print(f"🎯 Quality mode: {quality}")
+        
+        # Resize based on quality (preserve aspect ratio)
+        max_dimensions = {
+            'standard': 1920,  # Full HD
+            'hd': 2560,        # 2K
+            'ultra': 3840,     # 4K
+            '4k': 3840         # 4K alias
+        }
+        
+        max_size = max_dimensions.get(quality, 3840)  # Default to 4K
+        
+        # Only resize if image is larger than max
+        if max(input_image.size) > max_size:
+            ratio = max_size / max(input_image.size)
+            new_size = tuple(int(dim * ratio) for dim in input_image.size)
+            input_image = input_image.resize(new_size, Image.Resampling.LANCZOS)
+            print(f"📐 Resized to: {input_image.size}")
         
         # Get session and remove background
         session = get_session()
@@ -78,11 +100,19 @@ def remove_background():
         
         print(f"✅ Background removed successfully!")
         
-        # Convert to PNG bytes
+        # Save with MAXIMUM quality - no compression
         img_io = io.BytesIO()
-        output_image.save(img_io, 'PNG', optimize=False)
+        output_image.save(
+            img_io, 
+            'PNG',
+            compress_level=0,  # No compression (0-9, 0=no compression)
+            optimize=False     # Don't optimize file size
+        )
         img_io.seek(0)
         
+        output_size_mb = len(img_io.getvalue()) / (1024 * 1024)
+        print(f"📦 Output size: {output_size_mb:.2f} MB")
+        
         return send_file(
             img_io,
             mimetype='image/png',