docker-compose.yml

version: '3.8'

services:
  ml-api:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: ml-model-api
    ports:
      - "8000:8000"
    environment:
      - APP_NAME=ML Model API
      - APP_VERSION=1.0.0
      - DEBUG=false
      - LOG_LEVEL=INFO
      - HOST=0.0.0.0
      - PORT=8000
      - WORKERS=1
    volumes:
      # Mount model storage for persistent models
      - ./model_storage:/app/model_storage
      # Mount logs for persistent logging
      - ./logs:/app/logs
      # Optional: Mount app code for development (comment out for production)
      # - ./app:/app/app
    env_file:
      - .env
    restart: unless-stopped
    networks:
      - ml-network
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/api/v1/health"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 10s

  # Optional: Add monitoring services
  # prometheus:
  #   image: prom/prometheus:latest
  #   container_name: prometheus
  #   ports:
  #     - "9090:9090"
  #   volumes:
  #     - ./monitoring/prometheus.yml:/etc/prometheus/prometheus.yml
  #   networks:
  #     - ml-network

  # Optional: Add logging aggregation
  # elasticsearch:
  #   image: docker.elastic.co/elasticsearch/elasticsearch:8.11.0
  #   container_name: elasticsearch
  #   environment:
  #     - discovery.type=single-node
  #   ports:
  #     - "9200:9200"
  #   networks:
  #     - ml-network

networks:
  ml-network:
    driver: bridge

volumes:
  model_storage:
  logs: