version: "3.9"

services:
  mongo:
    image: mongo:6
    container_name: storyforge-mongo
    restart: unless-stopped
    ports:
      - "27017:27017"
    volumes:
      - ./data/mongo:/data/db

  vectorDB:
    image: pgvector/pgvector:pg16
    container_name: storyforge-pgvector
    restart: unless-stopped
    environment:
      POSTGRES_DB: ${POSTGRES_DB:-fastgpt}
      POSTGRES_USER: ${POSTGRES_USER:-postgres}
      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-postgres}
    ports:
      - "5432:5432"
    volumes:
      - ./data/pg:/var/lib/postgresql/data

  redis:
    image: redis:7-alpine
    container_name: storyforge-redis
    restart: unless-stopped
    ports:
      - "6379:6379"
    volumes:
      - ./data/redis:/data

  minio:
    image: minio/minio:RELEASE.2025-02-07T23-21-09Z
    container_name: storyforge-minio
    restart: unless-stopped
    command: server /data --console-address ":9001"
    environment:
      MINIO_ROOT_USER: ${MINIO_ROOT_USER:-minioadmin}
      MINIO_ROOT_PASSWORD: ${MINIO_ROOT_PASSWORD:-minioadmin}
    ports:
      - "9000:9000"
      - "9001:9001"
    volumes:
      - ./data/minio:/data

  collector:
    build:
      context: ./collector-service
    container_name: storyforge-collector
    restart: unless-stopped
    environment:
      DATA_DIR: /data/collector
      DATABASE_PATH: /data/collector/storyforge.db
      DEFAULT_EXTERNAL_BASE_URL: ${DEFAULT_EXTERNAL_BASE_URL:-https://test.hyzq.net/storyforge}
      LOCAL_OPENAI_BASE_URL: ${LOCAL_OPENAI_BASE_URL:-http://host.docker.internal:8317/v1}
      LOCAL_OPENAI_MODEL: ${LOCAL_OPENAI_MODEL:-GLM-5}
      LOCAL_OPENAI_API_KEY: ${LOCAL_OPENAI_API_KEY:-}
      FASTGPT_BASE_URL: ${FASTGPT_BASE_URL:-http://host.docker.internal:3000}
      FASTGPT_DATASET_API_KEY: ${FASTGPT_DATASET_API_KEY:-}
      YTDLP_BIN: ${YTDLP_BIN:-yt-dlp}
      FFMPEG_BIN: ${FFMPEG_BIN:-ffmpeg}
      WHISPER_BIN: ${WHISPER_BIN:-}
      WHISPER_MODEL: ${WHISPER_MODEL:-/data/collector/models/ggml-base.en.bin}
    ports:
      - "8081:8081"
    volumes:
      - ./data/collector:/data/collector
    command: uvicorn app.main:app --host 0.0.0.0 --port 8081

  fastgpt:
    image: ghcr.io/labring/fastgpt:latest
    container_name: storyforge-fastgpt
    restart: unless-stopped
    depends_on:
      - mongo
      - vectorDB
      - redis
      - minio
    ports:
      - "3000:3000"

  sandbox:
    image: ghcr.io/labring/fastgpt-sandbox:latest
    container_name: storyforge-sandbox
    restart: unless-stopped

  fastgpt-plugin:
    image: ghcr.io/labring/fastgpt-plugin:latest
    container_name: storyforge-fastgpt-plugin
    restart: unless-stopped

  cli-proxy-api:
    image: ${CLIPROXY_IMAGE:-storyforge/cli-proxy-api:patched}
    container_name: storyforge-cliproxyapi
    restart: unless-stopped
    ports:
      - "8317:8317"
      - "8085:8085"