version: '3.8'

services:
  astral-api:
    build:
      context: .
      dockerfile: Dockerfile
    image: astral-next-api:latest
    container_name: astral-next-api
    ports:
      - "8000:8000"
    volumes:
      - ./satellite_cache:/app/satellite_cache
      - ./models:/app/models
      - ./image_storage:/app/image_storage
      - ./test_flight_data:/app/test_flight_data
      - ./results_cache.db:/app/results_cache.db
      - ./flights.db:/app/flights.db
    environment:
      - USE_MOCK_MODELS=0  # Change to 1 if deploying to a CPU-only environment without a GPU
      - TEST_FLIGHT_DIR=/app/test_flight_data
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    restart: unless-stopped