Update setup_blender.sh

setup_blender.sh

@@ -1,19 +1,62 @@
 #!/bin/bash
 
-# Define the Blender version and download URL
+# Exit immediately if a command exits with a non-zero status.
+set -e
+
+# --- Configuration ---
 BLENDER_VERSION="4.2.0"
+BLENDER_MAJOR_MINOR="4.2" # Matches the version directory structure
+BLENDER_PYTHON_VERSION="python3.11" # Blender 4.2 uses Python 3.11
 BLENDER_TARBALL="blender-${BLENDER_VERSION}-linux-x64.tar.xz"
-BLENDER_URL="https://download.blender.org/release/Blender4.2/blender-${BLENDER_VERSION}-linux-x64.tar.xz"
+BLENDER_URL="https://download.blender.org/release/Blender${BLENDER_MAJOR_MINOR}/blender-${BLENDER_VERSION}-linux-x64.tar.xz"
 INSTALL_DIR="/opt/blender-${BLENDER_VERSION}-linux-x64"
+BLENDER_PY_EXEC="${INSTALL_DIR}/${BLENDER_MAJOR_MINOR}/python/bin/${BLENDER_PYTHON_VERSION}"
+# Assuming unirig_requirements.txt is in the root directory alongside this script and app.py
+# If UniRig repo is cloned first, adjust path e.g., "UniRig/unirig_requirements.txt"
+UNIRIG_REQS_FILE="unirig_requirements.txt"
+
+# --- Download and Extract Blender ---
+echo "Downloading Blender ${BLENDER_VERSION}..."
+wget -nv -O /tmp/${BLENDER_TARBALL} ${BLENDER_URL}
+echo "Download complete."
+
+echo "Extracting Blender to ${INSTALL_DIR}..."
+mkdir -p /opt # Ensure /opt exists
+tar -xJf /tmp/${BLENDER_TARBALL} -C /opt
+echo "Extraction complete."
+
+# --- Create Blender Symlink ---
+echo "Creating symlink for Blender executable..."
+ln -sf ${INSTALL_DIR}/blender /usr/local/bin/blender # Use -f to force overwrite if exists
+echo "Symlink created."
+
+# --- Install UniRig Dependencies into Blender's Python ---
+echo "Installing UniRig Python dependencies into Blender's Python (${BLENDER_PYTHON_VERSION})..."
+
+if [ ! -f "${BLENDER_PY_EXEC}" ]; then
+    echo "ERROR: Blender Python executable not found at ${BLENDER_PY_EXEC} after extraction!"
+    exit 1
+fi
+
+if [ ! -f "${UNIRIG_REQS_FILE}" ]; then
+    echo "ERROR: UniRig requirements file not found at ${UNIRIG_REQS_FILE}!"
+    echo "Ensure this file exists in the root directory."
+    exit 1
+fi
+
+# Upgrade pip within Blender's Python environment
+echo "Upgrading pip for Blender Python..."
+"${BLENDER_PY_EXEC}" -m pip install --upgrade pip setuptools wheel
 
-# Download Blender
-wget -O /tmp/${BLENDER_TARBALL} ${BLENDER_URL}
+# Install packages from unirig_requirements.txt
+echo "Installing packages from ${UNIRIG_REQS_FILE}..."
+"${BLENDER_PY_EXEC}" -m pip install --no-cache-dir -r "${UNIRIG_REQS_FILE}"
 
-# Extract to /opt
-tar -xvf /tmp/${BLENDER_TARBALL} -C /opt
+echo "UniRig dependency installation for Blender Python complete."
 
-# Create a symbolic link to the Blender executable
-ln -s ${INSTALL_DIR}/blender /usr/local/bin/blender
+# --- Cleanup ---
+echo "Cleaning up downloaded tarball..."
+rm /tmp/${BLENDER_TARBALL}
+echo "Cleanup complete."
 
-# Clean up the tarball
-rm /tmp/${BLENDER_TARBALL}
+echo "Blender setup finished successfully."