wasm support added

Dockerfile

@@ -3,7 +3,7 @@ FROM ubuntu:24.04
 # Disable interactive prompts during package installation
 ENV DEBIAN_FRONTEND=noninteractive
 
-# 1. Install dependencies (including python3, pip, and venv)
+# 1. Install dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     curl \
     jq \
@@ -18,20 +18,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3-venv \
     && rm -rf /var/lib/apt/lists/*
 
-# 2. Delete the pre-created 'ubuntu' user (UID 1000) and create your custom 'user' (UID 1000)
+# 2. Setup non-root user (UID 1000)
 RUN userdel -r ubuntu && useradd -m -u 1000 user
 
 WORKDIR /app
 
-# 3. Create a Python Virtual Environment and install packages
-# This avoids PEP 668 "externally-managed-environment" errors on Ubuntu 24.04
+# 3. Setup Python Virtual Environment
 RUN python3 -m venv /app/venv
 ENV PATH="/app/venv/bin:$PATH"
+RUN pip install --no-cache-dir fastapi uvicorn requests pydantic
 
-# Install FastAPI and Uvicorn
-RUN pip install  --no-cache-dir fastapi uvicorn requests pydantic
-
-# 4. Download & Extract ApexKit Binary
+# 4. Download & Install ApexKit Binary
 RUN echo "Downloading latest public release archive..." && \
     LATEST_URL=$(curl -s https://api.github.com/repos/deniskipeles/apexkit/releases/latest | \
     jq -r '.assets[] | select(.name | contains("linux-gnu.tar.gz")) | .browser_download_url') && \
@@ -42,25 +39,23 @@ RUN echo "Downloading latest public release archive..." && \
     chmod +x /usr/local/bin/apexkit && \
     rm apexkit.tar.gz
 
-# 5. Copy your FastAPI application code
-# Make sure to create a 'main.py' file in the same directory as your Dockerfile
+# 5. Bake WASM Binaries directly into NVMe Cache (/app/.cache/wasm)
+RUN mkdir -p /app/.cache/wasm && \
+    curl -fL -o /app/.cache/wasm/ffmpeg.wasm "https://kipeles-apexkit.hf.space/api/v1/storage/file/333dc27f-4698-45ad-96cc-57e801d1fa22.wasm" && \
+    curl -fL -o /app/.cache/wasm/ffprobe.wasm "https://kipeles-apexkit.hf.space/api/v1/storage/file/f3760ac4-f8d9-4a8b-9993-6fe00c256fc6.wasm"
+
+# 6. Copy FastAPI Application
 COPY --chown=user:user main.py /app/main.py
 
-# 6. Setup directories
-# /app/storage is the fast ephemeral NVMe
-# /data/storage is the mounted persistent storage provided by HF
+# 7. Setup storage directories
 RUN mkdir -p /app/storage /data/storage && \
     chown -R user:user /app /data
 
-# 7. Copy and prepare the entrypoint script
+# 8. Copy and prepare entrypoint script
 COPY --chown=user:user entrypoint.sh /app/entrypoint.sh
 RUN chmod +x /app/entrypoint.sh
 
-# 8. Switch to the non-root user
 USER user
-
-# Hugging Face exposes port 7860 by default
 ENV PORT=7860
 
-# 9. Start via the sync manager script
 ENTRYPOINT ["/app/entrypoint.sh"]

entrypoint.sh

@@ -4,18 +4,22 @@
 APP_STORAGE="/app/storage"
 PERSISTENT_STORAGE="/data/storage"
 
-# Set the mounted file storage environment variable for ApexKit
+# ApexKit persistent file/database storage
 export APEXKIT_MOUNTED_FILE_STORAGE="/data/"
 
+# Force WASM engine to strictly use the fast local NVMe cache in /app
+export APEXKIT_WASM_CACHE_DIR="/app/.cache/wasm"
+
 echo "[Init] Setting up Hugging Face Storage Sync..."
-echo "[Init] Set APEXKIT_MOUNTED_FILE_STORAGE to ${APEXKIT_MOUNTED_FILE_STORAGE}"
+echo "[Init] Persistent File Storage: ${APEXKIT_MOUNTED_FILE_STORAGE}"
+echo "[Init] Fast NVMe WASM Cache:   ${APEXKIT_WASM_CACHE_DIR}"
 
 # 1. Restore from persistent storage on boot (EXCLUDING uploads)
 if [ -d "$PERSISTENT_STORAGE" ] && [ "$(ls -A $PERSISTENT_STORAGE 2>/dev/null)" ]; then
     echo "[Sync] Restoring data from persistent storage ($PERSISTENT_STORAGE) to NVMe ($APP_STORAGE)..."
     mkdir -p "$APP_STORAGE"
     
-    # Exclude uploads so we don't pull large media files down to the ephemeral disk
+    # Exclude uploads so large media files remain directly on persistent disk
     rsync -a --exclude="**/uploads/" "$PERSISTENT_STORAGE/" "$APP_STORAGE/"
     echo "[Sync] Restore complete."
 else
