#!/bin/bash
JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"

echo "Starting Jupyter Lab with token $JUPYTER_TOKEN"

NOTEBOOK_DIR="/data"
cp -n data/hf_spark_utils.py $NOTEBOOK_DIR/hf_spark_utils.py
cp -n data/spark.ipynb $NOTEBOOK_DIR/spark.ipynb
DEFAULT_URL="/lab/tree/spark.ipynb"

jupyter-lab \
    --ip 0.0.0.0 \
    --port 7860 \
    --no-browser \
    --allow-root \
    --ServerApp.token="$JUPYTER_TOKEN" \
    --ServerApp.tornado_settings="{'headers': {'Content-Security-Policy': 'frame-ancestors *'}}" \
    --ServerApp.cookie_options="{'SameSite': 'None', 'Secure': True}" \
    --ServerApp.disable_check_xsrf=True \
    --LabApp.news_url=None \
    --LabApp.check_for_updates_class="jupyterlab.NeverCheckForUpdate" \
    --LabApp.default_url=$DEFAULT_URL \
    --notebook-dir=$NOTEBOOK_DIR