Update start.sh

start.sh

@@ -1,47 +1,29 @@
-# 1. Use an NVIDIA base image to ensure GPU drivers are available
-# (Even if running on CPU-only for now, this prepares for upgrade)
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+#!/bin/bash
 
-# 2. Prevent interactive prompts during build
-ENV DEBIAN_FRONTEND=noninteractive
+# 1. Start the Virtual Display (Fake Screen)
+export DISPLAY=:1
+Xvfb :1 -screen 0 1920x1080x24 &
+sleep 2
 
-# 3. Install dependencies for Chromium and KasmVNC
-RUN apt-get update && apt-get install -y \
-    wget \
-    xvfb \
-    libgbm1 \
-    libgtk-3-0 \
-    libnss3 \
-    libasound2 \
-    python3 \
-    dbus-x11 \
-    # Install FFMPEG for video processing capabilities
-    ffmpeg \
-    && rm -rf /var/lib/apt/lists/*
+# 2. Start KasmVNC (The Streaming Server)
+# This puts the video stream on port 7860
+vncserver -port 7860 \
+          -httpd /usr/share/kasmvnc/www \
+          -sslOnly 0 \
+          -depth 24 \
+          -geometry 1920x1080 \
+          -fg &
 
-# 4. Install KasmVNC (The secret sauce for video quality)
-RUN wget https://github.com/kasmtech/KasmVNC/releases/download/v1.1.0/kasmvncserver_focal_1.1.0_amd64.deb && \
-    apt-get install -y ./kasmvncserver_*.deb && \
-    rm kasmvncserver_*.deb
+# 3. Launch Chromium
+# We point to the custom build you copied earlier
+CHROME_BIN="/opt/chromium/chrome"
 
-# 5. COPY YOUR CUSTOM CHROMIUM BUILD
-# You must copy the custom binary we discussed to get H.264/HEVC support.
-# Assuming your build is in a local folder named 'chrome-linux'
-COPY ./chrome-linux /opt/chromium
-RUN chmod +x /opt/chromium/chrome
-
-# 6. Set up a user (Hugging Face prefers non-root users usually, but for X11 root is easier to debug initially. 
-# Let's create a standard user 'user' with ID 1000)
-RUN useradd -m -u 1000 user
-USER user
-ENV HOME=/home/user
-WORKDIR $HOME
-
-# 7. Configure Chromium Flags for Docker
-# --no-sandbox is required in Docker
-# --disable-dev-shm-usage prevents crashing on low memory
-ENV CHROME_OPTS="--no-sandbox --disable-dev-shm-usage --start-maximized --user-data-dir=/data"
-
-# 8. Start Script
-COPY start.sh /start.sh
-CMD ["/bin/bash", "/start.sh"]
+$CHROME_BIN \
+  --no-sandbox \
+  --disable-dev-shm-usage \
+  --start-maximized \
+  --user-data-dir=/home/user/data \
+  --enable-gpu \
+  --enable-unsafe-webgpu \
+  --ignore-gpu-blocklist \
+  https://huggingface.co