HartOMat/docker-compose.yml

name: hartomat

services:
  postgres:
    image: postgres:16-alpine
    environment:
      POSTGRES_DB: ${POSTGRES_DB:-hartomat}
      POSTGRES_USER: ${POSTGRES_USER:-hartomat}
      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-hartomat}
    volumes:
      - pgdata:/var/lib/postgresql/data
    ports:
      - "5432:5432"
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U ${POSTGRES_USER:-hartomat}"]
      interval: 5s
      timeout: 5s
      retries: 5

  redis:
    image: redis:7-alpine
    ports:
      - "6379:6379"
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 5s
      timeout: 5s
      retries: 5

  minio:
    image: minio/minio:latest
    command: server /data --console-address ":9001"
    environment:
      MINIO_ROOT_USER: ${MINIO_USER:-minioadmin}
      MINIO_ROOT_PASSWORD: ${MINIO_PASSWORD:-minioadmin}
    ports:
      - "9000:9000"
      - "9001:9001"
    volumes:
      - minio-data:/data
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"]
      interval: 10s
      timeout: 5s
      retries: 5

  backend:
    build:
      context: ./backend
      dockerfile: Dockerfile
    command: /start.sh
    environment:
      - PYTHONDONTWRITEBYTECODE=${PYTHONDONTWRITEBYTECODE:-1}
      - PYTHONPYCACHEPREFIX=${PYTHONPYCACHEPREFIX:-/tmp/pycache}
      - POSTGRES_DB=${POSTGRES_DB:-hartomat}
      - POSTGRES_USER=${POSTGRES_USER:-hartomat}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-hartomat}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=5432
      - REDIS_URL=${REDIS_URL:-redis://redis:6379/0}
      - JWT_SECRET_KEY=${JWT_SECRET_KEY:-changeme-in-production}
      - JWT_ALGORITHM=${JWT_ALGORITHM:-HS256}
      - JWT_ACCESS_TOKEN_EXPIRE_MINUTES=${JWT_ACCESS_TOKEN_EXPIRE_MINUTES:-480}
      - AZURE_OPENAI_API_KEY=${AZURE_OPENAI_API_KEY:-}
      - AZURE_OPENAI_ENDPOINT=${AZURE_OPENAI_ENDPOINT:-}
      - AZURE_OPENAI_DEPLOYMENT=${AZURE_OPENAI_DEPLOYMENT:-gpt-4o}
      - AZURE_OPENAI_API_VERSION=${AZURE_OPENAI_API_VERSION:-2024-02-01}
      - UPLOAD_DIR=/app/uploads
      - MAX_UPLOAD_SIZE_MB=${MAX_UPLOAD_SIZE_MB:-500}
      - MINIO_URL=${MINIO_URL:-http://minio:9000}
      - MINIO_USER=${MINIO_USER:-minioadmin}
      - MINIO_PASSWORD=${MINIO_PASSWORD:-minioadmin}
      - MINIO_BUCKET=${MINIO_BUCKET:-uploads}
      - COMPOSE_PROJECT_DIR=/compose
    volumes:
      - ./backend:/app
      - uploads:/app/uploads
      - /var/run/docker.sock:/var/run/docker.sock
      - .:/compose:ro
    ports:
      - "8888:8888"
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy
      minio:
        condition: service_healthy

  worker:
    build:
      context: ./backend
      dockerfile: Dockerfile
    user: "${APP_UID:-1000}:0"
    command: celery -A app.tasks.celery_app worker --loglevel=info -Q step_processing,ai_validation --autoscale=${MAX_CONCURRENCY:-8},${MIN_CONCURRENCY:-2} --concurrency=${MIN_CONCURRENCY:-2}
    environment:
      - PYTHONDONTWRITEBYTECODE=${PYTHONDONTWRITEBYTECODE:-1}
      - PYTHONPYCACHEPREFIX=${PYTHONPYCACHEPREFIX:-/tmp/pycache}
      - HOME=/tmp
      - POSTGRES_DB=${POSTGRES_DB:-hartomat}
      - POSTGRES_USER=${POSTGRES_USER:-hartomat}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-hartomat}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=5432
      - REDIS_URL=${REDIS_URL:-redis://redis:6379/0}
      - JWT_SECRET_KEY=${JWT_SECRET_KEY:-changeme-in-production}
      - AZURE_OPENAI_API_KEY=${AZURE_OPENAI_API_KEY:-}
      - AZURE_OPENAI_ENDPOINT=${AZURE_OPENAI_ENDPOINT:-}
      - AZURE_OPENAI_DEPLOYMENT=${AZURE_OPENAI_DEPLOYMENT:-gpt-4o}
      - AZURE_OPENAI_API_VERSION=${AZURE_OPENAI_API_VERSION:-2024-02-01}
      - UPLOAD_DIR=/app/uploads
      - MINIO_URL=${MINIO_URL:-http://minio:9000}
      - MINIO_USER=${MINIO_USER:-minioadmin}
      - MINIO_PASSWORD=${MINIO_PASSWORD:-minioadmin}
      - MINIO_BUCKET=${MINIO_BUCKET:-uploads}
      - CELERY_WORKER_CONCURRENCY=${CELERY_WORKER_CONCURRENCY:-8}
    volumes:
      - ./backend:/app
      - uploads:/app/uploads
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy

  render-worker:
    build:
      context: .
      dockerfile: render-worker/Dockerfile
      args:
        - BLENDER_VERSION=${BLENDER_VERSION:-5.0.1}
    user: "${APP_UID:-1000}:0"
    group_add:
      - "44"
      - "110"
    command: bash -c "python3 /check_version.py && celery -A app.tasks.celery_app worker --loglevel=info -Q asset_pipeline --autoscale=1,1 --concurrency=1"
    environment:
      - PYTHONDONTWRITEBYTECODE=${PYTHONDONTWRITEBYTECODE:-1}
      - PYTHONPYCACHEPREFIX=${PYTHONPYCACHEPREFIX:-/tmp/pycache}
      - HOME=/tmp
      - POSTGRES_DB=${POSTGRES_DB:-hartomat}
      - POSTGRES_USER=${POSTGRES_USER:-hartomat}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-hartomat}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=5432
      - REDIS_URL=${REDIS_URL:-redis://redis:6379/0}
      - JWT_SECRET_KEY=${JWT_SECRET_KEY:-changeme-in-production}
      - UPLOAD_DIR=/app/uploads
      - BLENDER_BIN=/opt/blender/blender
      - RENDER_SCRIPTS_DIR=/render-scripts
      - CYCLES_DEVICE=${CYCLES_DEVICE:-gpu}
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics
      - MINIO_URL=${MINIO_URL:-http://minio:9000}
      - MINIO_USER=${MINIO_USER:-minioadmin}
      - MINIO_PASSWORD=${MINIO_PASSWORD:-minioadmin}
      - MINIO_BUCKET=${MINIO_BUCKET:-uploads}
    volumes:
      - ./backend:/app
      - ./render-worker/scripts:/render-scripts
      - uploads:/app/uploads
      - /opt/blender:/opt/blender:ro
      - optix-cache:/var/tmp/OptixCache_root   # persist OptiX kernel cache across container restarts
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu, compute, utility, graphics]

  render-worker-light:
    profiles: ["multi-gpu"]   # Only start with: docker compose --profile multi-gpu up -d
    build:
      context: .
      dockerfile: render-worker/Dockerfile
      args:
        - BLENDER_VERSION=${BLENDER_VERSION:-5.0.1}
    user: "${APP_UID:-1000}:0"
    group_add:
      - "44"
      - "110"
    command: bash -c "python3 /check_version.py && celery -A app.tasks.celery_app worker --loglevel=info -Q asset_pipeline_light --autoscale=2,2 --concurrency=2"
    environment:
      - PYTHONDONTWRITEBYTECODE=${PYTHONDONTWRITEBYTECODE:-1}
      - PYTHONPYCACHEPREFIX=${PYTHONPYCACHEPREFIX:-/tmp/pycache}
      - HOME=/tmp
      - POSTGRES_DB=${POSTGRES_DB:-hartomat}
      - POSTGRES_USER=${POSTGRES_USER:-hartomat}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-hartomat}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=5432
      - REDIS_URL=${REDIS_URL:-redis://redis:6379/0}
      - JWT_SECRET_KEY=${JWT_SECRET_KEY:-changeme-in-production}
      - UPLOAD_DIR=/app/uploads
      - BLENDER_BIN=/opt/blender/blender
      - RENDER_SCRIPTS_DIR=/render-scripts
      - CYCLES_DEVICE=${CYCLES_DEVICE:-gpu}
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility,graphics
      - MINIO_URL=${MINIO_URL:-http://minio:9000}
      - MINIO_USER=${MINIO_USER:-minioadmin}
      - MINIO_PASSWORD=${MINIO_PASSWORD:-minioadmin}
      - MINIO_BUCKET=${MINIO_BUCKET:-uploads}
    volumes:
      - ./backend:/app
      - ./render-worker/scripts:/render-scripts
      - uploads:/app/uploads
      - /opt/blender:/opt/blender:ro
      - optix-cache:/var/tmp/OptixCache_root
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu, compute, utility, graphics]

  beat:
    build:
      context: ./backend
      dockerfile: Dockerfile
    user: "${APP_UID:-1000}:0"
    command: celery -A app.tasks.celery_app beat --loglevel=info --schedule=/tmp/celerybeat-schedule
    environment:
      - PYTHONDONTWRITEBYTECODE=${PYTHONDONTWRITEBYTECODE:-1}
      - PYTHONPYCACHEPREFIX=${PYTHONPYCACHEPREFIX:-/tmp/pycache}
      - HOME=/tmp
      - POSTGRES_DB=${POSTGRES_DB:-hartomat}
      - POSTGRES_USER=${POSTGRES_USER:-hartomat}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-hartomat}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=5432
      - REDIS_URL=${REDIS_URL:-redis://redis:6379/0}
      - JWT_SECRET_KEY=${JWT_SECRET_KEY:-changeme-in-production}
      - AZURE_OPENAI_API_KEY=${AZURE_OPENAI_API_KEY:-}
      - AZURE_OPENAI_ENDPOINT=${AZURE_OPENAI_ENDPOINT:-}
      - AZURE_OPENAI_DEPLOYMENT=${AZURE_OPENAI_DEPLOYMENT:-gpt-4o}
      - AZURE_OPENAI_API_VERSION=${AZURE_OPENAI_API_VERSION:-2024-02-01}
      - UPLOAD_DIR=/app/uploads
    volumes:
      - ./backend:/app
      - uploads:/app/uploads
    depends_on:
      postgres:
        condition: service_healthy
      redis:
        condition: service_healthy

  frontend:
    build:
      context: ./frontend
      dockerfile: Dockerfile
    ports:
      - "5173:5173"
    environment:
      - VITE_API_URL=http://backend:8888
    volumes:
      - ./frontend:/app
      - /app/node_modules
    depends_on:
      - backend

volumes:
  pgdata:
    external: true
    name: ${PGDATA_VOLUME:-hartomat_pgdata}
  uploads:
    external: true
    name: ${UPLOADS_VOLUME:-hartomat_uploads}
  minio-data:
    external: true
    name: ${MINIO_DATA_VOLUME:-hartomat_minio-data}
  optix-cache:     # OptiX BVH/kernel pipeline cache (survives render-worker rebuilds)
    external: true
    name: ${OPTIX_CACHE_VOLUME:-hartomat_optix-cache}