version: '3.8'

networks:
  emom_mesh:
    driver: bridge

services:
  emom_ui:
    build: 
      context: .
      dockerfile: docker/Dockerfile.ui
    container_name: emom_web_ui
    restart: unless-stopped
    ports:
      - "8080:8080"
    networks:
      - emom_mesh
    env_file:
      - .env
    volumes:
      - ./src:/app/src
      - ${DATA_DEAM_DIR}:/app/dataset/DEAM:ro
    depends_on:
      - emom_inference

  emom_inference:
    build:
      context: .
      dockerfile: docker/Dockerfile.api
    container_name: emom_pytorch_api
    restart: unless-stopped
    networks:
      - emom_mesh
    env_file:
      - .env
    volumes:
      - ${HOST_ARTIFACTS_DIR}/emoset_resnet50_best.pth:/app/src/emoset_resnet50_best.pth:ro
      - ${HOST_ARTIFACTS_DIR}/music_engine/va_regressor.pkl:/app/src/music_engine/va_regressor.pkl:ro
      - ${DATA_DEAM_DIR}:/app/dataset/DEAM:ro
      - ~/.cache/huggingface:/root/.cache/huggingface
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

  emom_ollama:
    image: ollama/ollama:latest
    container_name: emom_ollama_engine
    restart: unless-stopped
    networks:
      - emom_mesh
    volumes:
      - ~/.ollama:/root/.ollama
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]