# Use a multi-stage build to reduce final image size
# Start with a builder image
FROM python:3.10-slim as builder

# Set a working directory
WORKDIR /app

# Install dependencies in a virtual environment to make it easier to copy only what we need later
RUN python -m venv /venv
ENV PATH="/venv/bin:$PATH"

# Copy only requirements.txt initially to leverage Docker cache
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Continue with the final base image
FROM tiangolo/uvicorn-gunicorn-fastapi:python3.10-slim

# Install curl
RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*



# Copy the virtual environment from the builder stage
COPY --from=builder /venv /venv
ENV PATH="/venv/bin:$PATH"

# Set working directory
WORKDIR /app

# Copy the rest of the application from the current directory to /app inside the container
COPY . .


# Command to run the Uvicorn server using check_health.sh to wait for the model_server
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]