File size: 1,798 Bytes
fc8bf44 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
import requests
import logging
from PIL import Image
import io
from pathlib import Path
class BackgroundRemover:
def __init__(self, removebg_key=None):
self.removebg_key = removebg_key
def remove_local(self, input_path):
from rembg import remove
try:
img = Image.open(input_path)
result = remove(img)
out_path = Path(input_path).with_name(f"nobg_{Path(input_path).name}")
result.save(out_path)
return str(out_path), "✅ Background removed locally"
except Exception as e:
logging.exception("Local background removal failed")
return None, f"❌ Background removal error: {str(e)}"
def remove_with_removebg(self, input_path):
if not self.removebg_key:
return None, "❌ Remove.bg API key not provided"
try:
with open(input_path, 'rb') as img_file:
response = requests.post(
'https://api.remove.bg/v1.0/removebg',
files={'image_file': img_file},
data={'size': 'auto'},
headers={'X-Api-Key': self.removebg_key},
timeout=30
)
if response.status_code == 200:
out_path = Path(input_path).with_name(f"removebg_{Path(input_path).name}")
with open(out_path, 'wb') as out_file:
out_file.write(response.content)
return str(out_path), "✅ Background removed with Remove.bg"
else:
return None, f"❌ Remove.bg API Error: {response.text}"
except Exception as e:
logging.exception("Remove.bg API background removal failed")
return None, f"❌ Remove.bg error: {str(e)}"
|