FROM nvidia/cuda:12.1.0-base-ubuntu22.04

WORKDIR /app

# Install Python and dependencies
RUN apt-get update && apt-get install -y \
    python3.11 \
    python3-pip \
    libgomp1 \
    && rm -rf /var/lib/apt/lists/*

# Install Python packages (including attachment parsing)
RUN pip install --no-cache-dir \
    sentence-transformers \
    torch \
    chromadb \
    PyPDF2 \
    pdfminer.six \
    python-docx \
    openpyxl \
    python-pptx \
    transformers

# Copy scripts
COPY business_rag.py /app/
COPY query.py /app/

ENTRYPOINT ["python3", "/app/business_rag.py"]