rover-ui/docker-compose.yml at main · ByteSizedRobotics/rover-ui · 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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
services:
  web-app:
    image: bytesizedrobotics/rover-ui:latest
    ports:
      - '3000:3000'
    environment:
      - DATABASE_URL=postgres://root:mysecretpassword@postgres:5432/local
      - ORIGIN=http://127.0.0.1:3000 # Change to URL unless testing locally
    profiles:
      - default
      - gpu
    depends_on:
      - postgres

  postgres:
    image: postgis/postgis
    platform: linux/amd64
    environment:
      POSTGRES_USER: root
      POSTGRES_PASSWORD: mysecretpassword
      POSTGRES_DB: local
    volumes:
      - ./drizzle/schema.sql:/docker-entrypoint-initdb.d/schema.sql
    ports:
      - '5432:5432'

  # Default ai app without GPU support for CSI cam 1
  ai-pothole-app-csi1:
    image: bytesizedrobotics/ai-pothole-app:latest
    platform: linux/amd64
    environment:
      - RASPI_IP=100.85.202.20
      - WEBRTC_PORT=8765
      - API_IP=web-app
      - API_PORT=3000
      - ROVER_ID=1
    # volumes:
    #   - ./models:/models
    profiles:
      - default
    depends_on:
      - web-app
      - postgres

  # Default ai app without GPU support for CSI cam 2
  ai-pothole-app-csi2:
    image: bytesizedrobotics/ai-pothole-app:latest
    platform: linux/amd64
    environment:
      - RASPI_IP=100.85.202.20
      - WEBRTC_PORT=8767
      - API_IP=web-app
      - API_PORT=3000
      - ROVER_ID=1
    # volumes:
    #   - ./models:/models
    profiles:
      - default
    depends_on:
      - web-app
      - postgres

  # CSI CAM 1: Default ai app with GPU support: can run this with docker compose --profile gpu up
  ai-pothole-app-csi1-gpu:
    image: bytesizedrobotics/ai-pothole-app:latest
    platform: linux/amd64
    # Enable GPU support
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      - RASPI_IP=100.85.202.20
      - WEBRTC_PORT=8765
      - API_IP=web-app
      - API_PORT=3000
      - ROVER_ID=1
    # volumes:
    #   - ./models:/models
    profiles: ['gpu']
    depends_on:
      - app
      - postgres

  # CSI CAM 2: Default ai app with GPU support: can run this with docker compose --profile gpu up
  ai-pothole-app-csi2-gpu:
    image: bytesizedrobotics/ai-pothole-app:latest
    platform: linux/amd64
    # Enable GPU support
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      - RASPI_IP=100.85.202.20
      - WEBRTC_PORT=8767
      - API_IP=web-app
      - API_PORT=3000
      - ROVER_ID=1
    # volumes:
    #   - ./models:/models
    profiles: ['gpu']
    depends_on:
      - app
      - postgres

  # Default ai app without GPU support with local host docker internal connection
  ai-pothole-app-csi1-local:
    image: bytesizedrobotics/ai-pothole-app:latest
    platform: linux/amd64
    environment:
      - RASPI_IP=100.85.202.20
      - WEBRTC_PORT=8765
      - API_IP=host.docker.internal
      - API_PORT=5173
      - ROVER_ID=1
      - VERBOSE=true
    # volumes:
    #   - ./models:/models
    profiles:
      - local
    depends_on:
      - postgres

  ai-pothole-app-csi2-local:
    image: bytesizedrobotics/ai-pothole-app:latest
    platform: linux/amd64
    environment:
      - RASPI_IP=100.85.202.20
      - WEBRTC_PORT=8767
      - API_IP=host.docker.internal
      - API_PORT=5173
      - ROVER_ID=1
      - VERBOSE=true
    # volumes:
    #   - ./models:/models
    profiles:
      - local
    depends_on:
      - postgres