services:
  engine:
    build:
      context: ./ACE-Step-1.5
      dockerfile: Dockerfile.cpu
    container_name: resonance-engine
    environment:
      ACESTEP_MODE: api
      ACESTEP_API_HOST: 0.0.0.0
      ACESTEP_API_PORT: 8001
      ACESTEP_API_KEY: ${ACESTEP_API_KEY:-}
      ACESTEP_DEVICE: ${ACESTEP_DEVICE:-cpu}
      ACESTEP_LM_DEVICE: ${ACESTEP_LM_DEVICE:-cpu}
      ACESTEP_LM_BACKEND: ${ACESTEP_LM_BACKEND:-pt}
      ACESTEP_CONFIG_PATH: ${ACESTEP_CONFIG_PATH:-acestep-v15-turbo}
      ACESTEP_LM_MODEL_PATH: ${ACESTEP_LM_MODEL_PATH:-acestep-5Hz-lm-0.6B}
      ACESTEP_NO_INIT: ${ACESTEP_NO_INIT:-true}
      ACESTEP_USE_FLASH_ATTENTION: ${ACESTEP_USE_FLASH_ATTENTION:-false}
      ACESTEP_QUEUE_WORKERS: ${ACESTEP_QUEUE_WORKERS:-1}
      TOKENIZERS_PARALLELISM: "false"
    ports:
      - "127.0.0.1:${ACESTEP_API_PORT:-8001}:8001"
    volumes:
      - ./ACE-Step-1.5/checkpoints:/app/checkpoints
      - ./ACE-Step-1.5/output:/app/output
      - ./ACE-Step-1.5/gradio_outputs:/app/gradio_outputs
      - hf_cache:/root/.cache/huggingface
    shm_size: "2gb"
    restart: unless-stopped

  web:
    build:
      context: ./web
      dockerfile: Dockerfile
      args:
        ACE_API_URL: http://engine:8001
    container_name: resonance-web
    environment:
      ACE_API_URL: http://engine:8001
      NODE_ENV: production
      ENABLE_VISUALIZER: ${ENABLE_VISUALIZER:-false}
      VISUALIZER_PYTHON: python3
      VISUALIZER_SCRIPT_PATH: /app/tools/visualizer/render_visualizer.py
      VISUALIZER_DEVICE: ${VISUALIZER_DEVICE:-auto}
      VISUALIZER_WORKERS: ${VISUALIZER_WORKERS:-}
      VISUALIZER_UPSCALE: ${VISUALIZER_UPSCALE:-}
    ports:
      - "${WEB_PORT:-3000}:3000"
    volumes:
      - ./tools:/app/tools:ro
    depends_on:
      - engine
    restart: unless-stopped

volumes:
  hf_cache: