install-dearpygui.sh

#!/usr/bin/env bash

# Install Dear PyGui dependencies for the experimental UI

echo "[OneTrainer] Installing Dear PyGui dependencies..."

# Check for existing Python environment
if [ -d "venv" ]; then
    echo "[OneTrainer] Using existing Python virtual environment..."
    source venv/bin/activate
elif command -v conda &> /dev/null && [ -d "conda_env" ]; then
    echo "[OneTrainer] Using existing Conda environment..."
    conda run --prefix conda_env --no-capture-output pip install -r requirements-dearpygui.txt
    exit 0
else
    echo "[OneTrainer] No Python environment found. Please run install.sh first."
    exit 1
fi

# Install Dear PyGui dependencies
pip install -r requirements-dearpygui.txt

echo "[OneTrainer] Dear PyGui dependencies installed successfully."
echo "[OneTrainer] You can now run the experimental UI with: python start-dpg-ui.py"