FROM python:3.8

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

RUN 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 \
    && useradd -m -u 1000 user

WORKDIR /home/user/app

RUN pip install --no-cache-dir -U pip

# requirements.txt をコンテナにコピー（存在する場合）
COPY requirements.txt /tmp/requirements.txt

# packages.txt をコンテナにコピー（存在する場合）
COPY packages.txt /tmp/packages.txt

# packages.txtに記載のaptパッケージをインストール（存在する場合）
RUN if [ -f /tmp/packages.txt ]; then \
        apt-get update && xargs -r -a /tmp/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*; \
    fi

# requirements.txtに記載のpipパッケージをインストール（存在する場合）
RUN if [ -f /tmp/requirements.txt ]; then \
        pip install --no-cache-dir -r /tmp/requirements.txt; \
    fi

# 直接インストールしたいパッケージ（例）
RUN pip install --no-cache-dir \
        datasets \
        "huggingface-hub>=0.19" \
        "hf_xet>=1.0.0,<2.0.0" \
        "hf-transfer>=0.1.4" \
        "protobuf<4" \
        "click<8.1" \
        "pydantic~=1.0" \
        "gradio==5.31.0"


#packages.txt からパッケージインストール（オプション）
RUN --mount=target=/tmp/packages.txt,source=packages.txt \
    apt-get update && xargs -r -a /tmp/packages.txt apt-get install -y

# requirements.txt からpipパッケージをインストール（オプション）
RUN --mount=target=/tmp/requirements.txt,source=requirements.txt \
    pip install --no-cache-dir -r /tmp/requirements.txt

CMD ["python", "app.py"]