test

Dockerfile

@@ -9,12 +9,21 @@ RUN mkdir -p /var/lib/apt/lists/partial && \
 
 WORKDIR /work
 
-# Your conda env file is in binder/environment.yml inside the repo
+# 1) Create env first (better cache)
 COPY binder/environment.yml /tmp/environment.yml
 RUN micromamba env create -f /tmp/environment.yml && micromamba clean -a -y
 
-# Build StochasticCIL
-RUN git clone https://github.com/epapoutsellis/Split-Skip-and-Play.git && git clone https://github.com/epapoutsellis/StochasticCIL.git && \
+# 2) Copy the WHOLE Space repo into the image
+# This is what makes notebooks/scripts appear in JupyterLab
+COPY . /work
+
+# 3) Install Jupyter config + start script
+RUN mkdir -p /etc/jupyter && \
+    cp /work/jupyter_server_config.py /etc/jupyter/jupyter_server_config.py && \
+    chmod +x /work/start.sh
+
+# 4) (Optional) Build StochasticCIL (keep if you need it in the Space)
+RUN git clone https://github.com/epapoutsellis/StochasticCIL.git && \
     cd StochasticCIL && \
     git fetch --all --tags && \
     git checkout svrg && \
@@ -33,8 +42,5 @@ RUN git clone https://github.com/epapoutsellis/Split-Skip-and-Play.git && git cl
     make -j"$(nproc)" && \
     make install
 
-COPY start.sh /start.sh
-RUN chmod +x /start.sh
-
 EXPOSE 7860
-CMD ["/start.sh"]
+CMD ["/work/start.sh"]

jupyter_server_config.py

@@ -0,0 +1,34 @@
+import os
+
+PORT = int(os.environ.get("PORT", "7860"))
+TOKEN = os.environ.get("JUPYTER_TOKEN", "")
+
+c.ServerApp.ip = "0.0.0.0"
+c.ServerApp.port = PORT
+c.ServerApp.open_browser = False
+c.ServerApp.allow_root = True
+c.ServerApp.allow_remote_access = True
+
+# Show your repo contents
+c.ServerApp.root_dir = "/work"
+
+# Make the Space load JupyterLab directly (instead of a blank / page)
+# Change this path to your repo folder name if needed.
+c.ServerApp.default_url = "/lab/tree/Split-Skip-and-Play"
+
+# Allow embedding in the Hugging Face Space page iframe
+c.ServerApp.tornado_settings = {
+    "headers": {
+        "Content-Security-Policy": "frame-ancestors *"
+    }
+}
+
+# Cookies in an iframe need SameSite=None; Secure
+c.ServerApp.cookie_options = {"SameSite": "None", "Secure": True}
+
+# In iframe contexts, browsers sometimes drop the xsrf cookie; this avoids kernel-start failures.
+# IMPORTANT: keep a token (JUPYTER_TOKEN) if your Space is public.
+c.ServerApp.disable_check_xsrf = True
+
+# Require token if provided (set this in Space Settings -> Secrets)
+c.ServerApp.token = TOKEN

start.sh

@@ -1,12 +1,8 @@
 #!/usr/bin/env bash
 set -euo pipefail
 
-TOKEN="${JUPYTER_TOKEN:-}"
+# HF sets PORT; fallback to 7860
+export PORT="${PORT:-7860}"
 
 exec micromamba run -n ssp jupyter lab \
-  --allow-root \
-  --ip=0.0.0.0 --port=7860 \
-  --no-browser \
-  --ServerApp.allow_remote_access=True \
-  --ServerApp.root_dir=/work \
-  --IdentityProvider.token="${TOKEN}"
+  --config=/etc/jupyter/jupyter_server_config.py