FROM nvidia/cuda:12.3.1-devel-ubuntu22.04

# Install Python and basic utilities
RUN apt-get update && apt-get install -y \
    python3.10 \
    python3-pip \
    git \
    wget \
    && rm -rf /var/lib/apt/lists/*

# Set working directory
WORKDIR /app

# Copy application files
COPY . .

# Install Python dependencies
RUN pip3 install --no-cache-dir -r requirements.txt

# Install build dependencies
RUN pip3 install --no-cache-dir ninja pybind11

# Compile custom components
RUN cd /app/hy3dgen/texgen/custom_rasterizer && python3 setup.py install
RUN cd /app/hy3dgen/texgen/differentiable_renderer && bash compile_mesh_painter.sh

# Set environment variables
ENV NVIDIA_VISIBLE_DEVICES=all
ENV PYTHONPATH=/app

# Run the application
CMD ["python3", "hg_app.py"]