version: "3.9"

services:
  human-seg:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: human-seg
    # Enable GPU support (requires nvidia-container-toolkit)
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

    # Mount the repo to /app inside the container
    volumes:
      - .:/app

    # Optional: mount datasets or models
      - /home/isam/models:/models
      - /home/isam/datasets:/datasets

    # Environment variables
    environment:
      - PYTHONUNBUFFERED=1
      - PYTHONPATH=/app
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility

    # Default command — can be changed to your app’s entrypoint
    command: bash

    # Map ports if your app exposes a server (e.g. FastAPI/Streamlit)
    # ports:
    #   - "7860:7860"

    # Use host networking for simplicity (optional)
    # network_mode: host

    # Restart policy (useful for production)
    restart: unless-stopped