mindee/src/v2/cli.js

Mindee Client Library for Node.js

import { Command } from "commander";
import * as console from "console";
import { Client, PathInput, UrlInput } from "../index.js";
import { Classification, Crop, Extraction, Ocr, Split, } from "../v2/product/index.js";
const program = new Command();
//
// EXECUTE THE COMMANDS
//
function initClient(options) {
    return new Client({
        apiKey: options.apiKey,
        debug: options.debug,
    });
}
async function enqueueAndGetInference(product, inputPath, options) {
    const mindeeClient = initClient(options);
    let inputSource;
    if (inputPath.startsWith("https://")) {
        inputSource = new UrlInput({ url: inputPath });
    }
    else {
        inputSource = new PathInput({ inputPath: inputPath });
    }
    const response = await mindeeClient.enqueueAndGetResult(product, inputSource, { modelId: options.model }, {
        initialDelaySec: 2,
        delaySec: 1.5,
        maxRetries: 80,
    });
    if (!response.inference) {
        throw Error("Inference could not be retrieved");
    }
    printResponse(response.inference);
}
function printResponse(document) {
    if (document) {
        console.log(`\n${document}`);
    }
}
//
// BUILD THE COMMANDS
//
function addMainOptions(prog) {
    prog.requiredOption("-m, --model <model_id>", "Model ID (required)");
    prog.argument("<input_path>", "full path or URL to the file");
}
export function cli() {
    program.name("mindee")
        .description("Command line interface for Mindee V2 products.")
        .option("-d, --debug", "high verbosity mode")
        .option("-k, --api-key <api_key>", "your Mindee API key");
    const inferenceTypes = [
        { name: "extraction", description: "Extract data from a document.", productClass: Extraction },
        { name: "crop", description: "Crop a document.", productClass: Crop },
        { name: "split", description: "Split a document into pages.", productClass: Split },
        { name: "ocr", description: "Read text from a document.", productClass: Ocr },
        { name: "classification", description: "Classify a document.", productClass: Classification },
    ];
    for (const inference of inferenceTypes) {
        const inferenceCmd = program.command(inference.name)
            .description(inference.description);
        addMainOptions(inferenceCmd);
        inferenceCmd.action(function (inputPath, options) {
            const allOptions = { ...program.opts(), ...options };
            return enqueueAndGetInference(inference.productClass, inputPath, allOptions);
        });
    }
    program.parse(process.argv);
}