code

Dockerfile

@@ -15,6 +15,8 @@ RUN rm -f /etc/apt/sources.list.d/*.list && \
     aria2 \
     axel \
     procps \
+    fastapi \
+    uvicorn \
     git-lfs \
     zip \
     unzip \

main.py

@@ -0,0 +1,7 @@
+from fastapi import FastAPI
+
+app = FastAPI()
+
+@app.get("/")
+def read_root():
+    return {"message": "Hello, World!"}

start_server.sh

@@ -1,29 +1,16 @@
 #!/bin/bash
 JUPYTER_TOKEN="${JUPYTER_TOKEN:=huggingface}"
-nohup code-server --bind-addr 0.0.0.0:8080 /data > /tmp/code.log 2>&1 &
+nohup code-server --bind-addr 0.0.0.0:8080 /data >/tmp/code.log 2>&1 &
 sleep 5s
 ps aux | grep 'code-server' | grep -v 'grep' | awk '{print $2}' | xargs kill -9
 sed -i '/password/d' /home/user/.config/code-server/config.yaml
-echo "password: ${JUPYTER_TOKEN}" >> /home/user/.config/code-server/config.yaml
+echo "password: ${JUPYTER_TOKEN}" >>/home/user/.config/code-server/config.yaml
 cat /home/user/.config/code-server/config.yaml
 
 NOTEBOOK_DIR="/data"
 
-jupyter labextension disable "@jupyterlab/apputils-extension:announcements"
+nohup uvicorn main:app --host 0.0.0.0 --port 8080 >/data/start.log 2>&1 &
 
-nohup jupyter-lab \
-    --ip 0.0.0.0 \
-    --port 8080 \
-    --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" \
-    --notebook-dir=$NOTEBOOK_DIR > /data/jupyter.log 2>&1 &
-
-sleep 10s
-ps aux | grep 'jupyter' | grep -v 'grep' | awk '{print $2}' | xargs kill -9
-code-server --bind-addr 0.0.0.0:8080 /data > /data/start.log 2>&1 &
+sleep 20s
+ps aux | grep 'uvicorn' | grep -v 'grep' | awk '{print $2}' | xargs kill -9
+code-server --bind-addr 0.0.0.0:8080 /data >/data/start.log 2>&1 &