react-native-sherpa-onnx 0.3.9 → 0.4.0

package/ios/enhancement/sherpa-onnx-enhancement-wrapper.mm

@@ -0,0 +1,218 @@
+#include "sherpa-onnx-enhancement-wrapper.h"
+
+#include "sherpa-onnx-model-detect.h"
+
+#include <optional>
+
+#include "sherpa-onnx/c-api/cxx-api.h"
+
+namespace sherpaonnx {
+namespace {
+
+std::string EnhancementKindToString(EnhancementModelKind kind) {
+    switch (kind) {
+        case EnhancementModelKind::kGtcrn:
+            return "gtcrn";
+        case EnhancementModelKind::kDpdfNet:
+            return "dpdfnet";
+        default:
+            return "unknown";
+    }
+}
+
+sherpa_onnx::cxx::OfflineSpeechDenoiserModelConfig BuildModelConfig(
+    const EnhancementDetectResult& detect,
+    int32_t numThreads,
+    const std::optional<std::string>& provider,
+    bool debug
+) {
+    sherpa_onnx::cxx::OfflineSpeechDenoiserModelConfig cfg;
+    cfg.num_threads = numThreads;
+    cfg.debug = debug;
+    if (provider.has_value() && !provider->empty()) {
+        cfg.provider = *provider;
+    }
+    switch (detect.selectedKind) {
+        case EnhancementModelKind::kGtcrn:
+            cfg.gtcrn.model = detect.paths.model;
+            break;
+        case EnhancementModelKind::kDpdfNet:
+            cfg.dpdfnet.model = detect.paths.model;
+            break;
+        default:
+            break;
+    }
+    return cfg;
+}
+
+EnhancedAudioResult ToEnhancedAudioResult(
+    const sherpa_onnx::cxx::DenoisedAudio& audio
+) {
+    EnhancedAudioResult out;
+    out.samples = audio.samples;
+    out.sampleRate = audio.sample_rate;
+    return out;
+}
+
+} // namespace
+
+class EnhancementWrapper::Impl {
+public:
+    bool initialized = false;
+    std::optional<sherpa_onnx::cxx::OfflineSpeechDenoiser> denoiser;
+};
+
+EnhancementWrapper::EnhancementWrapper() : pImpl(std::make_unique<Impl>()) {}
+
+EnhancementWrapper::~EnhancementWrapper() { release(); }
+
+EnhancementInitializeResult EnhancementWrapper::initialize(
+    const std::string& modelDir,
+    const std::string& modelType,
+    int32_t numThreads,
+    const std::optional<std::string>& provider,
+    bool debug
+) {
+    EnhancementInitializeResult result;
+    if (pImpl->initialized) {
+        release();
+    }
+    if (modelDir.empty()) {
+        result.error = "Enhancement model directory is empty";
+        return result;
+    }
+
+    auto detect = DetectEnhancementModel(modelDir, modelType);
+    result.detectedModels = detect.detectedModels;
+    result.modelType = EnhancementKindToString(detect.selectedKind);
+    if (!detect.ok) {
+        result.error = detect.error;
+        return result;
+    }
+
+    sherpa_onnx::cxx::OfflineSpeechDenoiserConfig config;
+    config.model = BuildModelConfig(detect, numThreads, provider, debug);
+    pImpl->denoiser = sherpa_onnx::cxx::OfflineSpeechDenoiser::Create(config);
+    pImpl->initialized = true;
+
+    result.success = true;
+    result.sampleRate = pImpl->denoiser->GetSampleRate();
+    return result;
+}
+
+EnhancedAudioResult EnhancementWrapper::runSamples(
+    const std::vector<float>& samples,
+    int32_t sampleRate
+) {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) {
+        return {};
+    }
+    return ToEnhancedAudioResult(
+        pImpl->denoiser->Run(samples.data(), static_cast<int32_t>(samples.size()), sampleRate)
+    );
+}
+
+int32_t EnhancementWrapper::getSampleRate() const {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) return 0;
+    return pImpl->denoiser->GetSampleRate();
+}
+
+bool EnhancementWrapper::isInitialized() const { return pImpl->initialized; }
+
+void EnhancementWrapper::release() {
+    if (pImpl->denoiser.has_value()) {
+        pImpl->denoiser.reset();
+    }
+    pImpl->initialized = false;
+}
+
+class OnlineEnhancementWrapper::Impl {
+public:
+    bool initialized = false;
+    std::optional<sherpa_onnx::cxx::OnlineSpeechDenoiser> denoiser;
+};
+
+OnlineEnhancementWrapper::OnlineEnhancementWrapper()
+    : pImpl(std::make_unique<Impl>()) {}
+
+OnlineEnhancementWrapper::~OnlineEnhancementWrapper() { release(); }
+
+EnhancementInitializeResult OnlineEnhancementWrapper::initialize(
+    const std::string& modelDir,
+    const std::string& modelType,
+    int32_t numThreads,
+    const std::optional<std::string>& provider,
+    bool debug
+) {
+    EnhancementInitializeResult result;
+    if (pImpl->initialized) {
+        release();
+    }
+    if (modelDir.empty()) {
+        result.error = "Enhancement model directory is empty";
+        return result;
+    }
+
+    auto detect = DetectEnhancementModel(modelDir, modelType);
+    result.detectedModels = detect.detectedModels;
+    result.modelType = EnhancementKindToString(detect.selectedKind);
+    if (!detect.ok) {
+        result.error = detect.error;
+        return result;
+    }
+
+    sherpa_onnx::cxx::OnlineSpeechDenoiserConfig config;
+    config.model = BuildModelConfig(detect, numThreads, provider, debug);
+    pImpl->denoiser = sherpa_onnx::cxx::OnlineSpeechDenoiser::Create(config);
+    pImpl->initialized = true;
+
+    result.success = true;
+    result.sampleRate = pImpl->denoiser->GetSampleRate();
+    result.frameShiftInSamples = pImpl->denoiser->GetFrameShiftInSamples();
+    return result;
+}
+
+EnhancedAudioResult OnlineEnhancementWrapper::runSamples(
+    const std::vector<float>& samples,
+    int32_t sampleRate
+) {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) {
+        return {};
+    }
+    return ToEnhancedAudioResult(
+        pImpl->denoiser->Run(samples.data(), static_cast<int32_t>(samples.size()), sampleRate)
+    );
+}
+
+EnhancedAudioResult OnlineEnhancementWrapper::flush() {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) {
+        return {};
+    }
+    return ToEnhancedAudioResult(pImpl->denoiser->Flush());
+}
+
+void OnlineEnhancementWrapper::reset() {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) return;
+    pImpl->denoiser->Reset();
+}
+
+int32_t OnlineEnhancementWrapper::getSampleRate() const {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) return 0;
+    return pImpl->denoiser->GetSampleRate();
+}
+
+int32_t OnlineEnhancementWrapper::getFrameShiftInSamples() const {
+    if (!pImpl->initialized || !pImpl->denoiser.has_value()) return 0;
+    return pImpl->denoiser->GetFrameShiftInSamples();
+}
+
+bool OnlineEnhancementWrapper::isInitialized() const { return pImpl->initialized; }
+
+void OnlineEnhancementWrapper::release() {
+    if (pImpl->denoiser.has_value()) {
+        pImpl->denoiser.reset();
+    }
+    pImpl->initialized = false;
+}
+
+} // namespace sherpaonnx

package/ios/model_detect/sherpa-onnx-model-detect-enhancement.mm

