FROM python:3.10

# Install required packages via apt-get
RUN apt-get update && apt-get install -y \
    git \
    git-lfs \
    ffmpeg \
    libsm6 \
    libxext6 \
    cmake \
    rsync \
    libgl1-mesa-glx \
    && rm -rf /var/lib/apt/lists/* \
    && git lfs install

# Set up a user
RUN apt-get update && apt-get install -y fakeroot && \
    mv /usr/bin/apt-get /usr/bin/.apt-get && \
    echo '#!/usr/bin/env sh\nfakeroot /usr/bin/.apt-get $@' > /usr/bin/apt-get && \
    chmod +x /usr/bin/apt-get && \
    rm -rf /var/lib/apt/lists/* && \
    useradd -m -u 1000 user

# Set working directory
WORKDIR /home/user/app

# Install pip packages
RUN pip install --no-cache-dir pip==22.3.1 && \
    pip install --no-cache-dir \
    datasets \
    huggingface-hub>=0.19 \
    hf-transfer>=0.1.4 \
    protobuf<4 \
    click<8.1 \
    pydantic~=1.0

# Install torch and other Python packages
RUN pip install torch

# Copy requirements.txt and install additional requirements
COPY --chown=1000:1000 requirements.txt /tmp/requirements.txt
RUN pip install --no-cache-dir -r /tmp/requirements.txt

# Set environment variable
ENV OPENAI_API_KEY='/content/openkey.env'