Update Dockerfile

Dockerfile

@@ -12,7 +12,6 @@ RUN npm install --frozen-lockfile
 COPY frontend/ ./
 RUN npm run build
 
-# --- Stage 2: GPU-enabled Python backend ---
 # --- Stage 2: GPU-enabled Python backend ---
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04 AS backend
 
@@ -23,7 +22,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
     PIP_NO_CACHE_DIR=1 \
     HF_HOME=/app/.cache/huggingface
 
-# Install Python and system dependencies BEFORE switching user
+# Install system dependencies BEFORE switching user
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3 python3-pip python3-venv git curl && \
     rm -rf /var/lib/apt/lists/*
@@ -31,17 +30,16 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Create non-root user
 RUN useradd -m appuser
 
-# Create cache directory and set permissions
-RUN mkdir -p /app/.cache/huggingface && \
-    chown -R appuser:appuser /app/.cache
+# Create necessary directories and set permissions
+RUN mkdir -p /app/.cache/huggingface \
+    && mkdir -p /app/static \
+    && chown -R appuser:appuser /app
 
 # Switch to non-root user
 USER appuser
 
 WORKDIR /app
 
-# Continue with pip installs and app setup...
-
 # Upgrade pip and install CUDA-enabled PyTorch
 RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install --index-url https://download.pytorch.org/whl/cu121 torch==2.4.1+cu121
@@ -54,7 +52,6 @@ RUN python3 -m pip install -r /app/backend/requirements.txt
 COPY backend/ /app/backend/
 
 # Copy frontend build to static directory
-RUN mkdir -p /app/static
 COPY --from=frontend /app/frontend/dist/ /app/static/
 
 # App-specific environment variables
@@ -70,4 +67,4 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=10s --retries=3 \
 EXPOSE 8000
 
 # Start FastAPI app
-CMD ["uvicorn", "app.main:app", "--app-dir", "/app/backend", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "app.main:app", "--app-dir", "/app/backend", "--host", "0.0.0.0", "--port", "8000"]