@@ -0,0 +1,92 @@
+#include "sherpa-onnx-model-detect.h"
+#include "sherpa-onnx-model-detect-helper.h"
+#include "sherpa-onnx-validate-enhancement.h"
+
+#include <optional>
+#include <string>
+#include <vector>
+
+namespace sherpaonnx {
+namespace {
+
+using namespace model_detect;
+
+EnhancementModelKind ParseEnhancementModelType(const std::string& modelType) {
+    if (modelType == "gtcrn") return EnhancementModelKind::kGtcrn;
+    if (modelType == "dpdfnet") return EnhancementModelKind::kDpdfNet;
+    return EnhancementModelKind::kUnknown;
+}
+
+} // namespace
+
+EnhancementDetectResult DetectEnhancementModel(
+    const std::string& modelDir,
+    const std::string& modelType
+) {
+    EnhancementDetectResult result;
+
+    if (modelDir.empty()) {
+        result.error = "Enhancement: model directory is empty";
+        return result;
+    }
+    if (!FileExists(modelDir) || !IsDirectory(modelDir)) {
+        result.error =
+            "Enhancement: model directory does not exist or is not a directory: " +
+            modelDir;
+        return result;
+    }
+
+    const std::vector<FileEntry> files = ListFilesRecursive(modelDir, 4);
+    const std::string gtcrnModel =
+        FindOnnxByAnyToken(files, {"gtcrn"}, std::nullopt);
+    const std::string dpdfnetModel =
+        FindOnnxByAnyToken(files, {"dpdfnet"}, std::nullopt);
+
+    if (!gtcrnModel.empty()) {
+        result.detectedModels.push_back({"gtcrn", modelDir});
+    }
+    if (!dpdfnetModel.empty()) {
+        result.detectedModels.push_back({"dpdfnet", modelDir});
+    }
+
+    EnhancementModelKind selected = EnhancementModelKind::kUnknown;
+    if (modelType == "auto" || modelType.empty()) {
+        if (!gtcrnModel.empty()) {
+            selected = EnhancementModelKind::kGtcrn;
+        } else if (!dpdfnetModel.empty()) {
+            selected = EnhancementModelKind::kDpdfNet;
+        }
+    } else {
+        selected = ParseEnhancementModelType(modelType);
+        if (selected == EnhancementModelKind::kUnknown) {
+            result.error = "Enhancement: unknown model type: " + modelType;
+            return result;
+        }
+    }
+
+    switch (selected) {
+        case EnhancementModelKind::kGtcrn:
+            result.paths.model = gtcrnModel;
+            break;
+        case EnhancementModelKind::kDpdfNet:
+            result.paths.model = dpdfnetModel;
+            break;
+        default:
+            result.error = "Enhancement: no compatible model type detected in " +
+                           modelDir;
+            return result;
+    }
+
+    auto validation =
+        ValidateEnhancementPaths(selected, result.paths, modelDir);
+    if (!validation.ok) {
+        result.error = validation.error;
+        return result;
+    }
+
+    result.selectedKind = selected;
+    result.ok = true;
+    return result;
+}
+
+} // namespace sherpaonnx

package/ios/model_detect/sherpa-onnx-model-detect.h

@@ -42,6 +42,12 @@ enum class TtsModelKind {
     kSupertonic
 };
 
