feat: Add full Infinigen Docker setup with HuggingFace Inference API

- Dockerfile with Infinigen + Blender (cloned from GitHub)
- Gradio app with Scene Composer, Terrain Engineer, AI Recommendations
- HuggingFace Inference API support via pydantic-ai
- MCP Server enabled

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Dockerfile

@@ -0,0 +1,180 @@
+# Multi-stage build: Builder stage for compilation + Runtime stage for execution
+FROM python:3.11-slim AS builder
+
+# Python environment setup for UV
+ENV UV_COMPILE_BYTECODE=1
+ENV UV_LINK_MODE=copy
+ENV UV_CACHE_DIR=/tmp/.uv-cache
+
+# System dependencies installation for build stage
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        curl \
+        ca-certificates \
+        tzdata \
+        build-essential \
+        git \
+        cmake \
+        ninja-build \
+        libgtk-3-dev \
+        libpng-dev \
+        libjpeg-dev \
+        libwebp-dev \
+        libtiff5-dev \
+        libopenexr-dev \
+        libopenblas-dev \
+        libx11-dev \
+        libavutil-dev \
+        libavcodec-dev \
+        libavformat-dev \
+        libswscale-dev \
+        libswresample-dev \
+        libssl-dev \
+        libva-dev \
+        libmfx-dev \
+        libgstreamer1.0-dev \
+        libgstreamer-plugins-base1.0-dev \
+        opencl-headers \
+        ocl-icd-opencl-dev \
+        xvfb \
+        xauth \
+        g++ \
+        gcc \
+        libomp-dev \
+        libgomp1 \
+        coreutils \
+        findutils \
+        bash \
+        procps \
+        # Additional dependencies for terrain compilation
+        libc6-dev \
+        pkg-config \
+        make \
+        dos2unix \
+        # CUDA dependencies - install NVIDIA CUDA toolkit
+        wget \
+        software-properties-common \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install CUDA toolkit for terrain compilation
+RUN wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run -O cuda_installer.run && \
+    chmod +x cuda_installer.run && \
+    ./cuda_installer.run --no-opengl-libs --no-man-page --no-opengl-libs --override --silent --toolkit && \
+    rm cuda_installer.run && \
+    echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> /etc/profile && \
+    echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> /etc/profile
+
+# Create user for build process
+RUN useradd -ms /bin/bash user
+
+# Install uv
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh
+ENV PATH="/root/.local/bin:/usr/local/bin:/usr/bin:/bin:$PATH"
+
+# Create virtual environment
+RUN uv venv /app/.venv
+ENV VIRTUAL_ENV=/app/.venv
+ENV PATH="/app/.venv/bin:$PATH"
+
+# Set up working directory
+WORKDIR /app
+
+# Clone infinigen from GitHub
+RUN git clone https://github.com/bjoernbethge/infinigen.git /app
+
+# Install Python dependencies
+RUN sh -c "ulimit -n 4096 && uv sync --frozen --extra terrain --extra vis"
+
+# Compile terrain libraries with CUDA support
+RUN chmod +x scripts/install/compile_terrain.sh && \
+    dos2unix scripts/install/compile_terrain.sh && \
+    dos2unix infinigen/OcMesher/install.sh && \
+    bash -c "set -e; cd /app && bash scripts/install/compile_terrain.sh" && \
+    echo "Terrain compilation completed successfully with CUDA support"
+
+# Compile Cython terrain libraries
+RUN bash -c "source .venv/bin/activate && python setup.py build_ext --inplace" && \
+    echo "Cython terrain compilation completed successfully"
+
+# Install additional runtime dependencies
+RUN bash -c "source .venv/bin/activate && pip install PyOpenGL-accelerate"
+
+# Install Gradio and HuggingFace dependencies
+RUN sh -c "ulimit -n 4096 && uv add gradio>=5.0.0 pydantic-ai[huggingface]>=1.0.8"
+
+# Set ownership
+RUN chown -R user:user /app/.venv && \
+    chown -R user:user /root/.local
+
+# ============================================
+# Runtime stage - minimal image for execution
+# ============================================
+FROM python:3.11-slim AS runtime
+
+# Runtime environment setup
+ENV UV_COMPILE_BYTECODE=1
+ENV UV_CACHE_DIR=/tmp/.uv-cache
+ENV DISPLAY=:99
+ENV BLENDER_HEADLESS=1
+
+# Minimal runtime dependencies with NVIDIA GPU support
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        curl \
+        ca-certificates \
+        tzdata \
+        libx11-6 \
+        libglib2.0-0 \
+        libgtk-3-0 \
+        libgomp1 \
+        libomp5 \
+        libopenblas0 \
+        libpng16-16 \
+        libjpeg62-turbo \
+        libtiff6 \
+        libwebp7 \
+        libopenexr-3-1-30 \
+        libssl3 \
+        libgstreamer1.0-0 \
+        libgstreamer-plugins-base1.0-0 \
+        opencl-headers \
+        ocl-icd-opencl-dev \
+        xvfb \
+        xauth \
+        bash \
+        procps \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create user (HuggingFace Spaces standard)
+RUN useradd -ms /bin/bash user
+
+# Copy built artifacts from builder stage
+COPY --from=builder /app /app
+COPY --from=builder /root/.local /root/.local
+
+# Set up environment
+ENV PATH="/app/.venv/bin:/root/.local/bin:/usr/local/bin:/usr/bin:/bin:$PATH"
+ENV VIRTUAL_ENV=/app/.venv
+ENV HOME=/home/user
+
+# Set up working directory
+WORKDIR /app
+
+# Create startup script for Gradio
+RUN echo '#!/bin/bash\nXvfb :99 -screen 0 1024x768x24 > /dev/null 2>&1 &\nexport DISPLAY=:99\ncd /app\nexec python app.py' > /usr/local/bin/start-gradio.sh && \
+    chmod +x /usr/local/bin/start-gradio.sh
+
+# Copy Gradio app
+COPY app.py /app/app.py
+
+# Create logs directory
+RUN mkdir -p /app/logs && chown -R user:user /app
+
+# Switch to user
+USER user
+
+# Expose Gradio port
+EXPOSE 7860
+
+# Default command
+CMD ["/usr/local/bin/start-gradio.sh"]

