Update Dockerfile

Dockerfile

@@ -1,29 +1,48 @@
-# Use an official PyTorch runtime as a parent image with CUDA support
-FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime
+# Use an official PyTorch image WITH CUDA development headers required for compiling
+FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-devel
 
-# Set the working directory in the container
+# Set the working directory
 WORKDIR /app
 
-# Install system dependencies necessary for 3D and image processing
+# Install system dependencies for C++/CUDA extensions, 3D processing, and SSH access
 RUN apt-get update && apt-get install -y \
     git \
+    build-essential \
+    ninja-build \
     libgl1 \
     libglib2.0-0 \
+    tmate \
+    openssh-client \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements file first to leverage Docker layer caching
-COPY requirements.txt .
+# Clone the official Hunyuan3D-2 repository
+RUN git clone https://github.com/Tencent/Hunyuan3D-2.git /app/Hunyuan3D-2
 
-# Install Python dependencies
+# Copy your local requirements first
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Install the base Python requirements from the Tencent repository
+RUN pip install --no-cache-dir -r /app/Hunyuan3D-2/requirements.txt
+
+# Compile the Custom Rasterizer (Required for Textures)
+WORKDIR /app/Hunyuan3D-2/hy3dgen/texgen/custom_rasterizer
+RUN python setup.py install
+
+# Compile the Differentiable Renderer (Required for Textures)
+WORKDIR /app/Hunyuan3D-2/hy3dgen/texgen/differentiable_renderer
+RUN python setup.py install
+
+# Move back to the root app directory
+WORKDIR /app
+
+# Copy your API script into the container
+COPY app.py .
+
 # Create a directory to store generated outputs temporarily
 RUN mkdir -p /app/outputs
 
-# Copy the rest of the application code into the container
-COPY . .
-
-# Hugging Face Spaces requires services to run on port 7860
+# Expose the default Hugging Face Spaces port
 EXPOSE 7860
 
 # Command to start the FastAPI application