Dockerfile

FROM python:3.10-slim

# Add build argument
ARG HUGGINGFACE_API_KEY
ENV HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}

# System dependencies
RUN apt-get update && apt-get install -y \
    build-essential \
    curl \
    git \
    && rm -rf /var/lib/apt/lists/*

# Set working directory
WORKDIR /app

# Copy requirements first for caching
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Copy project files
COPY . .

# # Environment variables
# ENV PYTHONUNBUFFERED=1
# ENV HUGGINGFACE_API_KEY=${HUGGINGFACE_API_KEY}

# HuggingFace login at container startup
# RUN --mount=type=secret,id=hf_token \
#     HUGGINGFACE_API_KEY=$(cat /run/secrets/hf_token) && \
#     huggingface-cli login --token $HUGGINGFACE_API_KEY

RUN mkdir -p ~/.huggingface && \
    echo "${HUGGINGFACE_API_KEY}" > ~/.huggingface/token && \
    huggingface-cli login --token ${HUGGINGFACE_API_KEY}

# Expose port
EXPOSE 7860

# Run the application
CMD ["python", "main.py"]