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)}"