version: "3.9"

services:
  image-analysis:
    build:
      context: .
      dockerfile: Dockerfile
    image: wing-image-analysis:latest
    container_name: wing-image-analysis
    ports:
      - "5001:5001"

    volumes:
      # ── mx15hdi (EO 드론 카메라) ────────────────────────────────────────
      # 입력: 업로드된 원본 이미지
      - ./mx15hdi/Metadata/Image/Original_Images:/app/mx15hdi/Metadata/Image/Original_Images
      # 출력: 메타데이터 CSV
      - ./mx15hdi/Metadata/CSV:/app/mx15hdi/Metadata/CSV
      # 출력: 지리참조 GeoTIFF (컬러 / 마스크)
      - ./mx15hdi/Georeference/Tif:/app/mx15hdi/Georeference/Tif
      - ./mx15hdi/Georeference/Mask_Tif:/app/mx15hdi/Georeference/Mask_Tif
      # 출력: 유류 폴리곤 Shapefile
      - ./mx15hdi/Polygon/Shp:/app/mx15hdi/Polygon/Shp
      # 출력: 블렌딩 추론 결과 / 마스크 이미지
      - ./mx15hdi/Detect/result:/app/mx15hdi/Detect/result
      - ./mx15hdi/Detect/Mask_result:/app/mx15hdi/Detect/Mask_result
      # ── starsafire (열화상 카메라) ──────────────────────────────────────
      - ./starsafire/Metadata/Image/Original_Images:/app/starsafire/Metadata/Image/Original_Images
      - ./starsafire/Metadata/CSV:/app/starsafire/Metadata/CSV
      - ./starsafire/Georeference/Tif:/app/starsafire/Georeference/Tif
      - ./starsafire/Georeference/Mask_Tif:/app/starsafire/Georeference/Mask_Tif
      - ./starsafire/Polygon/Shp:/app/starsafire/Polygon/Shp
      - ./starsafire/Detect/result:/app/starsafire/Detect/result
      - ./starsafire/Detect/Mask_result:/app/starsafire/Detect/Mask_result
      # ── 스티칭 결과 ─────────────────────────────────────────────────────
      - ./stitch:/app/stitch

    # NVIDIA GPU 할당 (nvidia-container-toolkit 필수)
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

    restart: unless-stopped