FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 | |
# Set environment variables | |
ENV DEBIAN_FRONTEND=noninteractive \ | |
PYTHONUNBUFFERED=1 \ | |
PYTHONDONTWRITEBYTECODE=1 \ | |
PORT=7860 \ | |
BASE_DIR=/app \ | |
NLTK_DATA=/app/nltk_data | |
# Install Python and other dependencies | |
RUN apt-get update && apt-get install -y --no-install-recommends \ | |
python3 \ | |
python3-pip \ | |
python3-dev \ | |
build-essential \ | |
git \ | |
gcc \ | |
g++ \ | |
make \ | |
curl \ | |
&& rm -rf /var/lib/apt/lists/* | |
# Create a non-root user | |
RUN useradd -m -u 1000 user | |
# Set working directory | |
WORKDIR /app | |
# Create necessary directories and set permissions | |
RUN mkdir -p /app/data/pdfs /app/data/texts /app/data/index /app/nltk_data && \ | |
chown -R user:user /app | |
# Copy requirements first for better caching | |
COPY --chown=user:user requirements.txt . | |
# Switch to non-root user | |
USER user | |
# Install Python packages | |
RUN pip3 install --no-cache-dir --user -r requirements.txt | |
RUN pip3 install --no-cache-dir --user gradio | |
# Copy your application code and data | |
COPY --chown=user:user app/ /app/ | |
COPY --chown=user:user data/ /app/data/ | |
# Copy your Gradio interface | |
COPY --chown=user:user spaces_app.py /app/ | |
# Expose the port Hugging Face Spaces expects | |
EXPOSE 7860 | |
# Health check | |
HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \ | |
CMD curl -f http://localhost:${PORT}/health || exit 1 | |
# Command to run the application | |
CMD ["python3", "spaces_app.py"] |