@@ -24,42 +28,37 @@ else
     mkdir -p "$PERSISTENT_STORAGE"
 fi
 
-# Inject Replica ID if provided via Env Vars (Survives HF Space Restarts)
+# Inject Replica ID if provided via Env Vars
 if [ -n "$APEX_REPLICA_ID" ]; then
     mkdir -p "$APP_STORAGE/system"
     echo "$APEX_REPLICA_ID" > "$APP_STORAGE/system/.replica_id"
     echo "[Init] Injected static Replica ID: $APEX_REPLICA_ID"
 fi
 
-# 2. Background Sync Loop (Every 20 minutes = 1200 seconds)
+# 2. Background Sync Loop (Every 20 minutes)
 sync_data() {
     while true; do
         sleep 1200
         echo "[Sync] Backing up NVMe data to persistent storage..."
-        # --exclude protects the persistent uploads folder from being deleted by --delete
         rsync -a --delete --exclude="**/uploads/" "$APP_STORAGE/" "$PERSISTENT_STORAGE/"
         echo "[Sync] Backup complete."
     done
 }
 
-# Start sync loop in the background
 sync_data &
 SYNC_PID=$!
 
-# 3. Graceful Shutdown Handler (Catches Space restarts/pauses)
+# 3. Graceful Shutdown Handler
 cleanup() {
     echo "[Shutdown] Caught stop signal. Stopping application..."
     
-    # Terminate background services
     kill -TERM "$APP_PID" 2>/dev/null
     kill -TERM "$FASTAPI_PID" 2>/dev/null
     
-    # Wait for processes to exit
     wait "$APP_PID" 2>/dev/null
     wait "$FASTAPI_PID" 2>/dev/null
     
     echo "[Shutdown] Performing final data sync to persistent storage..."
-    # --exclude protects the persistent uploads folder from being deleted by --delete
     rsync -a --delete --exclude="**/uploads/" "$APP_STORAGE/" "$PERSISTENT_STORAGE/"
     echo "[Shutdown] Final sync complete. Safely exiting."
     
@@ -67,25 +66,21 @@ cleanup() {
     exit 0
 }
 
-# Trap termination signals to trigger the cleanup function
 trap cleanup SIGINT SIGTERM
 
-# 4. Start the FastAPI background server on port 8000
-# Run on localhost (127.0.0.1) if only ApexKit needs to communicate with it internally,
-# or 0.0.0.0 if other containers or processes on the host network need access.
+# 4. Start FastAPI background server on port 8000
 echo "[App] Starting FastAPI on port 8000..."
 uvicorn main:app --host 127.0.0.1 --port 8000 --workers 1 &
 FASTAPI_PID=$!
 
-# 5. Start the main ApexKit application on HF exposed port
+# 5. Start ApexKit on the exposed port
 echo "[App] Starting ApexKit on port ${PORT:-7860}..."
 apexkit --port ${PORT:-7860} &
 APP_PID=$!
 
-# Wait for the main application to exit
+# Wait for main process
 wait "$APP_PID"
 
-# If the app crashes or exits naturally, do a final sync before the container dies
 echo "[App] Application exited. Syncing data before container stops..."
 kill -TERM "$FASTAPI_PID" 2>/dev/null
 rsync -a --delete --exclude="**/uploads/" "$APP_STORAGE/" "$PERSISTENT_STORAGE/"