README.md

@@ -3,10 +3,8 @@ title: Infinigen Agents
 emoji: 🌍
 colorFrom: green
 colorTo: blue
-sdk: gradio
-sdk_version: 6.0.2
-python_version: "3.10"
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: false
 license: mit
 tags:
@@ -15,12 +13,13 @@ tags:
   - 3d
   - procedural-generation
   - infinigen
-short_description: "AI Agents for Procedural 3D World Generation"
+  - blender
+short_description: "AI Agents for Procedural 3D World Generation with Blender"
 ---
 
 # 🌍 Infinigen Agents
 
-AI-powered agents for procedural 3D world generation using [Infinigen](https://github.com/princeton-vl/infinigen).
+AI-powered agents for procedural 3D world generation using [Infinigen](https://github.com/princeton-vl/infinigen) + Blender.
 
 ## Features
 
@@ -32,7 +31,7 @@ AI-powered agents for procedural 3D world generation using [Infinigen](https://g
 
 ## AI Backend
 
-Uses HuggingFace Inference API by default:
+Uses HuggingFace Inference API:
 - Model: `openai/gpt-oss-20b`
 - Configurable providers: Cerebras, Together, Nebius, Groq
 

app.py

@@ -1,9 +1,16 @@
 """
 Infinigen Agents - AI-powered procedural 3D generation
+Full version with Infinigen + Blender in Docker container
 """
 import os
+import sys
 import gradio as gr
 from typing import Dict, Any
+from pathlib import Path
+
+# Add infinigen to path (in Docker container)
+sys.path.insert(0, "/app")
+sys.path.insert(0, "/app/infinigen")
 
 # HuggingFace token from Space secrets
 HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -19,11 +26,11 @@ def get_model():
     if AI_MODEL == "huggingface":
         from pydantic_ai.models.huggingface import HuggingFaceModel
         from pydantic_ai.providers.huggingface import HuggingFaceProvider
-        
+
         provider_kwargs = {"api_key": HF_TOKEN}
         if HF_PROVIDER:
             provider_kwargs["provider_name"] = HF_PROVIDER
-            
+
         return HuggingFaceModel(HF_MODEL_ID, provider=HuggingFaceProvider(**provider_kwargs))
     else:
         return f"openai:gpt-4o-mini"
@@ -33,14 +40,14 @@ def compose_scene(scene_type: str, seed: int, complexity: str) -> Dict[str, Any]
     """Compose a scene using AI agent."""
     try:
         from pydantic_ai import Agent
-        
+
         agent = Agent(
             get_model(),
             system_prompt=f"""You are a scene composer for Infinigen.
             Create a {complexity} complexity {scene_type} scene with seed {seed}.
             Respond with JSON containing: scene_type, seed, assets, lighting, camera."""
         )
-        
+
         result = agent.run_sync(f"Create a {scene_type} scene")
         return {
             "success": True,
@@ -57,14 +64,14 @@ def generate_terrain(terrain_type: str, seed: int, resolution: int) -> Dict[str,
     """Generate terrain using AI agent."""
     try:
         from pydantic_ai import Agent
-        
+
         agent = Agent(
             get_model(),
             system_prompt=f"""You are a terrain engineer for Infinigen.
             Generate {terrain_type} terrain with resolution {resolution}.
             Respond with terrain parameters: heightmap settings, erosion, materials."""
         )
-        
+
         result = agent.run_sync(f"Generate {terrain_type} terrain")
         return {
             "success": True,
@@ -81,13 +88,13 @@ def get_recommendations(scene_type: str) -> str:
     """Get AI recommendations for scene generation."""
     try:
         from pydantic_ai import Agent
-        
+
         agent = Agent(
             get_model(),
             system_prompt="""You are an expert on Infinigen procedural generation.
             Provide recommendations for assets, terrain, lighting, and camera setup."""
         )
-        
+
         result = agent.run_sync(f"Recommend settings for a {scene_type} scene in Infinigen")
         return str(result.data)
     except Exception as e:
@@ -99,10 +106,10 @@ with gr.Blocks(title="Infinigen Agents") as demo:
     gr.Markdown("""
     # 🌍 Infinigen Agents
     **AI-powered procedural 3D world generation**
-    
-    Using HuggingFace Inference API with pydantic-ai
+
+    Full version with Infinigen + Blender - Using HuggingFace Inference API
     """)
-    
+
     with gr.Tab("Scene Composer"):
         with gr.Row():
             scene_type = gr.Dropdown(
@@ -112,12 +119,12 @@ with gr.Blocks(title="Infinigen Agents") as demo:
             )
             scene_seed = gr.Number(label="Seed", value=42)
             complexity = gr.Dropdown(["low", "medium", "high"], label="Complexity", value="medium")
-        
+
         compose_btn = gr.Button("🎬 Compose Scene", variant="primary")
         scene_output = gr.JSON(label="Scene Result")
-        
+
         compose_btn.click(compose_scene, [scene_type, scene_seed, complexity], scene_output)
-    
+
     with gr.Tab("Terrain Engineer"):
         with gr.Row():
             terrain_type = gr.Dropdown(
@@ -127,12 +134,12 @@ with gr.Blocks(title="Infinigen Agents") as demo:
             )
             terrain_seed = gr.Number(label="Seed", value=42)
             resolution = gr.Slider(128, 2048, value=512, step=128, label="Resolution")
-        
+
         terrain_btn = gr.Button("🏔️ Generate Terrain", variant="primary")
         terrain_output = gr.JSON(label="Terrain Result")
-        
+
         terrain_btn.click(generate_terrain, [terrain_type, terrain_seed, resolution], terrain_output)
-    
+
     with gr.Tab("AI Recommendations"):
         rec_scene_type = gr.Dropdown(
             ["forest", "desert", "mountain", "canyon", "coast"],
@@ -141,16 +148,16 @@ with gr.Blocks(title="Infinigen Agents") as demo:
         )
         rec_btn = gr.Button("💡 Get Recommendations", variant="primary")
         rec_output = gr.Textbox(label="AI Recommendations", lines=10)
-        
+
         rec_btn.click(get_recommendations, rec_scene_type, rec_output)
-    
+
     gr.Markdown(f"""
     ---
     ### Configuration
     - **AI Model**: {AI_MODEL}
     - **HF Model**: {HF_MODEL_ID}
     - **Provider**: {HF_PROVIDER or 'auto'}
-    
+
     ### MCP Server
     ```json
     {{
@@ -165,4 +172,4 @@ with gr.Blocks(title="Infinigen Agents") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(server_name="0.0.0.0", server_port=7860, mcp_server=True)