# Use a Python base image
FROM  nvidia/cuda:12.4.1-runtime-ubuntu22.04

# Set the working directory
WORKDIR /DocQA

# Copy the requirements.txt file
COPY ./requirements.txt ./

# Install dependencies
RUN apt-get update && apt-get install -y \
    git \
    python3.10 \
    python3-pip \
    && apt-get clean
RUN apt-get install poppler-utils -y
RUN pip install --no-cache-dir -r requirements.txt 
RUN pip3 install torch --index-url https://download.pytorch.org/whl/cu121

# Copy the rest of the application codeDocQA
COPY images ./images
COPY app.py ./
COPY classication.py ./
COPY donut_inference.py ./
COPY non_form_llama_parse.py ./
COPY RAG.py ./
COPY best_resnet152_model.h5 ./
COPY Model ./

# Expose the port the app runs on
# EXPOSE 7860
EXPOSE 8501

# Start the application
# CMD ["streamlit", "run", "app.py"]
ENTRYPOINT ["streamlit", "run","app.py"]