RageStudio/comty
1
0
Fork 0
mirror of https://github.com/ragestudio/comty.git synced 2025-06-13 20:44:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

handle error on flagNsfwByAttachments

Browse Source
This commit is contained in:
SrGooglo 2023-07-20 13:25:42 +00:00
parent 0f96567a11
commit 7343f5df83
1 changed files with 41 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

77
packages/server/src/utils/indecent-prediction/index.js
View File

@ -23,46 +23,51 @@ if (global.isProduction) {
} }
export default async (payload) => { export default async (payload) => {
let { url, image, channels = 3 } = payload try {
let { url, image, channels = 3 } = payload
let file = null let file = null
const model = await nsfwjs.load() const model = await nsfwjs.load()
if (!image && url) { if (!image && url) {
file = await downloadFile({ url }) file = await downloadFile({ url })
image = file.destination image = file.destination
}
// check if image is not a jpg
if (image.indexOf(".jpg") === -1) {
// convert image to jpg
const converted = await sharp(image)
.jpeg()
.toBuffer()
// write converted image to disk (use cache)
const destination = path.resolve(global.uploadCachePath, `${Date.now()}.jpg`)
fs.writeFileSync(destination, converted)
// set image to the converted image
file = {
destination,
delete: () => fs.unlinkSync(destination),
} }
image = destination // check if image is not a jpg
if (image.indexOf(".jpg") === -1) {
// convert image to jpg
const converted = await sharp(image)
.jpeg()
.toBuffer()
// write converted image to disk (use cache)
const destination = path.resolve(global.uploadCachePath, `${Date.now()}.jpg`)
fs.writeFileSync(destination, converted)
// set image to the converted image
file = {
destination,
delete: () => fs.unlinkSync(destination),
}
image = destination
}
const logo = readImage(image)
const input = imageToInput(logo, channels)
const predictions = await model.classify(input)
if (typeof file.delete === "function") {
await file.delete()
}
return predictions
} catch (error) {
console.error(`Failed to process image >`, error)
console.trace()
} }
const logo = readImage(image)
const input = imageToInput(logo, channels)
const predictions = await model.classify(input)
if (typeof file.delete === "function") {
await file.delete()
}
return predictions
} }