Reranker and Debug Added - Optimized Dockerfile

Dockerfile

@@ -1,14 +1,47 @@
-FROM python:3.11-slim
+# ============================================================
+#  WorkWise Backend - Lightweight (GPU Optional) Build
+# ============================================================
 
-WORKDIR /app
+# ---- 1️⃣ Base Image ----------------------------------------------------------
+# Use CPU-friendly base by default; uncomment CUDA base if you need GPU
+FROM python:3.10-slim AS base
+# FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 AS base   # (optional GPU)
 
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+ENV DEBIAN_FRONTEND=noninteractive \
+    PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=1 \
+    PYTHONPATH=/home/user/app
 
-COPY . /app
-#COPY app/ ./app/
-#COPY data/ ./data/
+# ---- 2️⃣ System Dependencies -------------------------------------------------
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        git curl build-essential libgl1 libglib2.0-0 ffmpeg \
+        && rm -rf /var/lib/apt/lists/*
 
-EXPOSE 7860
+# ---- 3️⃣ Create Non-Root User ----------------------------------------------
+RUN useradd -m -u 1000 user
+WORKDIR /home/user/app
+USER user
+
+# ---- 4️⃣ Install Python Dependencies ----------------------------------------
+COPY --chown=user:user requirements.txt .
+RUN pip install --upgrade pip \
+    && pip install -r requirements.txt \
+    && pip install sentence-transformers faiss-cpu gradio[oauth] uvicorn fastapi \
+    && pip uninstall -y torch torchvision torchaudio torchtext torchdata 2>/dev/null || true
+
+# ---- 5️⃣ Copy Application Source --------------------------------------------
+COPY --chown=user:user . .
 
+# ---- 6️⃣ Environment Variables ----------------------------------------------
+ENV HF_HOME=/home/user/.cache/huggingface \
+    TRANSFORMERS_CACHE=/home/user/.cache/huggingface/transformers \
+    OMP_NUM_THREADS=4 \
+    TOKENIZERS_PARALLELISM=false \
+    HF_HUB_DISABLE_SYMLINKS_WARNING=1
+
+# ---- 7️⃣ Healthcheck ---------------------------------------------------------
+HEALTHCHECK CMD curl -f http://localhost:7860/health || exit 1
+
+# ---- 8️⃣ Entry Point ---------------------------------------------------------
+EXPOSE 7860
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

Dockerfile.bak

@@ -0,0 +1,14 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . /app
+#COPY app/ ./app/
+#COPY data/ ./data/
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]