docker-compose.yml

services:
  registry:
    build:
      context: .
      dockerfile: Dockerfile.registry
    container_name: ml-registry
    ports:
      - "8000:8000"
    environment:
      - LOG_LEVEL=INFO
      - APP_ENV=production
      - DATABASE_URL=sqlite:///./data/registry.db
      - MODEL_STORE_PATH=./model_store
      - API_KEY=your-api-key-here
    volumes:
      - registry-data:/app/data
      - model-store:/app/model_store
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"]
      interval: 30s
      timeout: 5s
      retries: 3
      start_period: 10s

  serving:
    build:
      context: .
      dockerfile: Dockerfile.serving
    container_name: ml-serving
    ports:
      - "8001:8001"
    depends_on:
      registry:
        condition: service_healthy

  gateway:
    build:
      context: .
      dockerfile: Dockerfile.serving
    container_name: ml-gateway
    ports:
      - "8002:8002"
    environment:
      - LOG_LEVEL=INFO
    command: ["uvicorn", "serving.gateway_app:app", "--host", "0.0.0.0", "--port", "8002", "--no-access-log"]
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8002/health')"]
      interval: 30s
      timeout: 3s
      start_period: 10s
      retries: 3
    depends_on:
      registry:
        condition: service_healthy
    restart: unless-stopped

volumes:
  registry-data:
    driver: local
  model-store:
    driver: local