local-gpu-docker-compose.yml

version: "3.8"
services:
  pdf_metadata_extraction_api:
    container_name: pdf_metadata_extraction_api
    entrypoint: [ "gunicorn", "-k", "uvicorn.workers.UvicornWorker", "--chdir", "./src", "app:app", "--bind", "0.0.0.0:5056", "--timeout", "300"  ]
    init: true
    restart: unless-stopped
    build:
      context: .
      dockerfile: Dockerfile
    volumes:
      - data:/app/models_data
    depends_on:
      - mongo_metadata_extraction
      - pdf_metadata_extraction_worker
    environment:
      - ENVIRONMENT=${ENVIRONMENT:-development}
      - SENTRY_DSN=${SENTRY_DSN:-}
    network_mode: 'host'
    env_file: .env.local
  pdf_metadata_extraction_worker:
    container_name: pdf_metadata_extraction_worker
    restart: unless-stopped
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [ gpu ]
    init: true
    entrypoint: [ "python", "-m", "src.start_queue_processor" ]
    build:
      context: .
      dockerfile: Dockerfile
    volumes:
      - data:/app/models_data
    shm_size: '11gb'
    depends_on:
      - mongo_metadata_extraction
    environment:
      - ENVIRONMENT=${ENVIRONMENT:-development}
      - SENTRY_DSN=${SENTRY_DSN:-}
    network_mode: 'host'
    env_file: .env.local
  mongo_metadata_extraction:
    restart: unless-stopped
    command: mongod --port 29017
    image: "mongo:7.0.12"
    network_mode: 'host'

volumes:
  data: