setup.sh

#!/usr/bin/env bash
# onwx - LLMs on the AMD Ryzen AI NPU: install the resident top-bar app (Ubuntu).
# Installs the system pieces the indicator and the window need (AppIndicator, WebKit; asks for sudo once), creates .venv,
# installs onwx + the tray, adds "onwx" to the application list and starts it in the top bar.
# GNOME shows indicators through the "Ubuntu AppIndicators" extension, which Ubuntu enables by default.
set -e
cd "$(dirname "$0")"
export PYTHONUTF8=1

if ! python3 -c "import sys; sys.exit(0 if sys.version_info >= (3, 10) else 1)" 2>/dev/null; then
    echo "Python 3.10 or newer is needed (Ubuntu 22.04 and later ship it):  sudo apt install python3"
    exit 1
fi

need=""
# the window is a WebKit web view (pywebview): 4.1 on Ubuntu 24.04 and later, 4.0 before
webkit=gir1.2-webkit2-4.1
apt-cache show $webkit >/dev/null 2>&1 || webkit=gir1.2-webkit2-4.0
# wl-clipboard / xclip: "Copy the OpenAI API URL" in the top-bar menu (Wayland / X11)
for p in python3-venv python3-gi gir1.2-ayatanaappindicator3-0.1 $webkit wl-clipboard xclip; do
    dpkg -s "$p" >/dev/null 2>&1 || need="$need $p"
done
if [ -n "$need" ]; then
    echo "[setup] installing system packages:$need"
    sudo apt-get install -y $need
fi

install_onwx() {
    # the bundled pip is fine (a newer pip once failed to install packages that have scripts, so it is not upgraded)
    .venv/bin/python -m pip install -e ".[tray]"
}
# pip ran for this pyproject.toml already (onwx.deps): nothing to install. Otherwise - new packages from an update, an
# update whose pip could not finish, an older install - pip runs again (a running onwx is stopped first, and started
# again at the end).
if [ ! -f .venv/bin/python ] || ! .venv/bin/python -m onwx.deps check >/dev/null 2>&1; then
    if [ ! -f .venv/bin/python ]; then
        echo "[setup] creating Python environment in .venv ..."
        # system site packages: the GObject bindings for the indicator and WebKit come from apt (python3-gi)
        python3 -m venv --system-site-packages .venv
    fi
    .venv/bin/python -m onwx.deps stop || true
    if ! install_onwx; then
        echo "[setup] installation failed; rebuilding the Python environment once ..."
        rm -rf .venv
        python3 -m venv --system-site-packages .venv
        install_onwx || exit 1
    fi
    .venv/bin/python -m onwx.deps done
    echo ok > .venv/installed.txt
fi

# the NPU from Python: XRT's bindings (pyxrt) - say now if they are missing or built for another Python
.venv/bin/python -c "from onwx.xdna import check_pyxrt; check_pyxrt()" || echo "[onwx] the NPU cannot be used until this is fixed (onwx still starts; models will not load)"

mkdir -p ~/.local/share/applications
cat > ~/.local/share/applications/onwx.desktop <<EOF
[Desktop Entry]
Type=Application
Name=onwx
Comment=LLMs on the AMD Ryzen AI NPU
Exec="$PWD/.venv/bin/onwx" tray
Path=$PWD
Icon=$PWD/onwx/web/icon.png
Terminal=false
Categories=Utility;
StartupWMClass=onwx
EOF
echo "[setup] application list: onwx"

echo "[start] onwx is starting in the top bar ..."
nohup .venv/bin/onwx tray >/dev/null 2>&1 &