index.js

const express = require("express");
const multer = require("multer");
const cors = require("cors");
const fs = require("fs");
const path = require("path");
const { removeBackground } = require("@imgly/background-removal-node");

const app = express();
const port = process.env.PORT || 3000;

// Enable CORS
app.use(cors());

// Configure multer for memory storage
const storage = multer.memoryStorage();
const upload = multer({ storage: storage });

// Create uploads directory if it doesn't exist
const uploadsDir = path.join(__dirname, "uploads");
if (!fs.existsSync(uploadsDir)) {
  fs.mkdirSync(uploadsDir, { recursive: true });
}

// Serve static files
app.use(express.static("public"));

// Simple home route
app.get("/", (req, res) => {
  res.send(
    "Background Removal API - Use POST /remove-background to process images"
  );
});

// Background removal endpoint
app.post("/remove-background", upload.single("image"), async (req, res) => {
  try {
    if (!req.file) {
      return res.status(400).json({ error: "No image file uploaded" });
    }

    // Save the uploaded file temporarily
    const inputPath = path.join(uploadsDir, `input-${Date.now()}.png`);
    fs.writeFileSync(inputPath, req.file.buffer);

    // Process the image with background removal
    const outputBlob = await removeBackground(inputPath);

    // Convert Blob to Buffer
    const arrayBuffer = await outputBlob.arrayBuffer();
    const outputBuffer = Buffer.from(arrayBuffer);

    // Convert Buffer to base64 and add data URI prefix
    const base64Image = `data:image/png;base64,${outputBuffer.toString(
      "base64"
    )}`;

    // Clean up temporary file
    fs.unlinkSync(inputPath);

    // Return the base64 string
    res.json({ base64Image });
  } catch (error) {
    console.error("Error processing image:", error);
    res.status(500).json({ error: "Failed to process image" });
  }
});

app.listen(port, () => {
  console.log(`Server running at http://localhost:${port}`);
});