@nanonets/optical-character-recognition/commonjs_example/index.js

NanoNets' Optical Character Recognition Node.js SDK.

const express = require("express");
const multer = require("multer");
const OpticalCharacterRecognition = 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}...`);
});