AI models shared registry + Code cleanup + Better async handling + Some fixes, etc

crates/jarvis-core/src/models.rs

@@ -0,0 +1,67 @@
+mod registry;
+mod catalog;
+pub mod structs;
+pub mod loaders;
+
+pub mod vosk_models;
+pub mod gliner_models;
+
+// re-export loaders
+#[cfg(feature = "jarvis_app")]
+pub use loaders::embedding;
+
+#[cfg(feature = "jarvis_app")]
+pub use loaders::gliner;
+
+#[cfg(feature = "jarvis_app")]
+pub use loaders::ort_model;
+
+#[cfg(feature = "jarvis_app")]
+pub use loaders::intent_classifier;
+
+#[cfg(feature = "vosk")]
+pub use loaders::vosk;
+
+#[cfg(feature = "nnnoiseless")]
+pub use loaders::nnnoiseless;
+
+pub use registry::ModelRegistry;
+pub use structs::{Task, ModelDef, BackendOption};
+
+use once_cell::sync::OnceCell;
+
+use crate::APP_DIR;
+
+pub const MODELS_PATH: &str = "resources/models";
+
+static REGISTRY: OnceCell<ModelRegistry> = OnceCell::new();
+
+pub fn init() -> Result<(), String> {
+    if REGISTRY.get().is_some() {
+        return Ok(());
+    }
+
+    let registry = ModelRegistry::new();
+
+    let models_dir = APP_DIR.join(MODELS_PATH);
+    let models = catalog::scan_models(&models_dir);
+    info!("Found {} model(s) in {:?}", models.len(), models_dir);
+    registry.set_catalog(models);
+
+    REGISTRY.set(registry)
+        .map_err(|_| "Models registry already initialized".to_string())?;
+
+    Ok(())
+}
+
+pub fn registry() -> &'static ModelRegistry {
+    REGISTRY.get().expect("Models registry not initialized - call models::init() first")
+}
+
+pub fn get_options(task: Task) -> Vec<BackendOption> {
+    registry().with_catalog(|models| catalog::get_options(task, models))
+}
+
+pub fn is_valid_backend(task: Task, backend_id: &str) -> bool {
+    registry().with_catalog(|models| catalog::is_valid_backend(task, backend_id, models))
+}