Update Dockerfile

Dockerfile

@@ -11,10 +11,14 @@ ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0 10.0 12.0"
 # Install dependencies
 RUN apt-get update && apt-get install --no-install-recommends -y \
     git \
+    git-lfs \
     curl \
     build-essential \
     cmake \
     wget \
+    procps \
+    vim \
+    nano \
     python3.10 \
     python3-pip \
     python3-dev \
@@ -43,41 +47,29 @@ RUN curl -sL https://deb.nodesource.com/setup_23.x -o nodesource_setup.sh && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+# Create Dev Mode compatible user
+RUN useradd -m -u 1000 user
+
+# Set working directory to /app
 WORKDIR /app
 
 # Set aliases for python and pip
 RUN ln -s /usr/bin/python3 /usr/bin/python
 
-# install pytorch before cache bust to avoid redownloading pytorch
-RUN pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128
-
-# Fix cache busting by moving CACHEBUST to right before git clone
-ARG CACHEBUST=1234
-ARG GIT_COMMIT=main
-RUN echo "Cache bust: ${CACHEBUST}" && \
-    git clone https://github.com/ostris/ai-toolkit.git && \
-    cd ai-toolkit && \
-    git checkout ${GIT_COMMIT}
-
-WORKDIR /app/ai-toolkit
+COPY --chown=1000 ./ /app
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt && \
     pip install --pre --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 --force && \
     pip install setuptools==69.5.1 --no-cache-dir
 
-# Build UI (generate temporary Prisma client for Next.js build)
-WORKDIR /app/ai-toolkit/ui
-RUN npm install && \
-    npx prisma generate --schema prisma/schema.prisma && \
-    npm run build
+RUN cd /app/ui && npm install
 
-# Expose port (assuming the application runs on port 3000)
 EXPOSE 8675
 
-WORKDIR /
-
-COPY start.sh /start.sh
-RUN chmod +x /start.sh
+WORKDIR /app
 
-CMD ["/start.sh"]
+RUN chown 1000 /app
+RUN chmod +x /app/start.sh
+USER 1000
+CMD ["/app/start.sh"]