@nanonets/image-classification/commonjs_example/index.js

NanoNets' Image Classification Node.js SDK.

const express = require("express");
const multer = require("multer");
const ImageClassification = require("../index.cjs");
require("dotenv").config();

// Server initialization
const app = express();
const upload = multer({ dest: "public/uploads" });

// Static files
app.use("/public", express.static(__dirname + "/public"));

// Body Parser
app.use(express.urlencoded({ extended: false }));
app.use(express.json());

// Class instantiations
// const ocr = new OpticalCharacterRecognition(
// 	process.env.API_KEY_1,
// 	process.env.MODEL_ID_1
// );
const ic = new ImageClassification(
	process.env.API_KEY_1,
	process.env.MODEL_ID_2
);

// Routes
app.get("/", async (req, res) => {
	// const ocrUrlArray = [process.env.FILE_URL_1, process.env.FILE_URL_2];
	// const icUrlArray = [process.env.FILE_URL_3, process.env.FILE_URL_4];
	// const startInterval = 18917;
	// const endInterval = 18919;

	// Optical Character Recognition (OCR) functions

	// console.log(
	// 	await ocr.getModelDetails(),
	// 	"\n-------------------------------------------------------\n"
	// );
	// console.log(
	// 	await ocr.getAllPredictedFileData(startInterval, endInterval),
	// 	"\n-------------------------------------------------------\n"
	// );
	// console.log(
	// 	await ocr.getPredictedFileData(process.env.FILE_ID),
	// 	"\n-------------------------------------------------------\n"
	// );
	// console.log(
	// 	await ocr.predictUsingUrls(ocrUrlArray, false),
	// 	"\n-------------------------------------------------------\n"
	// );
	// console.log(
	// 	await ocr.predictUsingUrls(ocrUrlArray, true),
	// 	"\n-------------------------------------------------------\n"
	// );

	// Image Classification (IC) functions

	console.log(
		await ic.getModelDetails(),
		"\n-------------------------------------------------------\n"
	);
	// console.log(
	// 	await ic.predictUsingUrls(icUrlArray),
	// 	"\n-------------------------------------------------------\n"
	// );

	res.sendFile(__dirname + "/public/index.html");
});

app.get("/uploadFile", (req, res) => {
	res.sendFile(__dirname + "/public/fileUpload.html");
});

app.post("/uploadFile", upload.single("file"), async (req, res) => {
	console.log("Uploaded file details: ", req.file, "\n\nResponse:");

	// if (req.body.operation === "ocr")
	// 	console.log(
	// 		await ocr.predictUsingFile(req.file.path, false),
	// 		"\n-------------------------------------------------------\n"
	// 	);
	// else if (req.body.operation === "ocr-async")
	// 	console.log(
	// 		await ocr.predictUsingFile(req.file.path, true),
	// 		"\n-------------------------------------------------------\n"
	// 	);
	// else if (req.body.operation === "ic")
	console.log(
		await ic.predictUsingFile(req.file.path),
		"\n-------------------------------------------------------\n"
	);

	res.send("Uploaded. Check the server console for the response.");
});

// Server port
const PORT = process.env.PORT || 5000;
app.listen(PORT, (err) => {
	if (err) console.error("Node.js server error: ", err);
	else console.log(`Server started on port ${PORT}...`);
});