docker-compose.yaml

version: '3.3'

services:
  flow:
    build: mlflow
    ports:
      - "${MLFLOW_SERVER_PORT_NOAUTH}:5000"
    restart: always
    volumes:
      - "${MLFLOW_BACKEND_DIR}:/workspace"
    networks:
      - default
    depends_on:
      - sftp
    working_dir: /workspace
    command:
      - sh
      - -c
      - sleep 5
        ;
        ssh-keyscan -H -p ${MLFLOW_ARTIFACT_PORT} ${MLFLOW_ARTIFACT_URI} | tee known_hosts
        ;
        mlflow server 
        --port 5000 
        --host 0.0.0.0
        --backend-store-uri sqlite:///mlruns.db
        --default-artifact-root sftp://${MLFLOW_ARTIFACT_USER}:${MLFLOW_ARTIFACT_PASS}@${MLFLOW_ARTIFACT_URI}:${MLFLOW_ARTIFACT_PORT}/artifacts
    user: "$DOCKER_UID:$DOCKER_GID" 

  nginx:
    image: 'nginx:1.21'
    restart: always
    ports:
      - "${MLFLOW_SERVER_PORT_AUTH}:80"
      - "${MLFLOW_SERVER_PORT_AUTH_SSL}:443"
    networks:
      - default
    depends_on:
      - flow
    volumes:
      - ./nginx/nginx.conf:/etc/nginx/nginx.conf
      - ./nginx/mlflow.conf:/etc/nginx/sites-enabled/mlflow.conf
      - ./nginx/mlflow_ssl.conf:/etc/nginx/sites-enabled/mlflow_ssl.conf
      - ./nginx/gen_htaccess.sh:/docker-entrypoint.d/90_gen_htaccess.sh
      - ./nginx/gen_key.sh:/docker-entrypoint.d/91_gen_key.sh
      - ${MLFLOW_SSL_CERT_DIR}:/cert
    environment:
      - "HTTP_USER=${MLFLOW_SERVER_USER}"
      - "HTTP_PASS=${MLFLOW_SERVER_PASS}"
      - "CUID=${DOCKER_UID}"
      - "CGID=${DOCKER_GID}"
      - "CCN=${MLFLOW_SERVER_SSL_CN}"
      - "CSAN=${MLFLOW_SERVER_SSL_SAN}"
  sftp:
    image: 'atmoz/sftp'
    volumes:
      - ${MLFLOW_ARTIFACT_DIR}:/home/${MLFLOW_ARTIFACT_USER}/artifacts
      - ${MLFLOW_SSL_CERT_DIR}:/cert
      - ./sftp/copy_host_keys.sh:/etc/sftp.d/00_copy_host_keys.sh
    ports:
      - "${MLFLOW_ARTIFACT_PORT}:22"
    environment:
      - "CUID=${DOCKER_UID}"
      - "CGID=${DOCKER_GID}"
    command: ${MLFLOW_ARTIFACT_USER}:${MLFLOW_ARTIFACT_PASS}:${DOCKER_UID}:${DOCKER_GID}:artifacts

networks:
 default: