services:

  comfyui:
    build:
      dockerfile: Dockerfile.comfyui
    image: comfyui:0.43.0
    pull_policy: never
    container_name: comfyui
    restart: always
    command: >
      --listen 0.0.0.0
      --enable-manager
      --enable-dynamic-vram
      --use-split-cross-attention
      --multi-user
    environment:
      ROCR_VISIBLE_DEVICES: 1
      TZ: Europe/Zurich
      PYOPENGL_PLATFORM: egl
      PYRENDER_PLATFORM: egl
      EGL_PLATFORM: surfaceless
    volumes:
      - type: bind
        source: ./comfyui/custom_nodes
        target: /app/ComfyUI/custom_nodes
        read_only: false
      - type: bind
        source: ./comfyui/input
        target: /app/ComfyUI/input
        read_only: false
      - type: bind
        source: ./comfyui/models
        target: /app/ComfyUI/models
        read_only: false
      - type: bind
        source: ./comfyui/output/
        target: /app/ComfyUI/output/
        read_only: false
      - type: bind
        source: ./comfyui/user
        target: /app/ComfyUI/user
        read_only: false
    ports:
      - 8188:8188
    group_add:
      - video
      - 985
      - 989
    ipc: host
    cap_add:
      - SYS_PTRACE
    security_opt:
      - seccomp=unconfined
    devices:
      - /dev/kfd:/dev/kfd
      - /dev/dri:/dev/dri

  docling:
    image: quay.io/docling-project/docling-serve:latest
    container_name: docling
    pull_policy: always
    restart: always
    mem_limit: 8g
    environment:
      TZ: Europe/Zurich
      DOCLING_SERVE_API_KEY: ${DOCLING_API_KEY}
      DOCLING_SERVE_ENABLE_UI: "true"
      DOCLING_SERVE_MAX_SYNC_WAIT: 600
      DOCLING_SERVE_ENG_LOC_NUM_WORKERS: 2
      UVICORN_WORKERS: 1
      OMP_NUM_THREADS: 8
      MKL_NUM_THREADS: 8
    ports:
      - 5001:5001

  ollama:
    image: ollama/ollama:rocm
    container_name: ollama
    pull_policy: always
    restart: always
    working_dir: /root
    environment:
      OLLAMA_FLASH_ATTENTION: 1
      OLLAMA_KV_CACHE_TYPE: q8_0
      OLLAMA_KEEP_ALIVE: 15m
      OLLAMA_MAX_LOADED_MODELS: 1
      OLLAMA_MAX_QUEUE: 512
      OLLAMA_NUM_PARALLEL: 2
      OLLAMA_NO_CLOUD: 1
      ROCR_VISIBLE_DEVICES: 0
      TZ: Europe/Zurich
    volumes:
      - type: bind
        source: ./ollama/data
        target: /root/.ollama
        read_only: false
      - type: bind
        source: ./ollama/modelfiles
        target: /root/modelfiles
        read_only: false
    ports:
      - 11434:11434
    devices:
      - /dev/kfd:/dev/kfd
      - /dev/dri:/dev/dri

  open-terminal-teachos:
    image: open-terminal/teachos:16.2.0
    build:
      dockerfile: Dockerfile.terminal-teachos
    pull_policy: never
    command: >
      open-terminal
      run
      --host 0.0.0.0
      --port 8000
      --cors-allowed-origins https://ai.arknet.ch
      --cwd /home/ubuntu/workspaces
    restart: always
    container_name: open-terminal-teachos
    environment:
      OPEN_TERMINAL_API_KEY: ${OPEN_TERMINAL_API_KEY}
      TZ: Europe/Zurich
    volumes:
      - type: bind
        source: ./open-terminal/teachos
        target: /home/ubuntu/workspaces
    ports:
      - 8000:8000