wrapture 0.0.2 → 0.1.0

package/.github/workflows/check.yml

@@ -48,6 +48,13 @@ jobs:
         uses: actions/setup-node@v4
         with:
           node-version: '>=22.8.0'
+      - uses: actions/setup-python@v4
+        with:
+          python-version: '3.10'
+      - name: Install Python dependencies
+        run: |
+          python3 -m pip install --upgrade pip
+          pip install torch onnx onnxsim onnxruntime
       - name: Install dependencies
         run: | # Install and link dependencies
           npm i
@@ -69,4 +76,4 @@ jobs:
       - uses: jayqi/failed-build-issue-action@v1.2
         with:
           github-token: ${{ secrets.GH_TOKEN }}
-          label-name: 'problems: build failed'
+          label-name: 'problems: 💔 build failed'

package/CHANGELOG.md

@@ -1,5 +1,33 @@
 # Changelog
 
+## [0.1.0](https://github.com/phun-ky/wrapture/compare/0.0.3...0.1.0) (2025-05-15)
+
+### Tasks
+
+* 🤖 add `bin/` to `.gitignore` ([a90b288](https://github.com/phun-ky/wrapture/commit/a90b2880a6179010b6133f719eb7696759e6ac06))
+* 🤖 Delete file from bin ([b707abd](https://github.com/phun-ky/wrapture/commit/b707abddd989c43da70bf91512d75e752c18456c))
+* 🤖 Remove old dep imports ([819e9e4](https://github.com/phun-ky/wrapture/commit/819e9e439f0bbf50383038db2cded3d3c167e3f3))
+* 🤖 Use correct label ([3d6acf0](https://github.com/phun-ky/wrapture/commit/3d6acf0311f18e9e2d90b4a129f0dc7cae9b955c))
+
+### Feature
+
+* 🎸 Add `--logLevel`-option and clean up messages ([c93f4ef](https://github.com/phun-ky/wrapture/commit/c93f4ef0570d4fe7501776f2fbce1ed362e048a7))
+* 🎸 Check for correct dependencies ([28d1d4d](https://github.com/phun-ky/wrapture/commit/28d1d4def395c700285caf82139e8b0786cc4855))
+
+### Bug
+
+* 🐛 Remove outtdated python module ([0b58d3c](https://github.com/phun-ky/wrapture/commit/0b58d3cc5c2758b606333e011766a3ed91fbfa74))
+
+### Refactoring
+
+* 💡 Adjust cli interface ([ddc80b9](https://github.com/phun-ky/wrapture/commit/ddc80b9d96f11133c2aa4e26ffcde40f316f12a8))
+
+## [0.0.3](https://github.com/phun-ky/wrapture/compare/0.0.2...0.0.3) (2025-05-14)
+
+### Bug
+
+* 🐛 Python env ([36d0a30](https://github.com/phun-ky/wrapture/commit/36d0a302aebfabf9860f7cc5c70bc8151b221632))
+
 ## [0.0.2](https://github.com/phun-ky/wrapture/compare/0.0.1...0.0.2) (2025-05-14)
 
 ### Documentation

package/README.md

@@ -62,7 +62,7 @@ Install Python if you don’t have it: 👉 <https://www.python.org/downloads/>
 ### Install required Python packages
 
 ```shell-session
-python3 -m pip install torch onnx onnxsim onnxruntime onnxruntime-tools
+python3 -m pip install torch onnx onnxsim onnxruntime
 ```
 
 Check your installation:

package/api/README.md

@@ -1,11 +1,13 @@
 # wrapture API documentation
 
-> Last updated 2025-05-14T13:10:23.461Z
+> Last updated 2025-05-15T10:00:06.699Z
 
 ## Modules
 
+- [utils/check-deps](utils/check-deps.md)
 - [utils/convert](utils/convert.md)
 - [utils/generate-wrapper](utils/generate-wrapper.md)
+- [utils/log-level](utils/log-level.md)
 - [wrapture](wrapture.md)
 
 ---

package/api/utils/check-deps.md

@@ -0,0 +1,61 @@
+[wrapture](../README.md) / utils/check-deps
+
+# utils/check-deps
+
+> Last updated 2025-05-15T10:00:06.852Z
+
+## Functions
+
+### checkPythonAvailable()
+
+```ts
+function checkPythonAvailable(): void;
+```
+
+Defined in:
+[utils/check-deps.ts:11](https://github.com/phun-ky/wrapture/blob/main/src/utils/check-deps.ts#L11)
+
+#### Returns
+
+`void`
+
+---
+
+### checkPythonDeps()
+
+```ts
+function checkPythonDeps(): void;
+```
+
+Defined in:
+[utils/check-deps.ts:23](https://github.com/phun-ky/wrapture/blob/main/src/utils/check-deps.ts#L23)
+
+#### Returns
+
+`void`
+
+---
+
+**Contributing**
+
+Want to contribute? Please read the
+[CONTRIBUTING.md](https://github.com/phun-ky/wrapture/blob/main/CONTRIBUTING.md)
+and
+[CODE_OF_CONDUCT.md](https://github.com/phun-ky/wrapture/blob/main/CODE_OF_CONDUCT.md)
+
+**Sponsor me**
+
+I'm an Open Source evangelist, creating stuff that does not exist yet to help
+get rid of secondary activities and to enhance systems already in place, be it
+documentation or web sites.
+
+The sponsorship is an unique opportunity to alleviate more hours for me to
+maintain my projects, create new ones and contribute to the large community
+we're all part of :)
+
+[Support me on GitHub Sponsors](https://github.com/sponsors/phun-ky).
+
+---
+
+This project created by [Alexander Vassbotn Røyne-Helgesen](http://phun-ky.net)
+is licensed under a [MIT License](https://choosealicense.com/licenses/mit/).

package/api/utils/convert.md

@@ -2,14 +2,14 @@
 
 # utils/convert
 
-> Last updated 2025-05-14T13:10:23.623Z
+> Last updated 2025-05-15T10:00:06.925Z
 
 ## Interfaces
 
 ### ConvertOptionsInterface
 
 Defined in:
-[utils/convert.ts:17](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L17)
+[utils/convert.ts:12](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L12)
 
 Options for the [convert](#convert) function.
 
@@ -17,8 +17,8 @@ Options for the [convert](#convert) function.
 
 | Property                          | Type      | Description                                                                                   | Defined in                                                                                    |
 | --------------------------------- | --------- | --------------------------------------------------------------------------------------------- | --------------------------------------------------------------------------------------------- |
-| <a id="format"></a> `format?`     | `string`  | The output format for the converted model (e.g., 'onnx'). Defaults to 'onnx' if not provided. | [utils/convert.ts:22](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L22) |
-| <a id="quantize"></a> `quantize?` | `boolean` | Whether to apply quantization to the model.                                                   | [utils/convert.ts:27](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L27) |
+| <a id="format"></a> `format?`     | `string`  | The output format for the converted model (e.g., 'onnx'). Defaults to 'onnx' if not provided. | [utils/convert.ts:17](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L17) |
+| <a id="quantize"></a> `quantize?` | `boolean` | Whether to apply quantization to the model.                                                   | [utils/convert.ts:22](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L22) |
 
 ## Functions
 
@@ -29,7 +29,7 @@ function convert(inputPath, outputDir, opts): Promise<void>;
 ```
 
 Defined in:
-[utils/convert.ts:54](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L54)
+[utils/convert.ts:49](https://github.com/phun-ky/wrapture/blob/main/src/utils/convert.ts#L49)
 
 Converts a machine learning model to ONNX or another supported format by
 delegating to a Python script (`convert.py`).

package/api/utils/generate-wrapper.md

@@ -2,14 +2,14 @@
 
 # utils/generate-wrapper
 
-> Last updated 2025-05-14T13:10:23.726Z
+> Last updated 2025-05-15T10:00:06.977Z
 
 ## Interfaces
 
 ### GenerateWrapperOptionsInterface
 
 Defined in:
-[utils/generate-wrapper.ts:9](https://github.com/phun-ky/wrapture/blob/main/src/utils/generate-wrapper.ts#L9)
+[utils/generate-wrapper.ts:10](https://github.com/phun-ky/wrapture/blob/main/src/utils/generate-wrapper.ts#L10)
 
 Options for generating ONNX wrapper files.
 
@@ -17,7 +17,7 @@ Options for generating ONNX wrapper files.
 
 | Property                       | Type     | Description                                                                                                                                                                      | Defined in                                                                                                      |
 | ------------------------------ | -------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | --------------------------------------------------------------------------------------------------------------- |
-| <a id="backend"></a> `backend` | `string` | The backend to use for inference. This affects the model file used. If set to `'wasm'`, the generated wrapper will load `model_quant.onnx`, otherwise it will load `model.onnx`. | [utils/generate-wrapper.ts:15](https://github.com/phun-ky/wrapture/blob/main/src/utils/generate-wrapper.ts#L15) |
+| <a id="backend"></a> `backend` | `string` | The backend to use for inference. This affects the model file used. If set to `'wasm'`, the generated wrapper will load `model_quant.onnx`, otherwise it will load `model.onnx`. | [utils/generate-wrapper.ts:16](https://github.com/phun-ky/wrapture/blob/main/src/utils/generate-wrapper.ts#L16) |
 
 ## Functions
 
@@ -28,7 +28,7 @@ function generateWrapper(outputDir, opts): Promise<void>;
 ```
 
 Defined in:
-[utils/generate-wrapper.ts:39](https://github.com/phun-ky/wrapture/blob/main/src/utils/generate-wrapper.ts#L39)
+[utils/generate-wrapper.ts:40](https://github.com/phun-ky/wrapture/blob/main/src/utils/generate-wrapper.ts#L40)
 
 Generates a TypeScript wrapper and type definition file (`wrapped.ts` and
 `wrapped.d.ts`) for use with `onnxruntime-web`, including utility functions like

package/api/utils/log-level.md

@@ -0,0 +1,161 @@
+[wrapture](../README.md) / utils/log-level
+
+# utils/log-level
+
+> Last updated 2025-05-15T10:00:07.021Z
+
+## Type Aliases
+
+### LogLevelType
+
+```ts
+type LogLevelType = 'silent' | 'error' | 'warn' | 'info' | 'debug';
+```
+
+Defined in:
+[utils/log-level.ts:2](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L2)
+
+## Variables
+
+### log
+
+```ts
+const log: {
+  debug: (...args) => false | void;
+  error: (...args) => false | void;
+  info: (...args) => false | void;
+  warn: (...args) => false | void;
+};
+```
+
+Defined in:
+[utils/log-level.ts:22](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L22)
+
+#### Type declaration
+
+<table>
+<thead>
+<tr>
+<th>Name</th>
+<th>Type</th>
+<th>Defined in</th>
+</tr>
+</thead>
+<tbody>
+<tr>
+<td>
+
+<a id="debug"></a> `debug()`
+
+</td>
+<td>
+
+(...`args`) => `false` | `void`
+
+</td>
+<td>
+
+[utils/log-level.ts:23](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L23)
+
+</td>
+</tr>
+<tr>
+<td>
+
+<a id="error"></a> `error()`
+
+</td>
+<td>
+
+(...`args`) => `false` | `void`
+
+</td>
+<td>
+
+[utils/log-level.ts:29](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L29)
+
+</td>
+</tr>
+<tr>
+<td>
+
+<a id="info"></a> `info()`
+
+</td>
+<td>
+
+(...`args`) => `false` | `void`
+
+</td>
+<td>
+
+[utils/log-level.ts:25](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L25)
+
+</td>
+</tr>
+<tr>
+<td>
+
+<a id="warn"></a> `warn()`
+
+</td>
+<td>
+
+(...`args`) => `false` | `void`
+
+</td>
+<td>
+
+[utils/log-level.ts:27](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L27)
+
+</td>
+</tr>
+</tbody>
+</table>
+
+## Functions
+
+### setLogLevel()
+
+```ts
+function setLogLevel(level): void;
+```
+
+Defined in:
+[utils/log-level.ts:14](https://github.com/phun-ky/wrapture/blob/main/src/utils/log-level.ts#L14)
+
+#### Parameters
+
+| Parameter | Type                            |
+| --------- | ------------------------------- |
+| `level`   | [`LogLevelType`](#logleveltype) |
+
+#### Returns
+
+`void`
+
+---
+
+**Contributing**
+
+Want to contribute? Please read the
+[CONTRIBUTING.md](https://github.com/phun-ky/wrapture/blob/main/CONTRIBUTING.md)
+and
+[CODE_OF_CONDUCT.md](https://github.com/phun-ky/wrapture/blob/main/CODE_OF_CONDUCT.md)
+
+**Sponsor me**
+
+I'm an Open Source evangelist, creating stuff that does not exist yet to help
+get rid of secondary activities and to enhance systems already in place, be it
+documentation or web sites.
+
+The sponsorship is an unique opportunity to alleviate more hours for me to
+maintain my projects, create new ones and contribute to the large community
+we're all part of :)
+
+[Support me on GitHub Sponsors](https://github.com/sponsors/phun-ky).
+
+---
+
+This project created by [Alexander Vassbotn Røyne-Helgesen](http://phun-ky.net)
+is licensed under a [MIT License](https://choosealicense.com/licenses/mit/).

package/api/wrapture.md

@@ -2,7 +2,7 @@
 
 # wrapture
 
-> Last updated 2025-05-14T13:10:23.760Z
+> Last updated 2025-05-15T10:00:07.077Z
 
 ---
 

package/bin/wrapture.js

@@ -3,25 +3,197 @@
 // src/wrapture.ts
 import chalk2 from "chalk";
 import { Command } from "commander";
+import ora from "ora";
 import { existsSync } from "node:fs";
 import path3 from "node:path";
 
-// src/utils/convert.ts
+// package.json
+var package_default = {
+  name: "wrapture",
+  version: "0.1.0",
+  description: "Wrapture lets you go from a Python-trained model to deployable JavaScript with a single command. It generates TypeScript bindings and a Web/Node-compatible wrapper, using WebGPU/WASM-ready ONNX runtimes.",
+  keywords: [
+    "onnx",
+    "pytorch",
+    "torch",
+    "model exporter",
+    "onnx export",
+    "onnx runtime",
+    "onnx to javascript",
+    "machine learning",
+    "ml model conversion",
+    "typescript",
+    "javascript",
+    "cli tool",
+    "webgpu",
+    "wasm",
+    "onnx simplifier",
+    "onnx quantization",
+    "deep learning",
+    "model conversion",
+    "wrapture",
+    "loadModel",
+    "predict"
+  ],
+  homepage: "https://github.com/phun-ky/wrapture#readme",
+  bugs: {
+    url: "https://github.com/phun-ky/wrapture/issues"
+  },
+  repository: {
+    type: "git",
+    url: "git+https://github.com/phun-ky/wrapture.git"
+  },
+  funding: "https://github.com/phun-ky/angle?sponsor=1",
+  license: "MIT",
+  author: "Alexander Vassbotn R\xF8yne-Helgesen <alexander@phun-ky.net>",
+  type: "module",
+  bin: {
+    wrapture: "./bin/wrapture.js"
+  },
+  scripts: {
+    build: "tsup",
+    commit: "npx git-cz",
+    "docs:gen": "node ./node_modules/.bin/typedoc",
+    release: "release-it",
+    start: "node ./bin/wrapture.js",
+    "style:code": "npx putout src",
+    "style:format": "./node_modules/.bin/eslint -c ./eslint.config.mjs src  --fix && ./node_modules/.bin/prettier --write ./eslint.config.mjs src",
+    "style:lint": "./node_modules/.bin/eslint -c ./eslint.config.mjs src && ./node_modules/.bin/prettier --check src",
+    test: 'NODE_ENV=test glob -c "node --import tsx --test --no-warnings" "./src/**/__tests__/**/*.[jt]s"',
+    "pretest:ci": "rm -rf coverage && mkdir -p coverage",
+    "test:ci": 'NODE_ENV=test glob -c "node --import tsx --test --no-warnings --experimental-test-coverage --test-reporter=cobertura --test-reporter-destination=coverage/cobertura-coverage.xml --test-reporter=spec --test-reporter-destination=stdout" "./src/**/__tests__/**/*.[jt]s"'
+  },
+  dependencies: {
+    chalk: "^5.4.1",
+    commander: "^13.1.0",
+    ora: "^8.2.0"
+  },
+  devDependencies: {
+    "@release-it/conventional-changelog": "^10.0.0",
+    "@rollup/plugin-node-resolve": "^16.0.1",
+    "@rollup/plugin-terser": "^0.4.4",
+    "@stylistic/eslint-plugin": "^4.2.0",
+    "@types/node": "^22.15.3",
+    cobertura: "^1.0.1",
+    eslint: "^9.20.0",
+    "eslint-config-phun-ky": "^1.0.0",
+    "git-cz": "^4.9.0",
+    "onnxruntime-web": "^1.22.0",
+    prettier: "^3.2.5",
+    putout: "^40.1.9",
+    "release-it": "^19.0.1",
+    "remark-github": "^12.0.0",
+    "remark-toc": "^9.0.0",
+    tslib: "^2.3.1",
+    tsup: "^8.4.0",
+    tsx: "^4.7.1",
+    typedoc: "^0.28.3",
+    "typedoc-plugin-frontmatter": "^1.0.0",
+    "typedoc-plugin-markdown": "^4.2.3",
+    "typedoc-plugin-mdn-links": "^5.0.1",
+    "typedoc-plugin-no-inherit": "^1.4.0",
+    "typedoc-plugin-remark": "^2.0.0",
+    "typedoc-plugin-rename-defaults": "^0.7.1",
+    typescript: "^5.0.0",
+    "unified-prettier": "^2.0.1"
+  },
+  engines: {
+    node: ">=22.0.0",
+    npm: ">=10.8.2"
+  },
+  publishConfig: {
+    access: "public"
+  }
+};
+
+// src/utils/check-deps.ts
 import chalk from "chalk";
-import ora from "ora";
+import { spawnSync } from "node:child_process";
+var printError = (title, body) => {
+  console.error(`
+${chalk.red.bold("\u2718")} ${chalk.red.bold(title)}`);
+  console.error(chalk.white(body));
+};
+var checkPythonAvailable = () => {
+  const result = spawnSync("python3", ["--version"], { encoding: "utf-8" });
+  if (result.error || result.status !== 0) {
+    printError(
+      "Python 3 is not available.",
+      "Please install it from https://www.python.org/downloads/ and ensure it is added to your PATH."
+    );
+    process.exit(1);
+  }
+};
+var checkPythonDeps = () => {
+  const check = spawnSync(
+    "python3",
+    [
+      "-c",
+      `
+import sys
+missing = []
+for module in ['torch', 'onnx', 'onnxsim', 'onnxruntime']:
+    try:
+        __import__(module)
+    except ImportError:
+        missing.append(module)
+if missing:
+    print(', '.join(missing))
+    sys.exit(1)
+  `
+    ],
+    { encoding: "utf-8" }
+  );
+  if (check.status !== 0) {
+    const missing = check.stdout.trim().split(",").filter(Boolean);
+    printError(
+      "Missing Python dependencies.",
+      `Please install the following packages:
+
+  ${chalk.yellow(
+        `python3 -m pip install ${missing.join(" ")}`
+      )}`
+    );
+    process.exit(1);
+  }
+};
+
+// src/utils/convert.ts
 import { spawn } from "node:child_process";
 import fs from "node:fs";
-import path, { dirname } from "node:path";
-import { fileURLToPath } from "node:url";
-var __filename = fileURLToPath(import.meta.url);
-var __dirname = dirname(__filename);
+import path from "node:path";
+
+// src/utils/log-level.ts
+var currentLevel = "info";
+var levels = {
+  silent: 0,
+  error: 1,
+  warn: 2,
+  info: 3,
+  debug: 4
+};
+var setLogLevel = (level) => {
+  currentLevel = level;
+};
+var shouldLog = (level) => {
+  return levels[level] <= levels[currentLevel];
+};
+var log = {
+  debug: (...args) => shouldLog("debug") && console.debug("[debug]", ...args),
+  info: (...args) => shouldLog("info") && console.info("[info]", ...args),
+  warn: (...args) => shouldLog("warn") && console.warn("[warn]", ...args),
+  error: (...args) => shouldLog("error") && console.error("[error]", ...args)
+};
+
+// src/utils/convert.ts
 var convert = async (inputPath, outputDir, opts) => {
   if (!fs.existsSync(inputPath)) {
     throw new Error(`Input model file not found: ${inputPath}`);
   }
-  const spinner = ora("\u{1F504} Converting model to ONNX...").start();
+  log.info("Converting model to ONNX...");
   return new Promise((resolve, reject) => {
-    const scriptPath = path.resolve(__dirname, "../../python/convert.py");
+    const scriptPath = path.resolve(process.cwd(), "python/convert.py");
+    log.debug("Script path for `convert.py`:", scriptPath);
     const args = [
       scriptPath,
       "--input",
@@ -33,17 +205,18 @@ var convert = async (inputPath, outputDir, opts) => {
     ];
     if (opts.quantize) args.push("--quantize");
     const python = spawn("python3", args);
-    python.stdout.on("data", (data) => process.stdout.write(data));
-    python.stderr.on(
-      "data",
-      (data) => process.stderr.write(chalk.red(data.toString()))
-    );
+    python.stdout.on("data", (data) => {
+      log.debug("[python stdout]", data.toString());
+    });
+    python.stderr.on("data", (data) => {
+      log.error("[python stderr]", data.toString());
+    });
     python.on("close", (code) => {
       if (code === 0) {
-        spinner.succeed("\u2705 Model converted successfully.");
+        log.info("Model converted successfully");
         resolve();
       } else {
-        spinner.fail("\u274C Model conversion failed.");
+        log.error("Model conversion failed");
         reject(new Error(`convert.py exited with code ${code}`));
       }
     });
@@ -51,11 +224,10 @@ var convert = async (inputPath, outputDir, opts) => {
 };
 
 // src/utils/generate-wrapper.ts
-import ora2 from "ora";
 import fs2 from "node:fs";
 import path2 from "node:path";
 var generateWrapper = async (outputDir, opts) => {
-  const spinner = ora2("\u{1F6E0} Generating wrapper files...").start();
+  log.info("Generating wrapper files...");
   const wrapper = `import { InferenceSession, Tensor } from 'onnxruntime-web';
 
 const softmax = (logits) => {
@@ -112,36 +284,55 @@ export function loadModel(): Promise<LoadedModel>;`;
   try {
     fs2.writeFileSync(path2.join(outputDir, "wrapped.ts"), wrapper);
     fs2.writeFileSync(path2.join(outputDir, "wrapped.d.ts"), typings);
-    spinner.succeed("\u2705 Wrapper files generated.");
+    log.info("Wrapper files generated");
   } catch (error) {
-    spinner.fail("\u274C Failed to generate wrapper files.");
+    log.error("Failed generate wrapper files");
     throw error;
   }
 };
 
 // src/wrapture.ts
 var program = new Command();
-program.name("wrapture").description("\u{1F300} One-click model exporter: from PyTorch to Web-ready JS/TS").version("0.1.0").requiredOption("-i, --input <file>", "Path to the PyTorch model (.pt)").requiredOption(
+program.name("wrapture").description(
+  `\u{1F300} ${chalk2.blue("One-click model exporter: ")}from PyTorch to Web-ready JS/TS.
+
+Wrapture lets you go from a Python-trained model to deployable JavaScript with a single command.
+It generates TypeScript bindings and a Web/Node-compatible wrapper, using WebGPU/WASM-ready ONNX runtimes.
+
+Report issues here: https://github.com/phun-ky/wrapture`
+).version(package_default.version).requiredOption("-i, --input <file>", "path to the PyTorch model (.pt)").requiredOption(
   "-o, --output <dir>",
-  "Output directory for the wrapped model"
-).option("--quantize", "Apply quantization to reduce model size").option("--format <type>", "Export format: onnx (default)", "onnx").option(
+  "output directory for the wrapped model"
+).option("--quantize", "apply quantization to reduce model size").option("--format <type>", "export format: onnx (default)", "onnx").option(
   "--backend <backend>",
-  "Inference backend: webgpu | wasm | cpu",
+  "inference backend: webgpu | wasm | cpu",
   "webgpu"
+).option(
+  "--logLevel <level>",
+  "set log level: silent | error | warn | info | debug",
+  "error"
 ).action(async (opts) => {
+  checkPythonAvailable();
+  checkPythonDeps();
   const input = path3.resolve(opts.input);
   const output = path3.resolve(opts.output);
+  setLogLevel(
+    process.env.LOGLEVEL || opts.logLevel || "error"
+  );
   if (!existsSync(input)) {
-    console.error(chalk2.red(`Input file not found: ${input}`));
+    console.error(
+      `${chalk2.red.bold("\u2718 Input file not found:")} ${chalk2.white(input)}`
+    );
     process.exit(1);
   }
-  console.log(chalk2.cyan("\u2728 Wrapture: Exporting model..."));
+  const spinner = ora("Wrapture: Exporting model...").start();
   try {
     await convert(input, output, opts);
     await generateWrapper(output, opts);
-    console.log(chalk2.green("\u2705 Done! Your model is wrapped and ready."));
+    spinner.succeed("Done! Your model is wrapped and ready.");
   } catch (err) {
-    console.error(chalk2.red("\u274C Failed to export model:"), err);
+    spinner.fail("Failed to export model:");
+    console.error(err);
     process.exit(1);
   }
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wrapture",
-  "version": "0.0.2",
+  "version": "0.1.0",
   "description": "Wrapture lets you go from a Python-trained model to deployable JavaScript with a single command. It generates TypeScript bindings and a Web/Node-compatible wrapper, using WebGPU/WASM-ready ONNX runtimes.",
   "keywords": [
     "onnx",

package/python/convert.py

@@ -43,9 +43,9 @@ def convert_to_onnx(input_path, output_path, quantize=False):
     assert check, "Simplified model could not be validated"
     onnx.checker.check_model(model_simp)  # throw if broken
     onnx.save(model_simp, output_model_path)
-    print("✅ ONNX model simplified.")
+    print("✓ ONNX model simplified.")
 
-    print(f"✅ Exported ONNX model to: {output_model_path}")
+    print(f"✓ Exported ONNX model to: {output_model_path}")
 
     if quantize:
         quant_model_path = os.path.join(output_path, 'model_quant.onnx')

package/src/utils/check-deps.ts

@@ -0,0 +1,55 @@
+/* global process */
+import chalk from 'chalk';
+
+import { spawnSync } from 'node:child_process';
+
+const printError = (title: string, body: string) => {
+  console.error(`\n${chalk.red.bold('✘')} ${chalk.red.bold(title)}`);
+  console.error(chalk.white(body));
+};
+
+export const checkPythonAvailable = (): void => {
+  const result = spawnSync('python3', ['--version'], { encoding: 'utf-8' });
+
+  if (result.error || result.status !== 0) {
+    printError(
+      'Python 3 is not available.',
+      'Please install it from https://www.python.org/downloads/ and ensure it is added to your PATH.'
+    );
+    process.exit(1);
+  }
+};
+
+export const checkPythonDeps = (): void => {
+  const check = spawnSync(
+    'python3',
+    [
+      '-c',
+      `
+import sys
+missing = []
+for module in ['torch', 'onnx', 'onnxsim', 'onnxruntime']:
+    try:
+        __import__(module)
+    except ImportError:
+        missing.append(module)
+if missing:
+    print(', '.join(missing))
+    sys.exit(1)
+  `
+    ],
+    { encoding: 'utf-8' }
+  );
+
+  if (check.status !== 0) {
+    const missing = check.stdout.trim().split(',').filter(Boolean);
+
+    printError(
+      'Missing Python dependencies.',
+      `Please install the following packages:\n\n  ${chalk.yellow(
+        `python3 -m pip install ${missing.join(' ')}`
+      )}`
+    );
+    process.exit(1);
+  }
+};

package/src/utils/convert.ts

@@ -1,15 +1,10 @@
 /* eslint-disable import/no-unused-modules */
 /* global process */
-import chalk from 'chalk';
-import ora from 'ora';
-
 import { spawn } from 'node:child_process';
 import fs from 'node:fs';
-import path, { dirname } from 'node:path';
-import { fileURLToPath } from 'node:url';
+import path from 'node:path';
 
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = dirname(__filename);
+import { log } from './log-level.js';
 
 /**
  * Options for the {@link convert} function.
@@ -60,10 +55,13 @@ export const convert = async (
     throw new Error(`Input model file not found: ${inputPath}`);
   }
 
-  const spinner = ora('🔄 Converting model to ONNX...').start();
+  log.info('Converting model to ONNX...');
 
   return new Promise((resolve, reject) => {
-    const scriptPath = path.resolve(__dirname, '../../python/convert.py');
+    const scriptPath = path.resolve(process.cwd(), 'python/convert.py');
+
+    log.debug('Script path for `convert.py`:', scriptPath);
+
     const args = [
       scriptPath,
       '--input',
@@ -78,18 +76,21 @@ export const convert = async (
 
     const python = spawn('python3', args);
 
-    python.stdout.on('data', (data) => process.stdout.write(data));
-    python.stderr.on('data', (data) =>
-      process.stderr.write(chalk.red(data.toString()))
-    );
+    python.stdout.on('data', (data) => {
+      log.debug('[python stdout]', data.toString());
+    });
+
+    python.stderr.on('data', (data) => {
+      log.error('[python stderr]', data.toString());
+    });
 
     python.on('close', (code) => {
       if (code === 0) {
-        spinner.succeed('✅ Model converted successfully.');
+        log.info('Model converted successfully');
 
         resolve();
       } else {
-        spinner.fail('❌ Model conversion failed.');
+        log.error('Model conversion failed');
 
         reject(new Error(`convert.py exited with code ${code}`));
       }

package/src/utils/generate-wrapper.ts

@@ -1,8 +1,9 @@
-import ora from 'ora';
-
+/* eslint-disable import/no-unused-modules */
 import fs from 'node:fs';
 import path from 'node:path';
 
+import { log } from './log-level.js';
+
 /**
  * Options for generating ONNX wrapper files.
  */
@@ -38,10 +39,11 @@ export interface GenerateWrapperOptionsInterface {
  */
 export const generateWrapper = async (
   outputDir: string,
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+
   opts: GenerateWrapperOptionsInterface
 ): Promise<void> => {
-  const spinner = ora('🛠 Generating wrapper files...').start();
+  log.info('Generating wrapper files...');
+
   const wrapper = `import { InferenceSession, Tensor } from 'onnxruntime-web';
 
 const softmax = (logits) => {
@@ -99,9 +101,10 @@ export function loadModel(): Promise<LoadedModel>;`;
   try {
     fs.writeFileSync(path.join(outputDir, 'wrapped.ts'), wrapper);
     fs.writeFileSync(path.join(outputDir, 'wrapped.d.ts'), typings);
-    spinner.succeed('✅ Wrapper files generated.');
+
+    log.info('Wrapper files generated');
   } catch (error) {
-    spinner.fail('❌ Failed to generate wrapper files.');
+    log.error('Failed generate wrapper files');
     throw error;
   }
 };

package/src/utils/log-level.ts

@@ -0,0 +1,31 @@
+/* eslint-disable no-console */
+export type LogLevelType = 'silent' | 'error' | 'warn' | 'info' | 'debug';
+
+let currentLevel: LogLevelType = 'info';
+
+const levels: Record<LogLevelType, number> = {
+  silent: 0,
+  error: 1,
+  warn: 2,
+  info: 3,
+  debug: 4
+};
+
+export const setLogLevel = (level: LogLevelType) => {
+  currentLevel = level;
+};
+
+const shouldLog = (level: LogLevelType): boolean => {
+  return levels[level] <= levels[currentLevel];
+};
+
+export const log = {
+  debug: (...args: unknown[]) =>
+    shouldLog('debug') && console.debug('[debug]', ...args),
+  info: (...args: unknown[]) =>
+    shouldLog('info') && console.info('[info]', ...args),
+  warn: (...args: unknown[]) =>
+    shouldLog('warn') && console.warn('[warn]', ...args),
+  error: (...args: unknown[]) =>
+    shouldLog('error') && console.error('[error]', ...args)
+};

package/src/wrapture.ts

@@ -2,48 +2,76 @@
 /* global process */
 import chalk from 'chalk';
 import { Command } from 'commander';
+import ora from 'ora';
 
 import { existsSync } from 'node:fs';
 import path from 'node:path';
 
+import pkg from '../package.json';
+
+import { checkPythonAvailable, checkPythonDeps } from './utils/check-deps.js';
 import { convert } from './utils/convert.js';
 import { generateWrapper } from './utils/generate-wrapper.js';
+import { LogLevelType, setLogLevel } from './utils/log-level.js';
 
 const program = new Command();
 
 program
   .name('wrapture')
-  .description('🌀 One-click model exporter: from PyTorch to Web-ready JS/TS')
-  .version('0.1.0')
-  .requiredOption('-i, --input <file>', 'Path to the PyTorch model (.pt)')
+  .description(
+    `🌀 ${chalk.blue('One-click model exporter: ')}from PyTorch to Web-ready JS/TS.
+
+Wrapture lets you go from a Python-trained model to deployable JavaScript with a single command.
+It generates TypeScript bindings and a Web/Node-compatible wrapper, using WebGPU/WASM-ready ONNX runtimes.
+
+Report issues here: https://github.com/phun-ky/wrapture`
+  )
+  .version(pkg.version)
+  .requiredOption('-i, --input <file>', 'path to the PyTorch model (.pt)')
   .requiredOption(
     '-o, --output <dir>',
-    'Output directory for the wrapped model'
+    'output directory for the wrapped model'
   )
-  .option('--quantize', 'Apply quantization to reduce model size')
-  .option('--format <type>', 'Export format: onnx (default)', 'onnx')
+  .option('--quantize', 'apply quantization to reduce model size')
+  .option('--format <type>', 'export format: onnx (default)', 'onnx')
   .option(
     '--backend <backend>',
-    'Inference backend: webgpu | wasm | cpu',
+    'inference backend: webgpu | wasm | cpu',
     'webgpu'
   )
+  .option(
+    '--logLevel <level>',
+    'set log level: silent | error | warn | info | debug',
+    'error'
+  )
   .action(async (opts) => {
+    checkPythonAvailable();
+    checkPythonDeps();
+
     const input = path.resolve(opts.input);
     const output = path.resolve(opts.output);
 
+    setLogLevel(
+      (process.env.LOGLEVEL as LogLevelType) || opts.logLevel || 'error'
+    );
+
     if (!existsSync(input)) {
-      console.error(chalk.red(`Input file not found: ${input}`));
+      console.error(
+        `${chalk.red.bold('✘ Input file not found:')} ${chalk.white(input)}`
+      );
       process.exit(1);
     }
 
-    console.log(chalk.cyan('✨ Wrapture: Exporting model...'));
+    const spinner = ora('Wrapture: Exporting model...').start();
 
     try {
       await convert(input, output, opts);
       await generateWrapper(output, opts);
-      console.log(chalk.green('✅ Done! Your model is wrapped and ready.'));
+      spinner.succeed('Done! Your model is wrapped and ready.');
     } catch (err) {
-      console.error(chalk.red('❌ Failed to export model:'), err);
+      spinner.fail('Failed to export model:');
+      console.error(err);
+
       process.exit(1);
     }
   });