docker-compose.yml

services:
  db:
    image: postgres:15-alpine
    container_name: rpg_postgres
    environment:
      POSTGRES_USER: rpg_user
      POSTGRES_PASSWORD: rpg_password
      POSTGRES_DB: rpg_documents
    volumes:
      - postgres_data:/var/lib/postgresql/data
    ports:
      - "5432:5432"
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U rpg_user -d rpg_documents"]
      interval: 10s
      timeout: 5s
      retries: 5
  app:
    build: .
    container_name: rpg_app
    environment:
      - DATABASE_URL=postgresql://rpg_user:rpg_password@db:5432/rpg_documents
      - PYTHONUNBUFFERED=1
      - PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
    volumes:
      - ./bucket:/app/bucket
      - ./sample_docs:/app/sample_docs
      - ./app:/app/app
      - huggingface_cache:/root/.cache/huggingface
    ports:
      - "8000:8000"
    depends_on:
      db:
        condition: service_healthy
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    command: bash -c "source /app/.venv/bin/activate && python3 -c 'import transformers' && uvicorn app.main:app --host 0.0.0.0 --port 8000 --reload"
  pgadmin:
    image: dpage/pgadmin4:latest
    container_name: rpg_pgadmin
    environment:
      PGADMIN_DEFAULT_EMAIL: admin@rpg.com
      PGADMIN_DEFAULT_PASSWORD: admin
      PGADMIN_CONFIG_SERVER_MODE: 'False'
    volumes:
      - pgadmin_data:/var/lib/pgadmin
    ports:
      - "5050:80"
    depends_on:
      - db
    profiles:
      - dev
volumes:
  postgres_data:
  pgadmin_data:
  huggingface_cache: