FROM docker.io/library/python:3.11
# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    git-lfs \
    ffmpeg \
    libsm6 \
    libxext6 \
    cmake \
    rsync \
    libgl1 \
    && rm -rf /var/lib/apt/lists/* \
    && git lfs install

# Upgrade pip and install base packages
RUN pip install --no-cache-dir pip -U && \
    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==2.10.6"

# Set working directory
WORKDIR /home/user/app

# Install fakeroot and setup 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

# Install Node.js
RUN apt-get update && \
    apt-get install -y curl && \
    curl -fsSL https://deb.nodesource.com/setup_20.x | bash - && \
    apt-get install -y nodejs && \
    rm -rf /var/lib/apt/lists/* && apt-get clean

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

# Copy application files
COPY --chown=1000:1000 . .

# Switch to user
USER user

# Expose port (adjust as needed)
EXPOSE 7860

# Set environment variables
ENV GRADIO_SERVER_NAME="0.0.0.0"

# Run the application (adjust command as needed)
CMD ["python", "app.py"]