layout_uv() {
    if [[ -d ".venv" ]]; then
        VIRTUAL_ENV="$(pwd)/.venv"
    fi

    if [[ -z $VIRTUAL_ENV || ! -d $VIRTUAL_ENV ]]; then
        log_status "No uv project exists. Executing \`uv init\` to create one."
        uv init
        uv venv
        VIRTUAL_ENV="$(pwd)/.venv"
    fi

    PATH_add "$VIRTUAL_ENV/bin"
    export UV_ACTIVE=1  # or VENV_ACTIVE=1
    export VIRTUAL_ENV
}

layout_uv


# Path to your virtual environment directory
VENV_DIR="./venv"

# Load and activate the virtual environment if it exists
if [[ -d "$VENV_DIR" ]]; then
  layout python "$VENV_DIR"
else
  echo "Virtual environment not found at $VENV_DIR"
fi