index.js

import path from "path";
import { fileURLToPath } from "url";
import express from "express";
import multer from "multer";
import ImageClassification from "@nanonets/image-classification";
import dotenv from "dotenv";

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

// Make "__dirname" available in ES module
const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

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

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

// Class instantiation
const ic = new ImageClassification(
	process.env.API_KEY_1,
	process.env.MODEL_ID_1
);

// Routes
app.get("/", async (req, res) => {
	// const icUrlArray = [process.env.FILE_URL_1, process.env.FILE_URL_2];

	// 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:");

	// 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}...`);
});