Update Dockerfile

Dockerfile

@@ -2,25 +2,24 @@
 FROM python:3.10-slim AS builder
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install build dependencies
+# Install build deps + OpenCV runtime libs
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential cmake git curl \
+    build-essential cmake git curl libgl1 libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
-
-# Install Python dependencies in a venv
 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 YOLOv8n weights to Ultralytics cache
+# ✅ Pre-download YOLOv8n weights
 RUN mkdir -p /root/.cache/torch/hub/checkpoints && \
     curl -L -o /root/.cache/torch/hub/checkpoints/yolov8n.pt \
     https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt
 
-# ✅ Pre-download EasyOCR model files
+# ✅ Pre-cache EasyOCR models into /app/.EasyOCR
 RUN mkdir -p /app/.EasyOCR && \
     /opt/venv/bin/python -c "import easyocr; easyocr.Reader(['en'], model_storage_directory='/app/.EasyOCR')"
 
@@ -29,29 +28,71 @@ FROM python:3.10-slim
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PATH="/opt/venv/bin:$PATH"
 
-# ✅ Force writable cache dirs
+# ✅ Fix EasyOCR cache location
 ENV EASYOCR_MODULE_PATH=/app/.EasyOCR
 ENV EASYOCR_CACHE_DIR=/app/.EasyOCR
 ENV YOLO_CONFIG_DIR=/app/.yolo
 
-# Runtime deps (OpenCV etc.)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy Python env + caches from builder
 COPY --from=builder /opt/venv /opt/venv
 COPY --from=builder /root/.cache /root/.cache
 COPY --from=builder /app/.EasyOCR /app/.EasyOCR
 
 WORKDIR /app
+RUN mkdir -p /app/.yolo && chmod -R 777 /app && chmod -R 777 /root/.cache
+
+# ✅ app.py baked in
+COPY <<EOF /app/app.py
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from fastapi.middleware.cors import CORSMiddleware
+import numpy as np
+from PIL import Image
+from io import BytesIO
+import easyocr
+from ultralytics import YOLO
+
+app = FastAPI()
+
+@app.get("/")
+async def root():
+    return {"message": "YOLO + EasyOCR API is running!"}
+
+reader = easyocr.Reader(['en'], model_storage_directory='/app/.EasyOCR')
+model = YOLO("yolov8n.pt")
+print("✅ YOLO model loaded")
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], allow_credentials=True,
+    allow_methods=["*"], allow_headers=["*"],
+)
+
+@app.post("/detect")
+async def detect_plate(file: UploadFile = File(...)):
+    try:
+        contents = await file.read()
+        image = Image.open(BytesIO(contents)).convert("RGB")
+        image_np = np.array(image)
+
+        results = model(image_np)[0]
+        plate_texts = []
+
+        for box in results.boxes:
+            x1, y1, x2, y2 = map(int, box.xyxy[0])
+            plate_crop = image_np[y1:y2, x1:x2]
+            text = reader.readtext(plate_crop, detail=0)
+            if text:
+                plate_texts.extend(text)
 
-# ✅ Create cache dirs & fix permissions
-RUN mkdir -p /app/.yolo && \
-    chmod -R 777 /app && \
-    chmod -R 777 /root/.cache
+        return JSONResponse(content={"plates": plate_texts})
 
-COPY . .
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)
+EOF
 
 EXPOSE 7860
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]