-
Notifications
You must be signed in to change notification settings - Fork 5
/
remove-background.js
69 lines (55 loc) · 2.24 KB
/
remove-background.js
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
//
// Copyright © 2023 Artizans. All rights reserved.
//
const https = require('https');
const fs = require('fs');
console.error('Please replace with your own apiKey');
const apiKey = 'YOUR_API_KEY_HERE';
function removeBackground(imagePath, savePath) {
return new Promise((resolve, reject) => {
const boundary = '--------------------------' + Date.now().toString(16);
const postOptions = {
hostname: 'sdk.photoroom.com',
path: '/v1/segment',
method: 'POST',
headers: {
'Content-Type': `multipart/form-data; boundary=${boundary}`,
'X-API-Key': apiKey
}
};
const req = https.request(postOptions, (res) => {
// Check if the response is an image
const isImage = ['image/jpeg', 'image/png', 'image/gif'].includes(res.headers['content-type']);
if (!isImage) {
let errorData = '';
res.on('data', (chunk) => errorData += chunk);
res.on('end', () => reject(new Error(`Expected an image response, but received: ${errorData}`)));
return;
}
// Create a write stream to save the image
const fileStream = fs.createWriteStream(savePath);
res.pipe(fileStream);
fileStream.on('finish', () => {
resolve(`Image saved to ${savePath}`);
});
fileStream.on('error', (error) => {
reject(new Error(`Failed to save the image: ${error.message}`));
});
});
req.on('error', (error) => {
reject(error);
});
// Write form data
req.write(`--${boundary}\r\n`);
req.write(`Content-Disposition: form-data; name="image_file"; filename="${imagePath.split('/').pop()}"\r\n`);
req.write('Content-Type: image/jpeg\r\n\r\n'); // assuming JPEG, adjust if another format is used
const uploadStream = fs.createReadStream(imagePath);
uploadStream.on('end', () => {
req.write('\r\n');
req.write(`--${boundary}--\r\n`);
req.end();
});
uploadStream.pipe(req, { end: false });
});
}
module.exports = removeBackground;