install_dpg.sh

#!/usr/bin/env bash
# Install Dear PyGui and its dependencies for OneTrainer

set -e

# Use python3 as backup if python command isn't available
if ! command -v python &> /dev/null && command -v python3 &> /dev/null; then
    export OT_PYTHON_CMD="python3"
fi

source "${BASH_SOURCE[0]%/*}/lib.include.sh"

prepare_runtime_environment

# Check if dearpygui is installed, install if missing
if ! run_python_in_active_env -c "import dearpygui" &> /dev/null; then
    echo "Dear PyGui is not installed. Installing now..."
    run_pip_in_active_env install dearpygui
fi

# Check if PIL/Pillow is installed, install if missing
if ! run_python_in_active_env -c "from PIL import Image" &> /dev/null; then
    echo "Pillow is not installed. Installing now..."
    run_pip_in_active_env install Pillow
fi

echo "All dependencies installed. You can now run the Dear PyGui UI with ./start-dpg-ui.sh"