version: '3.8'

services:
  mongodb:
    image: mongo:7.0
    container_name: face_recognition_mongo
    restart: always
    ports:
      - "27017:27017"
    volumes:
      - mongo_data:/data/db
    environment:
      MONGO_INITDB_ROOT_USERNAME: admin
      MONGO_INITDB_ROOT_PASSWORD: admin123
      MONGO_INITDB_DATABASE: face_recognition
    networks:
      - face_recognition_net

  chromadb:
    image: chromadb/chroma:0.4.15
    container_name: face_recognition_chroma
    restart: always
    ports:
      - "8000:8000"
    volumes:
      - chroma_data:/chroma/chroma
    environment:
      - IS_PERSISTENT=TRUE
      - PERSIST_DIRECTORY=/chroma/chroma
      - ANONYMIZED_TELEMETRY=FALSE
    networks:
      - face_recognition_net

  face_recognition:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: face_recognition_app
    restart: always
    ports:
      - "8080:8080"
    volumes:
      - ./data:/app/data
      - ./models:/app/models
    environment:
      - MONGODB_URL=mongodb://admin:admin123@mongodb:27017/face_recognition?authSource=admin
      - MONGODB_DB=face_recognition
      - CHROMA_HOST=chromadb
      - CHROMA_PORT=8000
      - UPLOAD_DIR=/app/data/uploads
      - NVIDIA_VISIBLE_DEVICES=all
    depends_on:
      - mongodb
      - chromadb
    networks:
      - face_recognition_net
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

  postgres:
    image: postgres:15
    container_name: vectoradmin_postgres
    restart: unless-stopped
    environment:
      POSTGRES_USER: vectoradmin
      POSTGRES_PASSWORD: "your-db-password"
      POSTGRES_DB: vdbms
    volumes:
      - pgdata:/var/lib/postgresql/data
    networks:
      - face_recognition_net

  vector-admin:
    image: mintplexlabs/vectoradmin:latest
    container_name: vectoradmin_ui
    restart: unless-stopped
    ports:
      - "3001:3001"
    depends_on:
      - postgres
      - chromadb
    environment:
      # Порт, на котором слушает UI
      SERVER_PORT: 3001
      # Секреты для JWT и Inngest
      JWT_SECRET: "your-jwt-secret"
      INNGEST_EVENT_KEY: "background_workers"
      INNGEST_SIGNING_KEY: "your-inngest-signing-key"
      INNGEST_LANDING_PAGE: "true"
      # База данных для метаданных VectorAdmin
      DATABASE_CONNECTION_STRING: "postgresql://vectoradmin:your-db-password@postgres:5432/vdbms"
      # Подключение к ChromaDB (основное хранилище векторов)
      VECTORS_PROVIDER_URL: "http://chromadb:8000"
      VECTORS_PROVIDER_TYPE: "chroma"
      # Автоматическая настройка администратора
      SYS_EMAIL: "admin@vectoradmin.com"
      SYS_PASSWORD: "admin123"
    networks:
      - face_recognition_net

volumes:
  mongo_data:
  chroma_data:
  pgdata:

networks:
  face_recognition_net:
    driver: bridge