docker-compose.yml

name: alphaops_ai

services:

  # ─── Bases de données ────────────────────────────────────────────────────────

  postgres_data:
    image: postgres:15
    container_name: postgres_data
    restart: always
    environment:
      - POSTGRES_USER=${PGUSER}
      - POSTGRES_PASSWORD=${PGPASSWORD}
      - POSTGRES_DB=${PGDATABASE}
    ports:
      - "127.0.0.1:5433:5432"
    volumes:
      - pgdata_financial:/var/lib/postgresql/data
      - ./data/schema:/docker-entrypoint-initdb.d:ro
    healthcheck:
      test: ["CMD", "pg_isready", "-U", "${PGUSER}", "-d", "${PGDATABASE}"]
      interval: 5s
      retries: 10

  postgres_airflow:
    image: postgres:15
    container_name: postgres_airflow
    restart: always
    environment:
      - POSTGRES_USER=${AIRFLOW_PGUSER}
      - POSTGRES_PASSWORD=${AIRFLOW_PGPASSWORD}
      - POSTGRES_DB=${AIRFLOW_PGDATABASE}
    ports:
      - "127.0.0.1:5434:5432"
    volumes:
      - pgdata_airflow:/var/lib/postgresql/data
    healthcheck:
      test: ["CMD", "pg_isready", "-U", "${AIRFLOW_PGUSER}", "-d", "${AIRFLOW_PGDATABASE}"]
      interval: 5s
      retries: 10

  # ─── MLflow Tracking Server ───────────────────────────────────────────────────

  mlflow:
    build:
      context: .
      dockerfile: mlflow/Dockerfile
    container_name: mlflow
    restart: unless-stopped
    ports:
      - "127.0.0.1:5000:5000"
    environment:
      - MLFLOW_SERVER_ALLOWED_HOSTS=mlflow.khar.fi,127.0.0.1,localhost
      - MLFLOW_SERVER_CORS_ALLOWED_ORIGINS=https://mlflow.khar.fi
    volumes:
      - ./mlruns:/mlruns

  # ─── Airflow ──────────────────────────────────────────────────────────────────

  airflow_init:
    build:
      context: .
      dockerfile: airflow/Dockerfile
    container_name: airflow_init
    depends_on:
      postgres_airflow:
        condition: service_healthy
      postgres_data:
        condition: service_healthy
    environment:
      AIRFLOW__CORE__EXECUTOR: LocalExecutor
      AIRFLOW__DATABASE__SQL_ALCHEMY_CONN: postgresql+psycopg2://${AIRFLOW_PGUSER}:${AIRFLOW_PGPASSWORD}@postgres_airflow:5432/${AIRFLOW_PGDATABASE}
      AIRFLOW__CORE__LOAD_EXAMPLES: "False"
      PYTHONPATH: /opt/airflow
      PGHOST: ${PGHOST}
      PGPORT: ${PGPORT}
      PGDATABASE: ${PGDATABASE}
      PGUSER: ${PGUSER}
      PGPASSWORD: ${PGPASSWORD}
    volumes:
      - ./airflow/dags:/opt/airflow/dags
      - ./airflow/logs:/opt/airflow/logs
      - ./airflow/plugins:/opt/airflow/plugins
      - ./src:/opt/airflow/src
      - ./data:/opt/airflow/data
      - ./ml:/opt/airflow/ml
      - ./artifacts:/opt/airflow/artifacts
    command: >
      bash -c "
      airflow db init &&
      airflow users create --username admin --password admin --firstname admin --lastname admin --role Admin --email admin@local || true
      "

  airflow_webserver:
    build:
      context: .
      dockerfile: airflow/Dockerfile
    container_name: airflow_webserver
    restart: unless-stopped
    depends_on:
      airflow_init:
        condition: service_completed_successfully
    environment:
      AIRFLOW__CORE__EXECUTOR: LocalExecutor
      AIRFLOW__DATABASE__SQL_ALCHEMY_CONN: postgresql+psycopg2://${AIRFLOW_PGUSER}:${AIRFLOW_PGPASSWORD}@postgres_airflow:5432/${AIRFLOW_PGDATABASE}
      AIRFLOW__CORE__LOAD_EXAMPLES: "False"
      PYTHONPATH: /opt/airflow
      PGHOST: ${PGHOST}
      PGPORT: ${PGPORT}
      PGDATABASE: ${PGDATABASE}
      PGUSER: ${PGUSER}
      PGPASSWORD: ${PGPASSWORD}
    volumes:
      - ./airflow/dags:/opt/airflow/dags
      - ./airflow/logs:/opt/airflow/logs
      - ./airflow/plugins:/opt/airflow/plugins
      - ./src:/opt/airflow/src
      - ./data:/opt/airflow/data
      - ./ml:/opt/airflow/ml
      - ./artifacts:/opt/airflow/artifacts
    ports:
      - "127.0.0.1:8081:8080"
    command: airflow webserver

  airflow_scheduler:
    build:
      context: .
      dockerfile: airflow/Dockerfile
    container_name: airflow_scheduler
    restart: unless-stopped
    depends_on:
      airflow_init:
        condition: service_completed_successfully
      mlflow:
        condition: service_started
    environment:
      AIRFLOW__CORE__EXECUTOR: LocalExecutor
      AIRFLOW__DATABASE__SQL_ALCHEMY_CONN: postgresql+psycopg2://${AIRFLOW_PGUSER}:${AIRFLOW_PGPASSWORD}@postgres_airflow:5432/${AIRFLOW_PGDATABASE}
      AIRFLOW__CORE__LOAD_EXAMPLES: "False"
      PYTHONPATH: /opt/airflow
      PGHOST: ${PGHOST}
      PGPORT: ${PGPORT}
      PGDATABASE: ${PGDATABASE}
      PGUSER: ${PGUSER}
      PGPASSWORD: ${PGPASSWORD}
      MLFLOW_TRACKING_URI: http://mlflow:5000
      GIT_PYTHON_REFRESH: quiet
    volumes:
      - ./airflow/dags:/opt/airflow/dags
      - ./airflow/logs:/opt/airflow/logs
      - ./airflow/plugins:/opt/airflow/plugins
      - ./src:/opt/airflow/src
      - ./data:/opt/airflow/data
      - ./ml:/opt/airflow/ml
      - ./artifacts:/opt/airflow/artifacts
      - ./scripts:/opt/airflow/scripts
      - ./mlruns:/mlruns
    command: airflow scheduler

  # ─── Serving API ──────────────────────────────────────────────────────────────

  serving:
    build:
      context: .
      dockerfile: serving/Dockerfile
    container_name: serving
    restart: unless-stopped
    env_file:
      - .env
    environment:
      - HMAC_SECRET=${HMAC_SECRET}
      - RETRAIN_K=${RETRAIN_K:-20}
    ports:
      - "127.0.0.1:8082:8080"
    volumes:
      - ./data:/data
      - ./artifacts:/artifacts

  # ─── Webapp Streamlit ─────────────────────────────────────────────────────────

  webapp:
    build:
      context: .
      dockerfile: webapp/Dockerfile
    container_name: webapp
    restart: unless-stopped
    depends_on:
      postgres_data:
        condition: service_healthy
    env_file:
      - .env
    environment:
      - PGHOST=postgres_data
      - PGPORT=5432
      - MLFLOW_TRACKING_URI=http://mlflow:5000
    volumes:
      - ./webapp/app.py:/app/app.py
      - ./ml:/app/ml
      - ./artifacts:/artifacts
      - ./artifacts:/app/artifacts
      - ./artifacts/monitoring_reports:/app/artifacts/monitoring_reports
    ports:
      - "127.0.0.1:8501:8501"

  # ─── LangChain Email Agent ───────────────────────────────────────────────────

  agent:
    build:
      context: ./agent
      dockerfile: Dockerfile
    container_name: agent
    restart: unless-stopped
    env_file:
      - .env
    environment:
      - FEEDBACK_BASE_URL=${FEEDBACK_BASE_URL:-http://serving:8080}
      - HMAC_SECRET=${HMAC_SECRET:-alphaops-secret-change-me}
      - SMTP_HOST=${SMTP_HOST:-smtp.gmail.com}
      - SMTP_PORT=${SMTP_PORT:-587}
      - SMTP_USER=${SMTP_USER}
      - SMTP_PASS=${SMTP_PASS}
      - SMTP_FROM=${SMTP_FROM}
    ports:
      - "127.0.0.1:8083:8083"

volumes:
  pgdata_financial:
  pgdata_airflow: