from flask import Flask, jsonify, request, send_file
from flask_cors import CORS
from rembg import remove
from PIL import Image
import io

app = Flask(__name__)
CORS(app)  # Enable CORS for all routes

@app.route('/remove_background', methods=['POST'])
def remove_background():
    if 'image' not in request.files:
        return jsonify({"error": "No image provided"}), 400

    input_image = request.files['image'].read()  # Read the image file

    # Apply background removal using rembg
    output_bytes = remove(input_image)

    # Convert the output bytes back into a PIL image
    output_image = Image.open(io.BytesIO(output_bytes))

    # Save the processed image to a BytesIO object
    img_byte_arr = io.BytesIO()
    output_image.save(img_byte_arr, format='PNG')
    img_byte_arr.seek(0)

    # Return the image as a response
    return send_file(img_byte_arr, mimetype='image/png')

if __name__ == "__main__":
    app.run(host='0.0.0.0', port=7860)