services:
  backend:
    build: ./backend
    restart: unless-stopped
    environment:
      # Prefill the setup wizard — uncomment and set your values:
      NAT20_HF_TOKEN: "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"
      NAT20_WHISPER_MODEL: medium
      NAT20_WHISPER_COMPUTE_TYPE: int8
      NAT20_OLLAMA_HOST: http://192.168.0.16:11434
      NAT20_OLLAMA_MODEL: qwen2.5:7b
      # NAT20_API_BASE_URL: https://api.openai.com/v1
      # NAT20_API_KEY: ""
      # NAT20_API_MODEL: gpt-4o-mini
      # NAT20_CHUNK_WORD_TARGET: "2500"
      # NAT20_WORLD_CONTEXT: "" # Add a string of context OR
      # NAT20_WORLD_CONTEXT_PATH: /data/campaign-context.txt # Link to a document of context
    volumes:
      - /mnt/media/media/dnd-sessions:/data
      - hf_cache:/root/.cache/huggingface
      - torch_cache:/root/.cache/torch

    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    ports:
      - "8000:8000"

  frontend:
    build: ./frontend
    restart: unless-stopped
    depends_on:
      - backend
    ports:
      - "8020:8020"

volumes:
  app_data:
  hf_cache:
  torch_cache: