run-ui.sh

#!/bin/bash

echo "Version-Info, Demo UI Branch: ${GIT_BRANCH}"
echo "Version-Info, Demo UI Commit: ${GIT_COMMIT}"
echo "Version-Info, Demo UI Diff: ${GIT_DIFF}"

# run python proxy
if [[ -z "${OPENAI_API_KEY}" ]]; then
    echo "ERROR: Missing OPENAI_API_KEY; will not work"
    exit 1
fi

# FIXME: If we unify Dockerfile and git dirs, this will become simpler.
if [[ -f /.dockerenv ]]; then
    PROXYDIR=/home/pn/py-proxy
    UIDIR=/home/pn/js-ui
else
    cd $(dirname $0)
    DIR=$(pwd)
    PROXYDIR="${DIR}/openai-proxy"
    UIDIR="${DIR}/ui"
fi

function cleanup {
    kill "${PROXYPID}"
}

cd "${PROXYDIR}"
: ${HOST:=localhost}
if [[ (! -f ${HOST}-key.pem) || (! -f ${HOST}-cert.pem) ]]; then
    openssl req -batch -x509 -newkey rsa:4096 -days 10000 \
    -subj "/C=US/ST=California/O=Aryn.ai/CN=${HOST}" \
    -extensions v3_req -addext "subjectAltName=DNS:${HOST}" \
    -noenc -keyout "${HOST}-key.pem" -out "${HOST}-cert.pem" 2> /dev/null
    echo "Created ${HOST} certificate"
fi
if [[ ${SSL} == 0 ]]; then
    BASEURL="http://${HOST}:3000"
else
    BASEURL="https://${HOST}:3000"
fi

poetry run python py_proxy/proxy.py "${HOST}" &
PROXYPID=$!
trap cleanup EXIT

tries=0
while ! curl -k "${BASEURL}/healthz" >/dev/null 2>&1; do
    echo "Demo-UI proxy not running after $tries seconds"
    if [[ $tries -ge 30 ]]; then
        echo "Demo UI probably broken, report a bug on slack"
    fi
    tries=$(expr ${tries} + 1)
    sleep 1
done

# run node ui
cd "${UIDIR}"
# Running the UI this way means that the html is unminified and hence easy to read
# Since the UI is open source, there's little downside of doing this and it helps with
# debugging.
BROWSER=none PORT=3001 WDS_SOCKET_PORT=3000 HTTPS=true \
SSL_CRT_FILE="${PROXYDIR}/${HOST}-cert.pem" \
SSL_KEY_FILE="${PROXYDIR}/${HOST}-key.pem" \
npm start | cat

# These are the steps that would build the UI and serve it minified.  We
# should move the build back into the Dockerfile now that we no longer need
# to patch 'localhost' in the source.
#npm run build
#npx serve -n -s build