+enum class EnhancementModelKind {
+    kUnknown,
+    kGtcrn,
+    kDpdfNet
+};
+
 struct SttModelPaths {
     std::string encoder;
     std::string decoder;
@@ -170,6 +176,10 @@ struct TtsModelPaths {
     std::string voiceStyle;
 };
 
+struct EnhancementModelPaths {
+    std::string model;
+};
+
 struct SttDetectResult {
     bool ok = false;
     std::string error;
@@ -191,6 +201,14 @@ struct TtsDetectResult {
     std::vector<std::string> lexiconLanguageCandidates;
 };
 
+struct EnhancementDetectResult {
+    bool ok = false;
+    std::string error;
+    std::vector<DetectedModel> detectedModels;
+    EnhancementModelKind selectedKind = EnhancementModelKind::kUnknown;
+    EnhancementModelPaths paths;
+};
+
 SttDetectResult DetectSttModel(
     const std::string& modelDir,
     const std::optional<bool>& preferInt8,
@@ -203,6 +221,11 @@ TtsDetectResult DetectTtsModel(
     const std::string& modelType
 );
 
+EnhancementDetectResult DetectEnhancementModel(
+    const std::string& modelDir,
+    const std::string& modelType
+);
+
 } // namespace sherpaonnx
 
 #endif // SHERPA_ONNX_MODEL_DETECT_H

package/ios/model_detect/sherpa-onnx-validate-enhancement.h

@@ -0,0 +1,30 @@
+#ifndef SHERPA_ONNX_VALIDATE_ENHANCEMENT_H
+#define SHERPA_ONNX_VALIDATE_ENHANCEMENT_H
+
+#include "sherpa-onnx-model-detect.h"
+#include <string>
+#include <vector>
+
+namespace sherpaonnx {
+
+struct EnhancementFieldRequirement {
+    const char* fieldName;
+    std::string EnhancementModelPaths::* field;
+    bool required;
+};
+
+struct EnhancementValidationResult {
+    bool ok = true;
+    std::vector<std::string> missingRequired;
+    std::string error;
+};
+
+EnhancementValidationResult ValidateEnhancementPaths(
+    EnhancementModelKind kind,
+    const EnhancementModelPaths& paths,
+    const std::string& modelDir
+);
+
+} // namespace sherpaonnx
+
+#endif // SHERPA_ONNX_VALIDATE_ENHANCEMENT_H

package/ios/model_detect/sherpa-onnx-validate-enhancement.mm

@@ -0,0 +1,69 @@
+#include "sherpa-onnx-validate-enhancement.h"
+
+#include <cstddef>
+
+namespace sherpaonnx {
+namespace {
+
+static const EnhancementFieldRequirement kGenericReqs[] = {
+    {"model", &EnhancementModelPaths::model, true},
+};
+
+static const EnhancementFieldRequirement* GetRequirements(
+    EnhancementModelKind kind,
+    size_t& count
+) {
+    switch (kind) {
+        case EnhancementModelKind::kGtcrn:
+        case EnhancementModelKind::kDpdfNet:
+            count = std::size(kGenericReqs);
+            return kGenericReqs;
+        default:
+            count = 0;
+            return nullptr;
+    }
+}
+
+static const char* EnhancementKindToName(EnhancementModelKind kind) {
+    switch (kind) {
+        case EnhancementModelKind::kGtcrn:
+            return "GTCRN";
+        case EnhancementModelKind::kDpdfNet:
+            return "DPDFNet";
+        default:
+            return "Unknown";
+    }
+}
+
+} // namespace
+
+EnhancementValidationResult ValidateEnhancementPaths(
+    EnhancementModelKind kind,
+    const EnhancementModelPaths& paths,
+    const std::string& modelDir
+) {
+    EnhancementValidationResult result;
+    size_t count = 0;
+    const auto* reqs = GetRequirements(kind, count);
+    if (!reqs) return result;
+
+    for (size_t i = 0; i < count; ++i) {
+        if (reqs[i].required && (paths.*(reqs[i].field)).empty()) {
+            result.missingRequired.push_back(reqs[i].fieldName);
+        }
+    }
+
+    if (!result.missingRequired.empty()) {
+        result.ok = false;
+        result.error = std::string("Enhancement ") + EnhancementKindToName(kind) +
+            ": missing required files in " + modelDir + ": ";
+        for (size_t i = 0; i < result.missingRequired.size(); ++i) {
+            if (i > 0) result.error += ", ";
+            result.error += result.missingRequired[i];
+        }
+    }
+
+    return result;
+}
+
+} // namespace sherpaonnx

package/lib/module/NativeSherpaOnnx.js.map

@@ -1 +1 @@
-{"version":3,"names":["TurboModuleRegistry","getEnforcing"],"sourceRoot":"../../src","sources":["NativeSherpaOnnx.ts"],"mappings":";;AAAA,SAASA,mBAAmB,QAA0B,cAAc;;AAEpE;;AA0sBA,eAAeA,mBAAmB,CAACC,YAAY,CAAO,YAAY,CAAC","ignoreList":[]}
+{"version":3,"names":["TurboModuleRegistry","getEnforcing"],"sourceRoot":"../../src","sources":["NativeSherpaOnnx.ts"],"mappings":";;AAAA,SAASA,mBAAmB,QAA0B,cAAc;;AAEpE;;AAixBA,eAAeA,mBAAmB,CAACC,YAAY,CAAO,YAAY,CAAC","ignoreList":[]}

package/lib/module/download/localModels.js

@@ -2,8 +2,8 @@
 
 import { Platform } from 'react-native';
 import { DocumentDirectoryPath, exists, readFile, readDir, unlink, writeFile } from '@dr.pogodin/react-native-fs';
-import { getCachePath, getModelsBaseDir, getModelDir, getManifestPath, getNativeAssetExtractedModelDir, getReadyMarkerPath, getTarArchivePath, getOnnxPath } from "./paths.js";
-import { resolveActualModelDir, removeDirectoryRecursive } from "./validation.js";
+import { getCachePath, getModelsBaseDir, getModelDir, getManifestPath, getReadyMarkerPath, getTarArchivePath, getOnnxPath } from "./paths.js";
+import { resolveActualModelDir } from "./validation.js";
 import { emitModelsListUpdated } from "./downloadEvents.js";
 import { clearMemoryCacheForCategory } from "./registry.js";
 export async function listDownloadedModelsByCategory(category) {
@@ -131,7 +131,6 @@ export async function deleteModelByCategory(category, id) {
   if (await exists(onnxPath)) {
     await unlink(onnxPath);
   }
-  await removeDirectoryRecursive(getNativeAssetExtractedModelDir(id));
   const list = await listDownloadedModelsByCategory(category);
   emitModelsListUpdated(category, list);
 }

package/lib/module/download/localModels.js.map

@@ -1 +1 @@
-{"version":3,"names":["Platform","DocumentDirectoryPath","exists","readFile","readDir","unlink","writeFile","getCachePath","getModelsBaseDir","getModelDir","getManifestPath","getNativeAssetExtractedModelDir","getReadyMarkerPath","getTarArchivePath","getOnnxPath","resolveActualModelDir","removeDirectoryRecursive","emitModelsListUpdated","clearMemoryCacheForCategory","listDownloadedModelsByCategory","category","baseDir","existsResult","entries","models","entry","isDirectory","manifestPath","name","manifestExists","raw","manifest","JSON","parse","model","push","isModelDownloadedByCategory","id","readyPath","getLocalModelPathByCategory","ready","updateModelLastUsed","installDir","lastUsed","Date","toISOString","stringify","error","console","warn","listDownloadedModelsWithMetadata","modelsWithMetadata","downloadedAt","sizeOnDisk","size","status","cleanupLeastRecentlyUsed","options","length","keepCount","sorted","sort","a","b","aTime","bTime","getTime","deletedIds","bytesFreed","targetBytes","maxToDelete","maxModelsToDelete","i","item","deleteModelByCategory","log","modelDir","tarPath","onnxPath","list","clearModelCacheByCategory","cachePath","getDownloadStorageBase","OS"],"sourceRoot":"../../../src","sources":["download/localModels.ts"],"mappings":";;AAAA,SAASA,QAAQ,QAAQ,cAAc;AACvC,SACEC,qBAAqB,EACrBC,MAAM,EACNC,QAAQ,EACRC,OAAO,EACPC,MAAM,EACNC,SAAS,QACJ,6BAA6B;AAOpC,SACEC,YAAY,EACZC,gBAAgB,EAChBC,WAAW,EACXC,eAAe,EACfC,+BAA+B,EAC/BC,kBAAkB,EAClBC,iBAAiB,EACjBC,WAAW,QACN,YAAS;AAChB,SAASC,qBAAqB,EAAEC,wBAAwB,QAAQ,iBAAc;AAC9E,SAASC,qBAAqB,QAAQ,qBAAkB;AACxD,SAASC,2BAA2B,QAAQ,eAAY;AAExD,OAAO,eAAeC,8BAA8BA,CAClDC,QAAuB,EACT;EACd,MAAMC,OAAO,GAAGb,gBAAgB,CAACY,QAAQ,CAAC;EAC1C,MAAME,YAAY,GAAG,MAAMpB,MAAM,CAACmB,OAAO,CAAC;EAC1C,IAAI,CAACC,YAAY,EAAE,OAAO,EAAE;EAE5B,MAAMC,OAAO,GAAG,MAAMnB,OAAO,CAACiB,OAAO,CAAC;EACtC,MAAMG,MAAW,GAAG,EAAE;EAEtB,KAAK,MAAMC,KAAK,IAAIF,OAAO,EAAE;IAC3B,IAAI,CAACE,KAAK,CAACC,WAAW,CAAC,CAAC,EAAE;IAC1B,MAAMC,YAAY,GAAGjB,eAAe,CAACU,QAAQ,EAAEK,KAAK,CAACG,IAAI,CAAC;IAC1D,MAAMC,cAAc,GAAG,MAAM3B,MAAM,CAACyB,YAAY,CAAC;IACjD,IAAI,CAACE,cAAc,EAAE;IACrB,IAAI;MACF,MAAMC,GAAG,GAAG,MAAM3B,QAAQ,CAACwB,YAAY,EAAE,MAAM,CAAC;MAChD,MAAMI,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACH,GAAG,CAAqB;MACpD,IAAIC,QAAQ,CAACG,KAAK,EAAE;QAClBV,MAAM,CAACW,IAAI,CAACJ,QAAQ,CAACG,KAAK,CAAC;MAC7B;IACF,CAAC,CAAC,MAAM;MACN;IAAA;EAEJ;EAEA,OAAOV,MAAM;AACf;AAEA,OAAO,eAAeY,2BAA2BA,CAC/ChB,QAAuB,EACvBiB,EAAU,EACQ;EAClB,MAAMC,SAAS,GAAG1B,kBAAkB,CAACQ,QAAQ,EAAEiB,EAAE,CAAC;EAClD,OAAOnC,MAAM,CAACoC,SAAS,CAAC;AAC1B;AAEA,OAAO,eAAeC,2BAA2BA,CAC/CnB,QAAuB,EACvBiB,EAAU,EACc;EACxB,MAAMG,KAAK,GAAG,MAAMJ,2BAA2B,CAAChB,QAAQ,EAAEiB,EAAE,CAAC;EAC7D,IAAI,CAACG,KAAK,EAAE,OAAO,IAAI;EAEvB,MAAMC,mBAAmB,CAACrB,QAAQ,EAAEiB,EAAE,CAAC;EAEvC,MAAMK,UAAU,GAAGjC,WAAW,CAACW,QAAQ,EAAEiB,EAAE,CAAC;EAC5C,OAAOtB,qBAAqB,CAAC2B,UAAU,CAAC;AAC1C;AAEA,OAAO,eAAeD,mBAAmBA,CACvCrB,QAAuB,EACvBiB,EAAU,EACK;EACf,MAAMV,YAAY,GAAGjB,eAAe,CAACU,QAAQ,EAAEiB,EAAE,CAAC;EAClD,MAAMf,YAAY,GAAG,MAAMpB,MAAM,CAACyB,YAAY,CAAC;EAC/C,IAAI,CAACL,YAAY,EAAE;EAEnB,IAAI;IACF,MAAMQ,GAAG,GAAG,MAAM3B,QAAQ,CAACwB,YAAY,EAAE,MAAM,CAAC;IAChD,MAAMI,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACH,GAAG,CAAkB;IACjDC,QAAQ,CAACY,QAAQ,GAAG,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;IAC5C,MAAMvC,SAAS,CAACqB,YAAY,EAAEK,IAAI,CAACc,SAAS,CAACf,QAAQ,CAAC,EAAE,MAAM,CAAC;EACjE,CAAC,CAAC,OAAOgB,KAAK,EAAE;IACdC,OAAO,CAACC,IAAI,CAAC,iCAAiC7B,QAAQ,IAAIiB,EAAE,GAAG,EAAEU,KAAK,CAAC;EACzE;AACF;AAEA,OAAO,eAAeG,gCAAgCA,CACpD9B,QAAuB,EACU;EACjC,MAAMC,OAAO,GAAGb,gBAAgB,CAACY,QAAQ,CAAC;EAC1C,MAAME,YAAY,GAAG,MAAMpB,MAAM,CAACmB,OAAO,CAAC;EAC1C,IAAI,CAACC,YAAY,EAAE,OAAO,EAAE;EAE5B,MAAMC,OAAO,GAAG,MAAMnB,OAAO,CAACiB,OAAO,CAAC;EACtC,MAAM8B,kBAA0C,GAAG,EAAE;EAErD,KAAK,MAAM1B,KAAK,IAAIF,OAAO,EAAE;IAC3B,IAAI,CAACE,KAAK,CAACC,WAAW,CAAC,CAAC,EAAE;IAE1B,MAAMC,YAAY,GAAGjB,eAAe,CAACU,QAAQ,EAAEK,KAAK,CAACG,IAAI,CAAC;IAC1D,MAAMC,cAAc,GAAG,MAAM3B,MAAM,CAACyB,YAAY,CAAC;IAEjD,IAAIE,cAAc,EAAE;MAClB,IAAI;QACF,MAAMC,GAAG,GAAG,MAAM3B,QAAQ,CAACwB,YAAY,EAAE,MAAM,CAAC;QAChD,MAAMI,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACH,GAAG,CAAqB;QACpD,IAAIC,QAAQ,CAACG,KAAK,EAAE;UAClBiB,kBAAkB,CAAChB,IAAI,CAAC;YACtBD,KAAK,EAAEH,QAAQ,CAACG,KAAK;YACrBkB,YAAY,EAAErB,QAAQ,CAACqB,YAAY;YACnCT,QAAQ,EAAEZ,QAAQ,CAACY,QAAQ,IAAI,IAAI;YACnCU,UAAU,EAAEtB,QAAQ,CAACsB,UAAU,IAAI5B,KAAK,CAAC6B,IAAI;YAC7CC,MAAM,EAAE;UACV,CAAC,CAAC;QACJ;MACF,CAAC,CAAC,OAAOR,KAAK,EAAE;QACdC,OAAO,CAACC,IAAI,CACV,+BAA+B7B,QAAQ,IAAIK,KAAK,CAACG,IAAI,GAAG,EACxDmB,KACF,CAAC;MACH;IACF;EACF;EAEA,OAAOI,kBAAkB;AAC3B;AAEA,OAAO,eAAeK,wBAAwBA,CAC5CpC,QAAuB,EACvBqC,OAIC,EACkB;EACnB,MAAMN,kBAAkB,GAAG,MAAMD,gCAAgC,CAAC9B,QAAQ,CAAC;EAE3E,IAAI+B,kBAAkB,CAACO,MAAM,KAAK,CAAC,EAAE;IACnC,OAAO,EAAE;EACX;EAEA,MAAMC,SAAS,GAAGF,OAAO,EAAEE,SAAS,IAAI,CAAC;EACzC,IAAIR,kBAAkB,CAACO,MAAM,IAAIC,SAAS,EAAE;IAC1C,OAAO,EAAE;EACX;EAEA,MAAMC,MAAM,GAAGT,kBAAkB,CAACU,IAAI,CAAC,CAACC,CAAC,EAAEC,CAAC,KAAK;IAC/C,MAAMC,KAAK,GAAGF,CAAC,CAACnB,QAAQ,IAAImB,CAAC,CAACV,YAAY;IAC1C,MAAMa,KAAK,GAAGF,CAAC,CAACpB,QAAQ,IAAIoB,CAAC,CAACX,YAAY;IAC1C,OAAO,IAAIR,IAAI,CAACoB,KAAK,CAAC,CAACE,OAAO,CAAC,CAAC,GAAG,IAAItB,IAAI,CAACqB,KAAK,CAAC,CAACC,OAAO,CAAC,CAAC;EAC9D,CAAC,CAAC;EAEF,MAAMC,UAAoB,GAAG,EAAE;EAC/B,IAAIC,UAAU,GAAG,CAAC;EAClB,MAAMC,WAAW,GAAGZ,OAAO,EAAEY,WAAW,IAAI,CAAC;EAC7C,MAAMC,WAAW,GAAGb,OAAO,EAAEc,iBAAiB,IAAIX,MAAM,CAACF,MAAM,GAAGC,SAAS;EAE3E,KAAK,IAAIa,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGZ,MAAM,CAACF,MAAM,GAAGC,SAAS,IAAIa,CAAC,GAAGF,WAAW,EAAEE,CAAC,EAAE,EAAE;IACrE,MAAMC,IAAI,GAAGb,MAAM,CAACY,CAAC,CAAC;IACtB,IAAI,CAACC,IAAI,EAAE;IAEX,IAAI;MACF,MAAMC,qBAAqB,CAACtD,QAAQ,EAAEqD,IAAI,CAACvC,KAAK,CAACG,EAAE,CAAC;MACpD8B,UAAU,CAAChC,IAAI,CAACsC,IAAI,CAACvC,KAAK,CAACG,EAAE,CAAC;MAC9B+B,UAAU,IAAIK,IAAI,CAACpB,UAAU,IAAI,CAAC;MAElCL,OAAO,CAAC2B,GAAG,CACT,yBAAyBvD,QAAQ,IAAIqD,IAAI,CAACvC,KAAK,CAACG,EAAE,WAChD,CAACoC,IAAI,CAACpB,UAAU,IAAI,CAAC,IAAI,IAAI,GAAG,IAAI,MAExC,CAAC;MAED,IAAIgB,WAAW,GAAG,CAAC,IAAID,UAAU,IAAIC,WAAW,EAAE;QAChD;MACF;IACF,CAAC,CAAC,OAAOtB,KAAK,EAAE;MACdC,OAAO,CAACC,IAAI,CACV,kCAAkC7B,QAAQ,IAAIqD,IAAI,CAACvC,KAAK,CAACG,EAAE,GAAG,EAC9DU,KACF,CAAC;IACH;EACF;EAEA,OAAOoB,UAAU;AACnB;AAEA,OAAO,eAAeO,qBAAqBA,CACzCtD,QAAuB,EACvBiB,EAAU,EACK;EACf,MAAMuC,QAAQ,GAAGnE,WAAW,CAACW,QAAQ,EAAEiB,EAAE,CAAC;EAC1C,MAAMwC,OAAO,GAAGhE,iBAAiB,CAACO,QAAQ,EAAEiB,EAAE,CAAC;EAC/C,MAAMyC,QAAQ,GAAGhE,WAAW,CAACM,QAAQ,EAAEiB,EAAE,CAAC;EAC1C,IAAI,MAAMnC,MAAM,CAAC0E,QAAQ,CAAC,EAAE;IAC1B,MAAMvE,MAAM,CAACuE,QAAQ,CAAC;EACxB;EACA,IAAI,MAAM1E,MAAM,CAAC2E,OAAO,CAAC,EAAE;IACzB,MAAMxE,MAAM,CAACwE,OAAO,CAAC;EACvB;EACA,IAAI,MAAM3E,MAAM,CAAC4E,QAAQ,CAAC,EAAE;IAC1B,MAAMzE,MAAM,CAACyE,QAAQ,CAAC;EACxB;EACA,MAAM9D,wBAAwB,CAACL,+BAA+B,CAAC0B,EAAE,CAAC,CAAC;EACnE,MAAM0C,IAAI,GAAG,MAAM5D,8BAA8B,CAAgBC,QAAQ,CAAC;EAC1EH,qBAAqB,CAACG,QAAQ,EAAE2D,IAAI,CAAC;AACvC;AAEA,OAAO,eAAeC,yBAAyBA,CAC7C5D,QAAuB,EACR;EACf,MAAM6D,SAAS,GAAG1E,YAAY,CAACa,QAAQ,CAAC;EACxC,IAAI,MAAMlB,MAAM,CAAC+E,SAAS,CAAC,EAAE;IAC3B,MAAM5E,MAAM,CAAC4E,SAAS,CAAC;EACzB;EACA/D,2BAA2B,CAACE,QAAQ,CAAC;AACvC;AAEA,OAAO,eAAe8D,sBAAsBA,CAAA,EAAoB;EAC9D,IAAIlF,QAAQ,CAACmF,EAAE,KAAK,KAAK,EAAE;IACzB,OAAOlF,qBAAqB;EAC9B;EACA,OAAOA,qBAAqB;AAC9B","ignoreList":[]}
+{"version":3,"names":["Platform","DocumentDirectoryPath","exists","readFile","readDir","unlink","writeFile","getCachePath","getModelsBaseDir","getModelDir","getManifestPath","getReadyMarkerPath","getTarArchivePath","getOnnxPath","resolveActualModelDir","emitModelsListUpdated","clearMemoryCacheForCategory","listDownloadedModelsByCategory","category","baseDir","existsResult","entries","models","entry","isDirectory","manifestPath","name","manifestExists","raw","manifest","JSON","parse","model","push","isModelDownloadedByCategory","id","readyPath","getLocalModelPathByCategory","ready","updateModelLastUsed","installDir","lastUsed","Date","toISOString","stringify","error","console","warn","listDownloadedModelsWithMetadata","modelsWithMetadata","downloadedAt","sizeOnDisk","size","status","cleanupLeastRecentlyUsed","options","length","keepCount","sorted","sort","a","b","aTime","bTime","getTime","deletedIds","bytesFreed","targetBytes","maxToDelete","maxModelsToDelete","i","item","deleteModelByCategory","log","modelDir","tarPath","onnxPath","list","clearModelCacheByCategory","cachePath","getDownloadStorageBase","OS"],"sourceRoot":"../../../src","sources":["download/localModels.ts"],"mappings":";;AAAA,SAASA,QAAQ,QAAQ,cAAc;AACvC,SACEC,qBAAqB,EACrBC,MAAM,EACNC,QAAQ,EACRC,OAAO,EACPC,MAAM,EACNC,SAAS,QACJ,6BAA6B;AAOpC,SACEC,YAAY,EACZC,gBAAgB,EAChBC,WAAW,EACXC,eAAe,EACfC,kBAAkB,EAClBC,iBAAiB,EACjBC,WAAW,QACN,YAAS;AAChB,SAASC,qBAAqB,QAAQ,iBAAc;AACpD,SAASC,qBAAqB,QAAQ,qBAAkB;AACxD,SAASC,2BAA2B,QAAQ,eAAY;AAExD,OAAO,eAAeC,8BAA8BA,CAClDC,QAAuB,EACT;EACd,MAAMC,OAAO,GAAGX,gBAAgB,CAACU,QAAQ,CAAC;EAC1C,MAAME,YAAY,GAAG,MAAMlB,MAAM,CAACiB,OAAO,CAAC;EAC1C,IAAI,CAACC,YAAY,EAAE,OAAO,EAAE;EAE5B,MAAMC,OAAO,GAAG,MAAMjB,OAAO,CAACe,OAAO,CAAC;EACtC,MAAMG,MAAW,GAAG,EAAE;EAEtB,KAAK,MAAMC,KAAK,IAAIF,OAAO,EAAE;IAC3B,IAAI,CAACE,KAAK,CAACC,WAAW,CAAC,CAAC,EAAE;IAC1B,MAAMC,YAAY,GAAGf,eAAe,CAACQ,QAAQ,EAAEK,KAAK,CAACG,IAAI,CAAC;IAC1D,MAAMC,cAAc,GAAG,MAAMzB,MAAM,CAACuB,YAAY,CAAC;IACjD,IAAI,CAACE,cAAc,EAAE;IACrB,IAAI;MACF,MAAMC,GAAG,GAAG,MAAMzB,QAAQ,CAACsB,YAAY,EAAE,MAAM,CAAC;MAChD,MAAMI,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACH,GAAG,CAAqB;MACpD,IAAIC,QAAQ,CAACG,KAAK,EAAE;QAClBV,MAAM,CAACW,IAAI,CAACJ,QAAQ,CAACG,KAAK,CAAC;MAC7B;IACF,CAAC,CAAC,MAAM;MACN;IAAA;EAEJ;EAEA,OAAOV,MAAM;AACf;AAEA,OAAO,eAAeY,2BAA2BA,CAC/ChB,QAAuB,EACvBiB,EAAU,EACQ;EAClB,MAAMC,SAAS,GAAGzB,kBAAkB,CAACO,QAAQ,EAAEiB,EAAE,CAAC;EAClD,OAAOjC,MAAM,CAACkC,SAAS,CAAC;AAC1B;AAEA,OAAO,eAAeC,2BAA2BA,CAC/CnB,QAAuB,EACvBiB,EAAU,EACc;EACxB,MAAMG,KAAK,GAAG,MAAMJ,2BAA2B,CAAChB,QAAQ,EAAEiB,EAAE,CAAC;EAC7D,IAAI,CAACG,KAAK,EAAE,OAAO,IAAI;EAEvB,MAAMC,mBAAmB,CAACrB,QAAQ,EAAEiB,EAAE,CAAC;EAEvC,MAAMK,UAAU,GAAG/B,WAAW,CAACS,QAAQ,EAAEiB,EAAE,CAAC;EAC5C,OAAOrB,qBAAqB,CAAC0B,UAAU,CAAC;AAC1C;AAEA,OAAO,eAAeD,mBAAmBA,CACvCrB,QAAuB,EACvBiB,EAAU,EACK;EACf,MAAMV,YAAY,GAAGf,eAAe,CAACQ,QAAQ,EAAEiB,EAAE,CAAC;EAClD,MAAMf,YAAY,GAAG,MAAMlB,MAAM,CAACuB,YAAY,CAAC;EAC/C,IAAI,CAACL,YAAY,EAAE;EAEnB,IAAI;IACF,MAAMQ,GAAG,GAAG,MAAMzB,QAAQ,CAACsB,YAAY,EAAE,MAAM,CAAC;IAChD,MAAMI,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACH,GAAG,CAAkB;IACjDC,QAAQ,CAACY,QAAQ,GAAG,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;IAC5C,MAAMrC,SAAS,CAACmB,YAAY,EAAEK,IAAI,CAACc,SAAS,CAACf,QAAQ,CAAC,EAAE,MAAM,CAAC;EACjE,CAAC,CAAC,OAAOgB,KAAK,EAAE;IACdC,OAAO,CAACC,IAAI,CAAC,iCAAiC7B,QAAQ,IAAIiB,EAAE,GAAG,EAAEU,KAAK,CAAC;EACzE;AACF;AAEA,OAAO,eAAeG,gCAAgCA,CACpD9B,QAAuB,EACU;EACjC,MAAMC,OAAO,GAAGX,gBAAgB,CAACU,QAAQ,CAAC;EAC1C,MAAME,YAAY,GAAG,MAAMlB,MAAM,CAACiB,OAAO,CAAC;EAC1C,IAAI,CAACC,YAAY,EAAE,OAAO,EAAE;EAE5B,MAAMC,OAAO,GAAG,MAAMjB,OAAO,CAACe,OAAO,CAAC;EACtC,MAAM8B,kBAA0C,GAAG,EAAE;EAErD,KAAK,MAAM1B,KAAK,IAAIF,OAAO,EAAE;IAC3B,IAAI,CAACE,KAAK,CAACC,WAAW,CAAC,CAAC,EAAE;IAE1B,MAAMC,YAAY,GAAGf,eAAe,CAACQ,QAAQ,EAAEK,KAAK,CAACG,IAAI,CAAC;IAC1D,MAAMC,cAAc,GAAG,MAAMzB,MAAM,CAACuB,YAAY,CAAC;IAEjD,IAAIE,cAAc,EAAE;MAClB,IAAI;QACF,MAAMC,GAAG,GAAG,MAAMzB,QAAQ,CAACsB,YAAY,EAAE,MAAM,CAAC;QAChD,MAAMI,QAAQ,GAAGC,IAAI,CAACC,KAAK,CAACH,GAAG,CAAqB;QACpD,IAAIC,QAAQ,CAACG,KAAK,EAAE;UAClBiB,kBAAkB,CAAChB,IAAI,CAAC;YACtBD,KAAK,EAAEH,QAAQ,CAACG,KAAK;YACrBkB,YAAY,EAAErB,QAAQ,CAACqB,YAAY;YACnCT,QAAQ,EAAEZ,QAAQ,CAACY,QAAQ,IAAI,IAAI;YACnCU,UAAU,EAAEtB,QAAQ,CAACsB,UAAU,IAAI5B,KAAK,CAAC6B,IAAI;YAC7CC,MAAM,EAAE;UACV,CAAC,CAAC;QACJ;MACF,CAAC,CAAC,OAAOR,KAAK,EAAE;QACdC,OAAO,CAACC,IAAI,CACV,+BAA+B7B,QAAQ,IAAIK,KAAK,CAACG,IAAI,GAAG,EACxDmB,KACF,CAAC;MACH;IACF;EACF;EAEA,OAAOI,kBAAkB;AAC3B;AAEA,OAAO,eAAeK,wBAAwBA,CAC5CpC,QAAuB,EACvBqC,OAIC,EACkB;EACnB,MAAMN,kBAAkB,GAAG,MAAMD,gCAAgC,CAAC9B,QAAQ,CAAC;EAE3E,IAAI+B,kBAAkB,CAACO,MAAM,KAAK,CAAC,EAAE;IACnC,OAAO,EAAE;EACX;EAEA,MAAMC,SAAS,GAAGF,OAAO,EAAEE,SAAS,IAAI,CAAC;EACzC,IAAIR,kBAAkB,CAACO,MAAM,IAAIC,SAAS,EAAE;IAC1C,OAAO,EAAE;EACX;EAEA,MAAMC,MAAM,GAAGT,kBAAkB,CAACU,IAAI,CAAC,CAACC,CAAC,EAAEC,CAAC,KAAK;IAC/C,MAAMC,KAAK,GAAGF,CAAC,CAACnB,QAAQ,IAAImB,CAAC,CAACV,YAAY;IAC1C,MAAMa,KAAK,GAAGF,CAAC,CAACpB,QAAQ,IAAIoB,CAAC,CAACX,YAAY;IAC1C,OAAO,IAAIR,IAAI,CAACoB,KAAK,CAAC,CAACE,OAAO,CAAC,CAAC,GAAG,IAAItB,IAAI,CAACqB,KAAK,CAAC,CAACC,OAAO,CAAC,CAAC;EAC9D,CAAC,CAAC;EAEF,MAAMC,UAAoB,GAAG,EAAE;EAC/B,IAAIC,UAAU,GAAG,CAAC;EAClB,MAAMC,WAAW,GAAGZ,OAAO,EAAEY,WAAW,IAAI,CAAC;EAC7C,MAAMC,WAAW,GAAGb,OAAO,EAAEc,iBAAiB,IAAIX,MAAM,CAACF,MAAM,GAAGC,SAAS;EAE3E,KAAK,IAAIa,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGZ,MAAM,CAACF,MAAM,GAAGC,SAAS,IAAIa,CAAC,GAAGF,WAAW,EAAEE,CAAC,EAAE,EAAE;IACrE,MAAMC,IAAI,GAAGb,MAAM,CAACY,CAAC,CAAC;IACtB,IAAI,CAACC,IAAI,EAAE;IAEX,IAAI;MACF,MAAMC,qBAAqB,CAACtD,QAAQ,EAAEqD,IAAI,CAACvC,KAAK,CAACG,EAAE,CAAC;MACpD8B,UAAU,CAAChC,IAAI,CAACsC,IAAI,CAACvC,KAAK,CAACG,EAAE,CAAC;MAC9B+B,UAAU,IAAIK,IAAI,CAACpB,UAAU,IAAI,CAAC;MAElCL,OAAO,CAAC2B,GAAG,CACT,yBAAyBvD,QAAQ,IAAIqD,IAAI,CAACvC,KAAK,CAACG,EAAE,WAChD,CAACoC,IAAI,CAACpB,UAAU,IAAI,CAAC,IAAI,IAAI,GAAG,IAAI,MAExC,CAAC;MAED,IAAIgB,WAAW,GAAG,CAAC,IAAID,UAAU,IAAIC,WAAW,EAAE;QAChD;MACF;IACF,CAAC,CAAC,OAAOtB,KAAK,EAAE;MACdC,OAAO,CAACC,IAAI,CACV,kCAAkC7B,QAAQ,IAAIqD,IAAI,CAACvC,KAAK,CAACG,EAAE,GAAG,EAC9DU,KACF,CAAC;IACH;EACF;EAEA,OAAOoB,UAAU;AACnB;AAEA,OAAO,eAAeO,qBAAqBA,CACzCtD,QAAuB,EACvBiB,EAAU,EACK;EACf,MAAMuC,QAAQ,GAAGjE,WAAW,CAACS,QAAQ,EAAEiB,EAAE,CAAC;EAC1C,MAAMwC,OAAO,GAAG/D,iBAAiB,CAACM,QAAQ,EAAEiB,EAAE,CAAC;EAC/C,MAAMyC,QAAQ,GAAG/D,WAAW,CAACK,QAAQ,EAAEiB,EAAE,CAAC;EAC1C,IAAI,MAAMjC,MAAM,CAACwE,QAAQ,CAAC,EAAE;IAC1B,MAAMrE,MAAM,CAACqE,QAAQ,CAAC;EACxB;EACA,IAAI,MAAMxE,MAAM,CAACyE,OAAO,CAAC,EAAE;IACzB,MAAMtE,MAAM,CAACsE,OAAO,CAAC;EACvB;EACA,IAAI,MAAMzE,MAAM,CAAC0E,QAAQ,CAAC,EAAE;IAC1B,MAAMvE,MAAM,CAACuE,QAAQ,CAAC;EACxB;EACA,MAAMC,IAAI,GAAG,MAAM5D,8BAA8B,CAAgBC,QAAQ,CAAC;EAC1EH,qBAAqB,CAACG,QAAQ,EAAE2D,IAAI,CAAC;AACvC;AAEA,OAAO,eAAeC,yBAAyBA,CAC7C5D,QAAuB,EACR;EACf,MAAM6D,SAAS,GAAGxE,YAAY,CAACW,QAAQ,CAAC;EACxC,IAAI,MAAMhB,MAAM,CAAC6E,SAAS,CAAC,EAAE;IAC3B,MAAM1E,MAAM,CAAC0E,SAAS,CAAC;EACzB;EACA/D,2BAA2B,CAACE,QAAQ,CAAC;AACvC;AAEA,OAAO,eAAe8D,sBAAsBA,CAAA,EAAoB;EAC9D,IAAIhF,QAAQ,CAACiF,EAAE,KAAK,KAAK,EAAE;IACzB,OAAOhF,qBAAqB;EAC9B;EACA,OAAOA,qBAAqB;AAC9B","ignoreList":[]}

package/lib/module/download/paths.js

@@ -86,7 +86,8 @@ export function getExtractionStatePath(category, modelId) {
 /**
  * Directory where native `resolveAssetPath` materializes a bundled model folder
  * (`DocumentDirectoryPath/models/{modelId}` — Android internal `files/models/...`).
- * Separate from {@link getModelDir}. Remove on delete so empty dirs do not break detection.
+ * Separate from {@link getModelDir}. `deleteModelByCategory` does not remove this tree; it
+ * only deletes download-manager installs under `sherpa-onnx/models/`.
  */
 export function getNativeAssetExtractedModelDir(modelId) {
   const safeId = modelId.replace(/[/\\]/g, '');

package/lib/module/download/paths.js.map

@@ -1 +1 @@
-{"version":3,"names":["DocumentDirectoryPath","ModelCategory","RELEASE_API_BASE","CATEGORY_CONFIG","Tts","tag","cacheFile","baseDir","Stt","Vad","Diarization","Enhancement","Separation","Qnn","getCacheDir","getCachePath","category","getModelsBaseDir","getModelDir","modelId","getArchiveFilename","archiveExt","getArchivePath","filename","getTarArchivePath","getOnnxPath","getReadyMarkerPath","getManifestPath","getDownloadStatePath","getExtractionStatePath","getNativeAssetExtractedModelDir","safeId","replace","getReleaseUrl"],"sourceRoot":"../../../src","sources":["download/paths.ts"],"mappings":";;AAAA,SAASA,qBAAqB,QAAQ,6BAA6B;AACnE,SAASC,aAAa,QAAQ,YAAS;AAEvC,SAASC,gBAAgB,QAAQ,gBAAa;AAE9C,OAAO,MAAMC,eAGZ,GAAG;EACF,CAACF,aAAa,CAACG,GAAG,GAAG;IACnBC,GAAG,EAAE,YAAY;IACjBC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACO,GAAG,GAAG;IACnBH,GAAG,EAAE,YAAY;IACjBC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACQ,GAAG,GAAG;IACnBJ,GAAG,EAAE,YAAY;IACjBC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACS,WAAW,GAAG;IAC3BL,GAAG,EAAE,6BAA6B;IAClCC,SAAS,EAAE,yBAAyB;IACpCC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACU,WAAW,GAAG;IAC3BN,GAAG,EAAE,2BAA2B;IAChCC,SAAS,EAAE,yBAAyB;IACpCC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACW,UAAU,GAAG;IAC1BP,GAAG,EAAE,0BAA0B;IAC/BC,SAAS,EAAE,wBAAwB;IACnCC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACY,GAAG,GAAG;IACnBR,GAAG,EAAE,uBAAuB;IAC5BC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC;AACF,CAAC;AAED,OAAO,SAASc,WAAWA,CAAA,EAAW;EACpC,OAAO,GAAGd,qBAAqB,oBAAoB;AACrD;AAEA,OAAO,SAASe,YAAYA,CAACC,QAAuB,EAAU;EAC5D,OAAO,GAAGF,WAAW,CAAC,CAAC,IAAIX,eAAe,CAACa,QAAQ,CAAC,CAACV,SAAS,EAAE;AAClE;AAEA,OAAO,SAASW,gBAAgBA,CAACD,QAAuB,EAAU;EAChE,OAAOb,eAAe,CAACa,QAAQ,CAAC,CAACT,OAAO;AAC1C;AAEA,OAAO,SAASW,WAAWA,CAACF,QAAuB,EAAEG,OAAe,EAAU;EAC5E,OAAO,GAAGF,gBAAgB,CAACD,QAAQ,CAAC,IAAIG,OAAO,EAAE;AACnD;AAEA,OAAO,SAASC,kBAAkBA,CAChCD,OAAe,EACfE,UAA2B,EACnB;EACR,OAAO,GAAGF,OAAO,IAAIE,UAAU,EAAE;AACnC;AAEA,OAAO,SAASC,cAAcA,CAC5BN,QAAuB,EACvBG,OAAe,EACfE,UAA2B,EACnB;EACR,MAAME,QAAQ,GAAGH,kBAAkB,CAACD,OAAO,EAAEE,UAAU,CAAC;EACxD,IAAIA,UAAU,KAAK,MAAM,EAAE;IACzB,OAAO,GAAGH,WAAW,CAACF,QAAQ,EAAEG,OAAO,CAAC,IAAII,QAAQ,EAAE;EACxD;EACA,OAAO,GAAGN,gBAAgB,CAACD,QAAQ,CAAC,IAAIO,QAAQ,EAAE;AACpD;AAEA,OAAO,SAASC,iBAAiBA,CAC/BR,QAAuB,EACvBG,OAAe,EACP;EACR,OAAOG,cAAc,CAACN,QAAQ,EAAEG,OAAO,EAAE,SAAS,CAAC;AACrD;AAEA,OAAO,SAASM,WAAWA,CAACT,QAAuB,EAAEG,OAAe,EAAU;EAC5E,OAAOG,cAAc,CAACN,QAAQ,EAAEG,OAAO,EAAE,MAAM,CAAC;AAClD;AAEA,OAAO,SAASO,kBAAkBA,CAChCV,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGD,WAAW,CAACF,QAAQ,EAAEG,OAAO,CAAC,SAAS;AACnD;AAEA,OAAO,SAASQ,eAAeA,CAC7BX,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGD,WAAW,CAACF,QAAQ,EAAEG,OAAO,CAAC,gBAAgB;AAC1D;AAEA,OAAO,SAASS,oBAAoBA,CAClCZ,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGF,gBAAgB,CAACD,QAAQ,CAAC,oBAAoBG,OAAO,OAAO;AACxE;;AAEA;AACA,OAAO,SAASU,sBAAsBA,CACpCb,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGF,gBAAgB,CAACD,QAAQ,CAAC,sBAAsBG,OAAO,OAAO;AAC1E;;AAEA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASW,+BAA+BA,CAACX,OAAe,EAAU;EACvE,MAAMY,MAAM,GAAGZ,OAAO,CAACa,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;EAC5C,OAAO,GAAGhC,qBAAqB,WAAW+B,MAAM,EAAE,CAACC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;AACzE;AAEA,OAAO,SAASC,aAAaA,CAACjB,QAAuB,EAAU;EAC7D,OAAO,GAAGd,gBAAgB,IAAIC,eAAe,CAACa,QAAQ,CAAC,CAACX,GAAG,EAAE;AAC/D","ignoreList":[]}
+{"version":3,"names":["DocumentDirectoryPath","ModelCategory","RELEASE_API_BASE","CATEGORY_CONFIG","Tts","tag","cacheFile","baseDir","Stt","Vad","Diarization","Enhancement","Separation","Qnn","getCacheDir","getCachePath","category","getModelsBaseDir","getModelDir","modelId","getArchiveFilename","archiveExt","getArchivePath","filename","getTarArchivePath","getOnnxPath","getReadyMarkerPath","getManifestPath","getDownloadStatePath","getExtractionStatePath","getNativeAssetExtractedModelDir","safeId","replace","getReleaseUrl"],"sourceRoot":"../../../src","sources":["download/paths.ts"],"mappings":";;AAAA,SAASA,qBAAqB,QAAQ,6BAA6B;AACnE,SAASC,aAAa,QAAQ,YAAS;AAEvC,SAASC,gBAAgB,QAAQ,gBAAa;AAE9C,OAAO,MAAMC,eAGZ,GAAG;EACF,CAACF,aAAa,CAACG,GAAG,GAAG;IACnBC,GAAG,EAAE,YAAY;IACjBC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACO,GAAG,GAAG;IACnBH,GAAG,EAAE,YAAY;IACjBC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACQ,GAAG,GAAG;IACnBJ,GAAG,EAAE,YAAY;IACjBC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACS,WAAW,GAAG;IAC3BL,GAAG,EAAE,6BAA6B;IAClCC,SAAS,EAAE,yBAAyB;IACpCC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACU,WAAW,GAAG;IAC3BN,GAAG,EAAE,2BAA2B;IAChCC,SAAS,EAAE,yBAAyB;IACpCC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACW,UAAU,GAAG;IAC1BP,GAAG,EAAE,0BAA0B;IAC/BC,SAAS,EAAE,wBAAwB;IACnCC,OAAO,EAAE,GAAGP,qBAAqB;EACnC,CAAC;EACD,CAACC,aAAa,CAACY,GAAG,GAAG;IACnBR,GAAG,EAAE,uBAAuB;IAC5BC,SAAS,EAAE,iBAAiB;IAC5BC,OAAO,EAAE,GAAGP,qBAAqB;EACnC;AACF,CAAC;AAED,OAAO,SAASc,WAAWA,CAAA,EAAW;EACpC,OAAO,GAAGd,qBAAqB,oBAAoB;AACrD;AAEA,OAAO,SAASe,YAAYA,CAACC,QAAuB,EAAU;EAC5D,OAAO,GAAGF,WAAW,CAAC,CAAC,IAAIX,eAAe,CAACa,QAAQ,CAAC,CAACV,SAAS,EAAE;AAClE;AAEA,OAAO,SAASW,gBAAgBA,CAACD,QAAuB,EAAU;EAChE,OAAOb,eAAe,CAACa,QAAQ,CAAC,CAACT,OAAO;AAC1C;AAEA,OAAO,SAASW,WAAWA,CAACF,QAAuB,EAAEG,OAAe,EAAU;EAC5E,OAAO,GAAGF,gBAAgB,CAACD,QAAQ,CAAC,IAAIG,OAAO,EAAE;AACnD;AAEA,OAAO,SAASC,kBAAkBA,CAChCD,OAAe,EACfE,UAA2B,EACnB;EACR,OAAO,GAAGF,OAAO,IAAIE,UAAU,EAAE;AACnC;AAEA,OAAO,SAASC,cAAcA,CAC5BN,QAAuB,EACvBG,OAAe,EACfE,UAA2B,EACnB;EACR,MAAME,QAAQ,GAAGH,kBAAkB,CAACD,OAAO,EAAEE,UAAU,CAAC;EACxD,IAAIA,UAAU,KAAK,MAAM,EAAE;IACzB,OAAO,GAAGH,WAAW,CAACF,QAAQ,EAAEG,OAAO,CAAC,IAAII,QAAQ,EAAE;EACxD;EACA,OAAO,GAAGN,gBAAgB,CAACD,QAAQ,CAAC,IAAIO,QAAQ,EAAE;AACpD;AAEA,OAAO,SAASC,iBAAiBA,CAC/BR,QAAuB,EACvBG,OAAe,EACP;EACR,OAAOG,cAAc,CAACN,QAAQ,EAAEG,OAAO,EAAE,SAAS,CAAC;AACrD;AAEA,OAAO,SAASM,WAAWA,CAACT,QAAuB,EAAEG,OAAe,EAAU;EAC5E,OAAOG,cAAc,CAACN,QAAQ,EAAEG,OAAO,EAAE,MAAM,CAAC;AAClD;AAEA,OAAO,SAASO,kBAAkBA,CAChCV,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGD,WAAW,CAACF,QAAQ,EAAEG,OAAO,CAAC,SAAS;AACnD;AAEA,OAAO,SAASQ,eAAeA,CAC7BX,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGD,WAAW,CAACF,QAAQ,EAAEG,OAAO,CAAC,gBAAgB;AAC1D;AAEA,OAAO,SAASS,oBAAoBA,CAClCZ,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGF,gBAAgB,CAACD,QAAQ,CAAC,oBAAoBG,OAAO,OAAO;AACxE;;AAEA;AACA,OAAO,SAASU,sBAAsBA,CACpCb,QAAuB,EACvBG,OAAe,EACP;EACR,OAAO,GAAGF,gBAAgB,CAACD,QAAQ,CAAC,sBAAsBG,OAAO,OAAO;AAC1E;;AAEA;AACA;AACA;AACA;AACA;AACA;AACA,OAAO,SAASW,+BAA+BA,CAACX,OAAe,EAAU;EACvE,MAAMY,MAAM,GAAGZ,OAAO,CAACa,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC;EAC5C,OAAO,GAAGhC,qBAAqB,WAAW+B,MAAM,EAAE,CAACC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC;AACzE;AAEA,OAAO,SAASC,aAAaA,CAACjB,QAAuB,EAAU;EAC7D,OAAO,GAAGd,gBAAgB,IAAIC,eAAe,CAACa,QAAQ,CAAC,CAACX,GAAG,EAAE;AAC/D","ignoreList":[]}

package/lib/module/enhancement/index.js

@@ -1,54 +1,69 @@
 "use strict";
 
-/**
- * Speech Enhancement feature module
- *
- * @remarks
- * This feature is not yet implemented. This module serves as a placeholder
- * for future speech enhancement functionality.
- *
- * @example
- * ```typescript
- * // Future usage:
- * import { initializeEnhancement, enhanceAudio } from 'react-native-sherpa-onnx/enhancement';
- *
- * await initializeEnhancement({ modelPath: { type: 'auto', path: 'models/enhancement-model' } });
- * const enhancedPath = await enhanceAudio('path/to/noisy-audio.wav');
- * ```
- */
-
-/**
- * Enhancement initialization options (placeholder)
- */
-
-/**
- * Enhancement result
- */
-
-/**
- * Initialize Speech Enhancement with model directory.
- *
- * @throws {Error} Not yet implemented
- */
-export async function initializeEnhancement(_options) {
-  throw new Error('Speech Enhancement feature is not yet implemented. This is a placeholder module.');
+import SherpaOnnx from "../NativeSherpaOnnx.js";
+import { resolveModelPath } from "../utils.js";
+let enhancementInstanceCounter = 0;
+function normalizeEnhancedAudio(raw) {
+  const samplesArray = Array.isArray(raw.samples) ? raw.samples : Array.from(raw.samples ?? []);
+  return {
+    samples: Float32Array.from(samplesArray),
+    sampleRate: Number(raw.sampleRate ?? 0)
+  };
 }
-
-/**
- * Enhance speech quality in an audio file.
- *
- * @throws {Error} Not yet implemented
- */
-export function enhanceAudio(_filePath) {
-  throw new Error('Speech Enhancement feature is not yet implemented. This is a placeholder module.');
+export async function detectEnhancementModel(modelPath, options) {
+  const resolvedPath = await resolveModelPath(modelPath);
+  const raw = await SherpaOnnx.detectEnhancementModel(resolvedPath, options?.modelType);
+  const err = typeof raw.error === 'string' ? raw.error.trim() : '';
+  return {
+    success: raw.success,
+    ...(err.length > 0 ? {
+      error: err
+    } : {}),
+    detectedModels: raw.detectedModels ?? [],
+    ...(raw.modelType != null && raw.modelType !== '' ? {
+      modelType: raw.modelType
+    } : {})
+  };
 }
-
-/**
- * Release enhancement resources.
- *
- * @throws {Error} Not yet implemented
- */
-export function unloadEnhancement() {
-  throw new Error('Speech Enhancement feature is not yet implemented. This is a placeholder module.');
+export async function createEnhancement(options) {
+  const instanceId = `enhancement_${++enhancementInstanceCounter}`;
+  const resolvedPath = await resolveModelPath(options.modelPath);
+  const init = await SherpaOnnx.initializeEnhancement(instanceId, resolvedPath, options.modelType ?? 'auto', options.numThreads, options.provider, options.debug);
+  if (!init.success) {
+    const nativeError = typeof init.error === 'string' ? init.error.trim() : '';
+    throw new Error(nativeError.length > 0 ? `Enhancement initialization failed: ${nativeError}` : `Enhancement initialization failed for ${instanceId}`);
+  }
+  let destroyed = false;
+  const guard = () => {
+    if (destroyed) {
+      throw new Error(`Enhancement instance ${instanceId} has been destroyed; cannot call methods on it.`);
+    }
+  };
+  return {
+    get instanceId() {
+      return instanceId;
+    },
+    async enhanceFile(inputPath, outputPath) {
+      guard();
+      const raw = await SherpaOnnx.enhanceFile(instanceId, inputPath, outputPath);
+      return normalizeEnhancedAudio(raw);
+    },
+    async enhanceSamples(samples, sampleRate) {
+      guard();
+      const raw = await SherpaOnnx.enhanceSamples(instanceId, samples, sampleRate);
+      return normalizeEnhancedAudio(raw);
+    },
+    async getSampleRate() {
+      guard();
+      return SherpaOnnx.getEnhancementSampleRate(instanceId);
+    },
+    async destroy() {
+      if (destroyed) return;
+      destroyed = true;
+      await SherpaOnnx.unloadEnhancement(instanceId);
+    }
+  };
 }
+export { createStreamingEnhancement } from "./streaming.js";
+export { ENHANCEMENT_MODEL_TYPES } from "./types.js";
 //# sourceMappingURL=index.js.map