docker-compose.yaml

version: '3'

volumes:
  pb_volume:

services:
  pocketbase:
    build:
      context: ./pocketbase

    ports:
      - 8090:8090

    command: ["/pocketbase/pocketbase", "serve", "--http", "0.0.0.0:8090"]

    volumes:
      - pb_volume:/pocketbase

  frontend-dev:
    build:
      context: ./frontend/MASSter-frontend

    ports:
      - 8080:8080

    command: ["npm", "run", "preview", "--", "--host", "0.0.0.0", "--port", "8080"]

  preprocessor:
    build:
      context: model/generator/
      dockerfile: docker/Dockerfile

    environment:
      - PB_LINK=http://pocketbase:8090/
      - PB_LOGIN=dev@email.local
      - PB_PWD=1234567890
      - COL_NAME=text_generation_mvp

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

  generator:
    build:
      context: model/generator/
      dockerfile: docker/Dockerfile

    environment:
      - PB_LINK=http://pocketbase:8090/
      - PB_LOGIN=dev@email.local
      - PB_PWD=1234567890
      - COL_NAME=text_generation_mvp

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