Heart-Disease-Prediction/docker-compose.yml at main · BimsaraIUHS/Heart-Disease-Prediction · 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
# Docker Compose configuration for Heart Disease Prediction Application
# Defines services for API and Streamlit web app

version: '3.8'

services:
  # ==========================================================================
  # FastAPI Backend Service
  # ==========================================================================
  api:
    build:
      context: .
      dockerfile: Dockerfile
      target: runtime
    container_name: heart-disease-api
    ports:
      - "${API_PORT:-8000}:8000"
    environment:
      - API_HOST=0.0.0.0
      - API_PORT=8000
      - PYTHONPATH=/app
    volumes:
      # Mount models directory for easy model updates without rebuild
      - ./models:/app/models:ro
      - ./data:/app/data:ro
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 10s
    restart: unless-stopped
    networks:
      - heart-disease-network

  # ==========================================================================
  # Streamlit Frontend Service
  # ==========================================================================
  streamlit:
    build:
      context: .
      dockerfile: Dockerfile
      target: runtime
    container_name: heart-disease-streamlit
    ports:
      - "${STREAMLIT_PORT:-8501}:8501"
    environment:
      - STREAMLIT_SERVER_PORT=8501
      - STREAMLIT_SERVER_ADDRESS=0.0.0.0
      - STREAMLIT_SERVER_HEADLESS=true
      - STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
      # API URL points to the api service within Docker network
      - API_URL=http://api:8000
    command: >
      streamlit run app/streamlit_app.py
      --server.port=8501
      --server.address=0.0.0.0
      --server.headless=true
      --browser.gatherUsageStats=false
    depends_on:
      api:
        condition: service_healthy
    restart: unless-stopped
    networks:
      - heart-disease-network

# =============================================================================
# Networks
# =============================================================================
networks:
  heart-disease-network:
    driver: bridge
    name: heart-disease-network

# =============================================================================
# Volumes (optional - for persistent data)
# =============================================================================
volumes:
  model-data:
    driver: local