Paper2Any/docker-compose.yml at main · OpenDCAI/Paper2Any · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
services:
  paper2any-backend:
    build:
      context: .
      dockerfile: Dockerfile
      args:
        INSTALL_CUDA: ${DOCKER_BACKEND_INSTALL_CUDA:-0}
        PYTHON_BASE_IMAGE: ${PYTHON_BASE_IMAGE:-python:3.11-slim}
    image: paper2any-backend:local
    container_name: paper2any-backend
    env_file:
      - ./fastapi_app/.env
    environment:
      PYTHONUNBUFFERED: "1"
      PAPER2ANY_RUNTIME_TMPDIR: /app/outputs/system/tmp
      RMBG_MODEL_PATH: ${RMBG_MODEL_PATH:-/app/models/RMBG-2.0}
      SAM3_SERVER_URLS: ${SAM3_SERVER_URLS:-}
      SAM3_HOME: ${SAM3_HOME:-}
      SAM3_CHECKPOINT_PATH: ${SAM3_CHECKPOINT_PATH:-/app/models/sam3/sam3.pt}
      SAM3_BPE_PATH: ${SAM3_BPE_PATH:-/app/models/sam3/bpe_simple_vocab_16e6.txt.gz}
      PAPER2DRAWIO_SAM3_CHECKPOINT_PATH: ${SAM3_CHECKPOINT_PATH:-/app/models/sam3/sam3.pt}
      PAPER2DRAWIO_SAM3_BPE_PATH: ${SAM3_BPE_PATH:-/app/models/sam3/bpe_simple_vocab_16e6.txt.gz}
    command:
      - python
      - -m
      - uvicorn
      - fastapi_app.main:app
      - --host
      - 0.0.0.0
      - --port
      - "${DOCKER_BACKEND_BIND_PORT:-8000}"
      - --workers
      - "${DOCKER_APP_WORKERS:-1}"
      - --log-level
      - info
    volumes:
      - ./outputs:/app/outputs
      - ./models:/app/models
      - ./data:/app/data
      - ./database:/app/database
      - ./logs:/app/logs
      - ./raw_data_store:/app/raw_data_store
      - ./rebuttal_sessions:/app/rebuttal_sessions
      - ./fastapi_app/.env:/app/fastapi_app/.env:ro
    ports:
      - "${BACKEND_PORT:-8000}:${DOCKER_BACKEND_BIND_PORT:-8000}"
    healthcheck:
      test: ["CMD", "curl", "-fsS", "http://127.0.0.1:${DOCKER_BACKEND_BIND_PORT:-8000}/health"]
      interval: 20s
      timeout: 5s
      retries: 10
      start_period: 30s
    restart: unless-stopped

  paper2any-frontend:
    build:
      context: .
      dockerfile: frontend-workflow/Dockerfile
      args:
        NODE_BASE_IMAGE: ${NODE_BASE_IMAGE:-node:20-alpine}
        NGINX_BASE_IMAGE: ${NGINX_BASE_IMAGE:-nginx:alpine}
        VITE_API_KEY: ${VITE_API_KEY:-}
        VITE_API_BASE_URL: ${VITE_API_BASE_URL:-}
        VITE_DEFAULT_LLM_API_URL: ${VITE_DEFAULT_LLM_API_URL:-}
        VITE_LLM_API_URLS: ${VITE_LLM_API_URLS:-}
        VITE_SUPABASE_URL: ${VITE_SUPABASE_URL:-}
        VITE_SUPABASE_ANON_KEY: ${VITE_SUPABASE_ANON_KEY:-}
    image: paper2any-frontend:local
    container_name: paper2any-frontend
    depends_on:
      paper2any-backend:
        condition: service_healthy
    environment:
      NGINX_LISTEN_PORT: "${DOCKER_FRONTEND_BIND_PORT:-80}"
      BACKEND_UPSTREAM_URL: "${BACKEND_UPSTREAM_URL:-http://paper2any-backend:${DOCKER_BACKEND_BIND_PORT:-8000}}"
    ports:
      - "${FRONTEND_PORT:-3000}:${DOCKER_FRONTEND_BIND_PORT:-80}"
    healthcheck:
      test: ["CMD", "wget", "-qO-", "http://127.0.0.1:${DOCKER_FRONTEND_BIND_PORT:-80}/"]
      interval: 20s
      timeout: 5s
      retries: 10
      start_period: 10s
    restart: unless-stopped

  paper2any-sam3:
    profiles: ["sam3"]
    build:
      context: .
      dockerfile: Dockerfile
      args:
        INSTALL_CUDA: 1
        PYTHON_BASE_IMAGE: ${PYTHON_BASE_IMAGE:-python:3.11-slim}
    image: paper2any-backend:local-gpu
    container_name: paper2any-sam3
    environment:
      PYTHONUNBUFFERED: "1"
      SAM3_HOME: ${SAM3_HOME:-/app/models/sam3-official/sam3}
      SAM3_CHECKPOINT_PATH: ${SAM3_CHECKPOINT_PATH:-/app/models/sam3/sam3.pt}
      SAM3_BPE_PATH: ${SAM3_BPE_PATH:-/app/models/sam3/bpe_simple_vocab_16e6.txt.gz}
    command:
      - python
      - -m
      - dataflow_agent.toolkits.model_servers.sam3_server
      - --host
      - 0.0.0.0
      - --port
      - "${DOCKER_SAM3_BIND_PORT:-8021}"
      - --checkpoint
      - "${SAM3_CHECKPOINT_PATH:-/app/models/sam3/sam3.pt}"
      - --bpe
      - "${SAM3_BPE_PATH:-/app/models/sam3/bpe_simple_vocab_16e6.txt.gz}"
      - --device
      - cuda
    volumes:
      - ./models:/app/models
      - ./logs:/app/logs
    ports:
      - "${SAM3_PORT:-8021}:${DOCKER_SAM3_BIND_PORT:-8021}"
    healthcheck:
      test: ["CMD", "curl", "-fsS", "http://127.0.0.1:${DOCKER_SAM3_BIND_PORT:-8021}/health"]
      interval: 20s
      timeout: 5s
      retries: 10
      start_period: 30s
    gpus: all
    restart: unless-stopped