from rembg import remove, new_session
from PIL import Image
import io

# Best results ke liye 'u2net_human_seg' try karein
session = new_session("u2net_human_seg") 

def process_image(image_bytes: bytes):
    input_image = Image.open(io.BytesIO(image_bytes))
    
    # Background remove karein
    # alpha_matting=True edges ko sharp rakhega
    output_image = remove(input_image, session=session, alpha_matting=True)
    
    img_byte_arr = io.BytesIO()
    output_image.save(img_byte_arr, format='PNG')
    return img_byte_arr.getvalue()