docker-compose.prd.yaml

version: '3.8'

services:
    db:
      image: postgres:latest
      restart: always
      container_name: shipped-brain-db
      env_file:
        - .env
      ports:
        - 5432:5432
      volumes:
        - /var/lib/postgres-data:/var/lib/postgresql/data
    mlflow_server:
      build: ./api/
      restart: always
      container_name: mlflow-server
      depends_on: 
        - db
      env_file:
        - .env
      ports:
        - 5000:5000
      volumes:
        - ${CONDA_ENVS_PATH_VOL}:/opt/conda/envs:rw
        - .env:/app/.env:ro
      command: mlflow server --host 0.0.0.0 --backend-store-uri ${DB_URL} --default-artifact-root s3://${ARTIFACT_DIR_NAME}/ --workers ${MLFLOW_SERVER_WORKERS}
    api_server:
      build: ./api/
      restart: always
      container_name: shipped-brain-server
      depends_on: 
        - db
        - mlflow_server
      env_file:
        - .env
      ports:
        - ${API_SERVER_PORT}:${API_SERVER_PORT}
      volumes:
        - ${CONDA_ENVS_PATH_VOL}:/opt/conda/envs:rw
        - .env:/app/.env:ro
        - ./resources/data:/data:rw
      command: gunicorn -k uvicorn.workers.UvicornWorker -b 0.0.0.0:${API_SERVER_PORT} -w ${API_SERVER_WORKERS} main:app
    prediction_server:
      build: ./api/
      restart: always
      container_name: shipped-brain-prediction-server
      depends_on:
        - db
        - mlflow_server
      env_file:
        - .env
      ports:
        - ${PREDICTION_SERVER_PORT}:${PREDICTION_SERVER_PORT}
      volumes:
        - ${CONDA_ENVS_PATH_VOL}:/opt/conda/envs:rw
        - .env:/app/.env:ro
      command: gunicorn -k uvicorn.workers.UvicornWorker -b 0.0.0.0:${PREDICTION_SERVER_PORT} -w ${PREDICTION_SERVER_WORKERS} prediction_server:app
    upload_server:
      build: ./api/
      restart: always
      container_name: shipped-brain-upload-server
      depends_on:
        - db
        - mlflow_server
      env_file:
        - .env
      ports:
        - ${UPLOAD_SERVER_PORT}:${UPLOAD_SERVER_PORT}
      volumes:
        - ${CONDA_ENVS_PATH_VOL}:/opt/conda/envs:rw
        - .env:/app/.env:ro
      command: gunicorn -k uvicorn.workers.UvicornWorker -b 0.0.0.0:${UPLOAD_SERVER_PORT} -w ${UPLOAD_SERVER_WORKERS} model_upload_server:app
    frontend:
      build: ./app/
      container_name: shipped-brain-ui
      volumes:
        - /var/lib/shipped-brain-ui/dist/shipped-brain/:/usr/share/nginx/html/:rw
        - /var/lib/shipped-brain-ui/nginx_conf/:/etc/nginx/conf.d/:rw
        - /var/lib/shipped-brain-ui/letsencrypt/:/etc/letsencrypt/:rw
      ports:
        - 80:80
        - 443:443
      depends_on:
        - api_server
        - mlflow_server
      command: /bin/bash /start_nginx.prd.sh