Update Dockerfile

Dockerfile

@@ -1,31 +1,27 @@
-# Configure image
-ARG PYTHON_VERSION=3.10
+FROM node:18-slim
 
-FROM python:${PYTHON_VERSION}-slim
-ARG PYTHON_VERSION
-ARG DEBIAN_FRONTEND=noninteractive
-
-# Install apt dependencies
+# Install git and clean up in one step to keep the image size down
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential cmake git wget \
-    libglib2.0-0 libgl1-mesa-glx libegl1-mesa ffmpeg \
-    && apt-get clean && rm -rf /var/lib/apt/lists/*
-
-# Create virtual environment
-RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
-RUN python -m venv /opt/venv
-ENV PATH="/opt/venv/bin:$PATH"
-RUN echo "source /opt/venv/bin/activate" >> /root/.bashrc
-
-RUN useradd -m -u 1000 user
-
-# Install LeRobot
-RUN git clone https://github.com/huggingface/lerobot.git /lerobot
-WORKDIR /lerobot
-RUN pip install --upgrade --no-cache-dir pip
-RUN pip install --no-cache-dir "." \
-    --extra-index-url https://download.pytorch.org/whl/cpu
-RUN pip install --no-cache-dir flask
-
-COPY --chown=user . /lerobot
-CMD ["python", "lerobot/scripts/visualize_dataset_html.py", "--host", "0.0.0.0", "--port", "7860"]
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
+
+# Clone the repository with depth 1 (shallow clone)
+RUN git clone --depth 1 -b update_html_visualizer https://github.com/mishig25/lerobot.git .
+
+# Change to the HTML visualizer directory
+WORKDIR /app/lerobot/html_dataset_visualizer
+
+# Install production dependencies only
+RUN npm ci --production
+
+# Expose port 7860
+EXPOSE 7860
+
+# Set environment variable for port
+ENV PORT=7860
+
+# Start the application
+CMD ["npm", "start"]