@cfasim-ui/pyodide 0.1.4 → 0.1.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cfasim-ui/pyodide",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "type": "module",
   "description": "Pyodide (Python-in-browser) integration for cfasim-ui",
   "license": "Apache-2.0",
@@ -9,6 +9,9 @@
     "url": "https://github.com/CDCgov/cfa-simulator.git",
     "directory": "cfasim-ui/pyodide"
   },
+  "publishConfig": {
+    "access": "public"
+  },
   "files": [
     "src"
   ],
@@ -18,7 +21,7 @@
   },
   "dependencies": {
     "pyodide": "^0.29.3",
-    "@cfasim-ui/shared": "0.1.4"
+    "@cfasim-ui/shared": "0.1.6"
   },
   "peerDependencies": {
     "vue": "^3.5.0"

package/src/pyodide.worker.ts

@@ -44,14 +44,26 @@ const pyodideReadyPromise = (async () => {
   return pyodide;
 })();
 
+let wheelsInstalled = false;
+
+async function installAllWheels() {
+  if (wheelsInstalled) return;
+  wheelsInstalled = true;
+  const urls = Object.values(wheelMap).map(
+    (f) => `${self.location.origin}${baseUrl}${f}`,
+  );
+  if (urls.length > 0) {
+    await micropip.install(urls);
+  }
+}
+
 async function ensureModule(
   pyodide: Awaited<typeof pyodideReadyPromise>,
   moduleName: string,
 ) {
   if (loadedModules.has(moduleName)) return;
-  const filename = wheelMap[moduleName];
-  if (!filename) throw new Error(`Unknown module: ${moduleName}`);
-  await micropip.install(`${self.location.origin}${baseUrl}${filename}`);
+  if (!wheelMap[moduleName]) throw new Error(`Unknown module: ${moduleName}`);
+  await installAllWheels();
   pyodide.pyimport(moduleName);
   loadedModules.add(moduleName);
 }

package/src/vitePlugin.ts

@@ -6,11 +6,16 @@ import type { Plugin } from "vite";
 interface CfasimPyodideOptions {
   /** Path to the Python model directory (default: "model") */
   model?: string;
+  /** PyPI packages to download at build time and serve locally */
+  pypiDeps?: string[];
 }
 
 /**
  * Vite plugin that builds a Python wheel from a local model directory
  * and generates wheels.json in the public directory.
+ *
+ * Use `pypiDeps` to prebuild PyPI packages (like cfasim-model) as local
+ * wheels for faster browser runtime — avoids PyPI round-trips on page load.
  */
 export function cfasimPyodide(options?: CfasimPyodideOptions): Plugin {
   const modelDir = options?.model ?? "model";
@@ -18,6 +23,12 @@ export function cfasimPyodide(options?: CfasimPyodideOptions): Plugin {
   function build(root: string) {
     const publicDir = resolve(root, "public");
     mkdirSync(publicDir, { recursive: true });
+    for (const dep of options?.pypiDeps ?? []) {
+      execSync(
+        `uv pip download ${dep} --dest public --no-deps --python-version 3.12 --platform any`,
+        { cwd: root, stdio: "pipe" },
+      );
+    }
     execSync(`uv build ${modelDir} --wheel --out-dir public`, {
       cwd: root,
       stdio: "pipe",