docker-compose.yaml

services:
  backend:
    build:
      context: .
      dockerfile: Dockerfile
      target: production
    image: autovoice-backend:${AUTOVOICE_IMAGE_TAG:-local}
    restart: unless-stopped

    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - LOG_LEVEL=${LOG_LEVEL:-info}
      - SECRET_KEY=${SECRET_KEY:?SECRET_KEY must be set}
      - CUDA_VISIBLE_DEVICES=0
      - MAX_WORKERS=${MAX_WORKERS:-4}
      - DATA_DIR=/app/data
      - CORS_ORIGINS=${CORS_ORIGINS:-http://localhost:3000,http://localhost:10001}
      - AUTOVOICE_PUBLIC_DEPLOYMENT=${AUTOVOICE_PUBLIC_DEPLOYMENT:-false}
      - AUTOVOICE_REQUIRE_API_AUTH=${AUTOVOICE_REQUIRE_API_AUTH:-false}
      - AUTOVOICE_API_TOKEN=${AUTOVOICE_API_TOKEN:-}
      - AUTOVOICE_STRICT_PATH_SANDBOX=${AUTOVOICE_STRICT_PATH_SANDBOX:-true}
      - AUTOVOICE_REQUIRE_MEDIA_CONSENT=${AUTOVOICE_REQUIRE_MEDIA_CONSENT:-false}
      - RATE_LIMIT=${RATE_LIMIT:-60}

    env_file:
      - path: .env
        required: false

    ports:
      - "${HOST_PORT:-10001}:5000"

    volumes:
      - ${AUTOVOICE_PRETRAINED_DIR:-./models/pretrained}:/app/models/pretrained:ro
      - autovoice-app-state:/app/data/app_state
      - autovoice-profiles:/app/data/voice_profiles
      - autovoice-samples:/app/data/samples
      - autovoice-trained-models:/app/data/trained_models
      - autovoice-checkpoints:/app/data/checkpoints
      - autovoice-training-vocals:/app/data/training_vocals
      - autovoice-youtube-audio:/app/data/youtube_audio
      - autovoice-separated-youtube:/app/data/separated_youtube
      - autovoice-diarized-youtube:/app/data/diarized_youtube
      - autovoice-uploads:/app/data/uploads
      - autovoice-outputs:/app/data/outputs
      - autovoice-swarm-runs:/app/data/swarm_runs
      - autovoice-swarm-memory:/app/data/swarm_memory
      - autovoice-logs:/app/logs
      - ./config:/app/config:ro

    deploy:
      resources:
        limits:
          memory: 32G
        reservations:
          memory: 8G
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:5000/ready"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 60s

    networks:
      - autovoice-net

    logging:
      driver: json-file
      options:
        max-size: "10m"
        max-file: "3"

  frontend:
    build:
      context: ./frontend
      dockerfile: Dockerfile.frontend
    image: autovoice-frontend:${AUTOVOICE_IMAGE_TAG:-local}
    restart: unless-stopped
    ports:
      - "${FRONTEND_PORT:-3000}:80"
    depends_on:
      backend:
        condition: service_healthy
    networks:
      - autovoice-net
    logging:
      driver: json-file
      options:
        max-size: "10m"
        max-file: "3"

  prometheus:
    image: prom/prometheus:v2.54.1@sha256:69961df6ffa67598048a31aa2822d61f3c93b91d7db24e44d9bb03f99d520da9
    restart: unless-stopped
    command:
      - '--config.file=/etc/prometheus/prometheus.yml'
      - '--storage.tsdb.path=/prometheus'
      - '--web.console.libraries=/usr/share/prometheus/console_libraries'
      - '--web.console.templates=/usr/share/prometheus/consoles'
    ports:
      - "${PROMETHEUS_PORT:-10002}:9090"
    volumes:
      - ./config/prometheus.yml:/etc/prometheus/prometheus.yml:ro
      - ./config/prometheus/alert_rules.yml:/etc/prometheus/alert_rules.yml:ro
      - prometheus-data:/prometheus
    networks:
      - autovoice-net
    profiles:
      - monitoring

  grafana:
    image: grafana/grafana:11.2.0@sha256:37a5d8860aef847dfa09f5f8947f010f6479f98cf7820b5186f9c6314b44be60
    restart: unless-stopped
    ports:
      - "${GRAFANA_PORT:-10003}:3000"
    environment:
      - GF_SECURITY_ADMIN_USER=${GRAFANA_USER:-admin}
      - GF_SECURITY_ADMIN_PASSWORD=${GRAFANA_PASSWORD:?GRAFANA_PASSWORD must be set when monitoring profile is enabled}
      - GF_INSTALL_PLUGINS=grafana-piechart-panel
    volumes:
      - grafana-data:/var/lib/grafana
      - ./config/grafana/dashboards:/etc/grafana/provisioning/dashboards:ro
      - ./config/grafana/datasources:/etc/grafana/provisioning/datasources:ro
    networks:
      - autovoice-net
    profiles:
      - monitoring
    depends_on:
      - prometheus

volumes:
  autovoice-app-state:
    name: autovoice-app-state
  autovoice-profiles:
    name: autovoice-profiles
  autovoice-samples:
    name: autovoice-samples
  autovoice-trained-models:
    name: autovoice-trained-models
  autovoice-checkpoints:
    name: autovoice-checkpoints
  autovoice-training-vocals:
    name: autovoice-training-vocals
  autovoice-youtube-audio:
    name: autovoice-youtube-audio
  autovoice-separated-youtube:
    name: autovoice-separated-youtube
  autovoice-diarized-youtube:
    name: autovoice-diarized-youtube
  autovoice-uploads:
    name: autovoice-uploads
  autovoice-outputs:
    name: autovoice-outputs
  autovoice-swarm-runs:
    name: autovoice-swarm-runs
  autovoice-swarm-memory:
    name: autovoice-swarm-memory
  autovoice-logs:
    name: autovoice-logs
  prometheus-data:
    name: autovoice-prometheus-data
  grafana-data:
    name: autovoice-grafana-data

networks:
  autovoice-net:
    name: autovoice-net
    driver: bridge