43 lines
1.2 KiB
JavaScript
43 lines
1.2 KiB
JavaScript
// server.js
|
|
const express = require("express");
|
|
const { spawn } = require("child_process");
|
|
const path = require("path");
|
|
const upload = require("./Middleware/multer");
|
|
|
|
const app = express();
|
|
const PORT = 3000;
|
|
|
|
// Endpoint to run the Python script
|
|
app.post("/predict", upload.single("image"), (req, res) => {
|
|
console.log("File is uploaded successfully" + req.file?.path);
|
|
// const imagePath = path.join(__dirname, req.file?.path); // Path to your image
|
|
|
|
const imagePath = req.file?.path;
|
|
|
|
console.log("Image Path is : ", imagePath);
|
|
|
|
// Spawn a new Python process
|
|
const pythonProcess = spawn("python", ["app.py", imagePath]);
|
|
|
|
// Collect data from the script
|
|
pythonProcess.stdout.on("data", (data) => {
|
|
console.log(`Output: ${data}`);
|
|
res.send(data.toString()); // Send the output back to the client
|
|
});
|
|
|
|
// Handle errors
|
|
pythonProcess.stderr.on("data", (data) => {
|
|
console.error(`Error: ${data}`);
|
|
res.status(500).send(data.toString());
|
|
});
|
|
|
|
// When the process is done
|
|
pythonProcess.on("close", (code) => {
|
|
console.log(`Python process exited with code ${code}`);
|
|
});
|
|
});
|
|
|
|
app.listen(PORT, () => {
|
|
console.log(`Server is running on http://localhost:${PORT}`);
|
|
});
|