Prepare job build context

deploy.sh

@@ -3,7 +3,8 @@
 
 set -e
 
-IMAGE_NAME="gpue/foundationpose-base-l2"
+IMAGE_NAME_L1="gpue/foundationpose-base-l1"
+IMAGE_NAME_L2="gpue/foundationpose-base-l2"
 TAG="latest"
 PLATFORM="linux/amd64"
 HF_SPACE="gpue/foundationpose"
@@ -65,17 +66,17 @@ git push "https://huggingface.co/spaces/${HF_SPACE}" "HEAD:${JOB_REF}" --force
 echo "✓ Job ref pushed"
 echo ""
 
-echo "Stage 1: Building base image via HF Job"
+echo "Stage 1: Building L1 base image via HF Job"
 echo "Platform: ${PLATFORM}"
-echo "Image: ${IMAGE_NAME}:${TAG}"
+echo "Image: ${IMAGE_NAME_L1}:${TAG}"
 echo ""
 
 JOB_OUTPUT=$("${PY_BIN}" scripts/run_hf_image_job.py \
-    --image-name "${IMAGE_NAME}" \
+    --image-name "${IMAGE_NAME_L1}" \
     --tag "${TAG}" \
     --platform "${PLATFORM}" \
     --dockerfile "Dockerfile.base" \
-    --target "foundationpose-base-l2" \
+    --target "foundationpose-base-l1" \
     --flavor "l40sx1" \
     --git-repo "https://huggingface.co/spaces/${HF_SPACE}" \
     --git-ref "${JOB_REF}" 2>&1 | tee /tmp/hf_image_job.log)
@@ -115,7 +116,7 @@ fi
 
 if [ -n "${JOB_ID}" ] && [ -x "${HF_BIN}" ]; then
     echo ""
-    echo "Waiting for image build job to complete..."
+    echo "Waiting for L1 image build job to complete..."
     for i in $(seq 1 40); do
         JOB_STAGE=$("${HF_BIN}" jobs inspect "${JOB_ID}" | python3 -c "import sys, json; data=json.load(sys.stdin)[0]; print(data.get('status', {}).get('stage', 'UNKNOWN'))" 2>/dev/null || echo "UNKNOWN")
         echo "  Job stage: ${JOB_STAGE}"
@@ -136,7 +137,76 @@ if [ -n "${JOB_ID}" ] && [ -x "${HF_BIN}" ]; then
 fi
 
 echo ""
-echo "Stage 2: Deploying to HuggingFace Space"
+echo "Stage 2: Building L2 base image via HF Job"
+echo ""
+
+JOB_OUTPUT_L2=$("${PY_BIN}" scripts/run_hf_image_job.py \
+    --image-name "${IMAGE_NAME_L2}" \
+    --tag "${TAG}" \
+    --platform "${PLATFORM}" \
+    --dockerfile "Dockerfile.base" \
+    --target "foundationpose-base-l2" \
+    --flavor "l40sx1" \
+    --git-repo "https://huggingface.co/spaces/${HF_SPACE}" \
+    --git-ref "${JOB_REF}" 2>&1 | tee /tmp/hf_image_job_l2.log)
+
+JOB_ID_L2=$(echo "${JOB_OUTPUT_L2}" | awk '/Job ID:/ {print $3}')
+if [ -z "${JOB_ID_L2}" ]; then
+    echo "Warning: Could not parse HF job id for L2. See /tmp/hf_image_job_l2.log"
+else
+    echo "Following L2 job logs for 1 minute..."
+    if [ -x "${HF_BIN}" ]; then
+        HF_BIN_PATH="${HF_BIN}" JOB_ID="${JOB_ID_L2}" "${PY_BIN}" - <<'PY'
+import os
+import subprocess
+import sys
+import time
+
+hf = os.environ["HF_BIN_PATH"]
+job_id = os.environ["JOB_ID"]
+
+proc = subprocess.Popen([hf, "jobs", "logs", job_id], stdout=sys.stdout, stderr=sys.stderr)
+try:
+    time.sleep(60)
+finally:
+    proc.terminate()
+    try:
+        proc.wait(timeout=5)
+    except Exception:
+        proc.kill()
+PY
+        echo ""
+        echo "L2 job status:"
+        "${HF_BIN}" jobs inspect "${JOB_ID_L2}" || true
+    else
+        echo "hf CLI not available; job logs skipped"
+    fi
+fi
+
+if [ -n "${JOB_ID_L2}" ] && [ -x "${HF_BIN}" ]; then
+    echo ""
+    echo "Waiting for L2 image build job to complete..."
+    for i in $(seq 1 60); do
+        JOB_STAGE=$("${HF_BIN}" jobs inspect "${JOB_ID_L2}" | python3 -c "import sys, json; data=json.load(sys.stdin)[0]; print(data.get('status', {}).get('stage', 'UNKNOWN'))" 2>/dev/null || echo "UNKNOWN")
+        echo "  Job stage: ${JOB_STAGE}"
+        case "${JOB_STAGE}" in
+            SUCCESS|SUCCEEDED|COMPLETED|DONE)
+                echo "✓ L2 image build job completed"
+                break
+                ;;
+            FAILED|ERROR|CANCELED|CANCELLED)
+                echo "✗ L2 image build job failed: ${JOB_STAGE}"
+                exit 1
+                ;;
+            *)
+                sleep 30
+                ;;
+        esac
+    done
+fi
+
+echo ""
+echo "Stage 3: Deploying to HuggingFace Space"
 echo ""
 
 # Initialize git repo if needed
@@ -167,7 +237,7 @@ echo ""
 echo "✓ Pushed to HuggingFace"
 echo ""
 echo "HuggingFace will now:"
-echo "  1. Pull base image from DockerHub (${IMAGE_NAME}:${TAG})"
+echo "  1. Pull base image from DockerHub (${IMAGE_NAME_L2}:${TAG})"
 echo "  2. Build CUDA extensions"
 echo "  3. Download model weights"
 echo "  4. Start the Gradio app"