Update Dockerfile

Dockerfile

@@ -1,41 +1,48 @@
-FROM python:3.10-slim
+FROM python:3.10-slim AS builder
 
-WORKDIR /app
+ENV DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies
+# Install build tools
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential \
-    libgl1 \
-    libglib2.0-0 \
-    ffmpeg \
-    libsm6 \
-    wget \
+    build-essential cmake git curl \
     && rm -rf /var/lib/apt/lists/*
 
-# Environment variables for EasyOCR
-ENV EASYOCR_MODULE_PATH=/app/.EasyOCR
-ENV EASYOCR_CACHE_DIR=/app/.EasyOCR
-
-# Create EasyOCR dir
-RUN mkdir -p /app/.EasyOCR && chmod -R 777 /app/.EasyOCR
+WORKDIR /app
 
-# Copy requirements first
+# Copy and install Python dependencies
 COPY requirements.txt .
+RUN python -m venv /opt/venv && \
+    /opt/venv/bin/pip install --upgrade pip && \
+    /opt/venv/bin/pip install --no-cache-dir -r requirements.txt
+
+# Pre-download YOLOv8 model weights into writable path
+RUN mkdir -p /app/models && \
+    curl -L -o /app/models/yolov8n.pt \
+    https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt
+
+# -------- RUNTIME IMAGE --------
+FROM python:3.10-slim
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PATH="/opt/venv/bin:$PATH"
+ENV EASYOCR_MODULE_PATH=/app/.EasyOCR
+ENV EASYOCR_CACHE_DIR=/app/.EasyOCR
+ENV YOLO_CONFIG_DIR=/app/.yolo
 
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+# Install minimal OpenCV runtime dependencies
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    libgl1 libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
 
-# Pre-download EasyOCR English model
-RUN python3 -c "import easyocr; easyocr.Reader(['en'])"
+# Copy venv and preloaded model
+COPY --from=builder /opt/venv /opt/venv
+COPY --from=builder /app/models/yolov8n.pt /app/models/yolov8n.pt
 
-# Pre-download YOLOv8n model into /app
-RUN wget -O /app/yolov8n.pt https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt
+# Create writable dirs for EasyOCR & YOLO configs
+RUN mkdir -p /app/.EasyOCR /app/.yolo && chmod -R 777 /app/.EasyOCR /app/.yolo
 
-# Copy app code
+WORKDIR /app
 COPY . .
 
-# Expose port
 EXPOSE 7860
-
-# Run the app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]