@drakulavich/parakeet-cli 0.7.1 → 0.7.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@drakulavich/parakeet-cli",
-  "version": "0.7.1",
+  "version": "0.7.2",
   "description": "Fast local speech-to-text CLI. CoreML on Apple Silicon, ONNX on CPU. 25 languages.",
   "type": "module",
   "bin": {
@@ -53,6 +53,7 @@
     "typescript": "^6.0.2"
   },
   "dependencies": {
-    "onnxruntime-node": "^1.24.0"
+    "onnxruntime-node": "^1.24.0",
+    "picocolors": "^1.1.1"
   }
 }

package/src/cli.ts

@@ -4,6 +4,7 @@ import { transcribe } from "./lib";
 import { downloadModel } from "./onnx-install";
 import { downloadCoreML } from "./coreml-install";
 import { isMacArm64 } from "./coreml";
+import { log } from "./log";
 
 async function main(): Promise<void> {
   const args = process.argv.slice(2);
@@ -24,7 +25,7 @@ async function main(): Promise<void> {
     try {
       if (forceCoreML) {
         if (!isMacArm64()) {
-          console.error("Error: CoreML backend is only available on macOS Apple Silicon.");
+          log.error("CoreML backend is only available on macOS Apple Silicon.");
           process.exit(1);
         }
         await downloadCoreML(noCache);
@@ -37,7 +38,7 @@ async function main(): Promise<void> {
       }
     } catch (err: unknown) {
       const message = err instanceof Error ? err.message : String(err);
-      console.error(`Error: ${message}`);
+      log.error(message);
       process.exit(1);
     }
     process.exit(0);
@@ -46,8 +47,8 @@ async function main(): Promise<void> {
   const file = positional[0];
 
   if (!file) {
-    console.error("Usage: parakeet [--version] <audio_file>");
-    console.error("       parakeet install [--coreml | --onnx] [--no-cache]");
+    log.info("Usage: parakeet [--version] <audio_file>");
+    log.info("       parakeet install [--coreml | --onnx] [--no-cache]");
     process.exit(1);
   }
 
@@ -56,7 +57,7 @@ async function main(): Promise<void> {
     if (text) process.stdout.write(text + "\n");
   } catch (err: unknown) {
     const message = err instanceof Error ? err.message : String(err);
-    console.error(message);
+    log.error(message);
     process.exit(1);
   }
 }

package/src/coreml-install.ts

@@ -2,6 +2,7 @@ import { join, dirname } from "path";
 import { homedir } from "os";
 import { existsSync, mkdirSync, chmodSync } from "fs";
 import { getCoreMLBinPath } from "./coreml";
+import { log } from "./log";
 
 const COREML_BINARY_NAME = "parakeet-coreml-darwin-arm64";
 const GITHUB_REPO = "drakulavich/parakeet-cli";
@@ -226,12 +227,12 @@ export async function downloadCoreML(
   const plan = planCoreMLInstall(state, noCache);
 
   if (!plan.downloadBinary && !plan.downloadModels) {
-    console.log("CoreML backend already installed.");
+    log.success("CoreML backend already installed.");
     return binPath;
   }
 
   if (plan.downloadBinary) {
-    console.error("Downloading parakeet-coreml binary...");
+    log.progress("Downloading parakeet-coreml binary...");
     const res = await fetchCoreMLBinary();
     mkdirSync(dirname(binPath), { recursive: true });
     await Bun.write(binPath, res);
@@ -242,6 +243,6 @@ export async function downloadCoreML(
     await ensureCoreMLModels(binPath, runner, output);
   }
 
-  console.log("CoreML backend installed successfully.");
+  log.success("CoreML backend installed successfully.");
   return binPath;
 }

package/src/log.ts

@@ -0,0 +1,9 @@
+import pc from "picocolors";
+
+export const log = {
+  info: (msg: string) => console.log(msg),
+  success: (msg: string) => console.log(pc.green(msg)),
+  progress: (msg: string) => console.log(pc.cyan(msg)),
+  warn: (msg: string) => console.error(pc.yellow(msg)),
+  error: (msg: string) => console.error(pc.red(msg)),
+};

package/src/onnx-install.ts

@@ -1,6 +1,7 @@
 import { join } from "path";
 import { homedir } from "os";
 import { existsSync, mkdirSync } from "fs";
+import { log } from "./log";
 
 export const HF_REPO = "istupakov/parakeet-tdt-0.6b-v3-onnx";
 
@@ -48,7 +49,7 @@ export async function downloadModel(noCache = false, modelDir?: string): Promise
   const dir = modelDir ?? getModelDir();
 
   if (!noCache && isModelCached(dir)) {
-    console.log("Model already downloaded.");
+    log.success("Model already downloaded.");
     return dir;
   }
 
@@ -60,7 +61,7 @@ export async function downloadModel(noCache = false, modelDir?: string): Promise
 
     if (!noCache && existsSync(dest)) continue;
 
-    console.error(`Downloading ${file}...`);
+    log.progress(`Downloading ${file}...`);
 
     let res: Response;
     try {
@@ -93,6 +94,6 @@ export async function downloadModel(noCache = false, modelDir?: string): Promise
     }
   }
 
-  console.log("Model downloaded successfully.");
+  log.success("Model downloaded successfully.");
   return dir;
 }