FROM tensorflow/tensorflow:1.14.0-gpu-py3

# Copy requirements first for efficient caching
COPY requirements.txt ./

# Update and install system dependencies
RUN echo 'deb [signed-by=/usr/share/keyrings/cloud-archive:ubuntu-archive-keyring.gpg] http://archive.ubuntu.com/ubuntu bionic main universe restricted' >> /etc/apt/sources.list.d/offending-repo.list \
    && apt-get update -o Acquire::AllowInsecureRepositories=true \
    && apt-get install -y libglib2.0-0 libgl1-mesa-glx ffmpeg \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Upgrade pip and install python packages
RUN pip install --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt \
    && pip install jupyter

WORKDIR /data
EXPOSE 7860

# Copy the rest of the files
COPY . .

CMD ["streamlit", "run", "app.py", "--server.port", "7860"]