Federated_Learning/launch-platform.sh at main · Nikhil172913832/Federated_Learning · 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
#!/bin/bash

# Federated Learning Platform Launcher
# Builds images sequentially to prevent system overload

set -e

echo "Federated Learning Platform Launcher"
echo "========================================"
echo ""
echo "This will launch a complete federated learning platform:"
echo "  - SuperLink (coordination service)"
echo "  - 3 Client containers"
echo "  - Server container"
echo "  - Real-time monitoring UI"
echo "  - MLflow tracking server"
echo ""

# Check prerequisites
echo "Checking prerequisites..."

# Check Docker
if ! docker info > /dev/null 2>&1; then
    echo "❌ Error: Docker is not running"
    exit 1
fi
echo "✓ Docker is running"

# Check Docker Compose
if ! command -v docker compose > /dev/null 2>&1; then
    echo "❌ Error: Docker Compose V2 is not installed"
    exit 1
fi
echo "✓ Docker Compose V2 is available"

# Check if complete directory exists
if [ ! -d "complete" ]; then
    echo "❌ Error: complete directory not found"
    exit 1
fi
echo "✓ Complete directory exists"

# Check if platform-ui directory exists
if [ ! -d "platform-ui" ]; then
    echo "❌ Error: platform-ui directory not found"
    exit 1
fi
echo "✓ Platform UI directory exists"

echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  BUILDING IMAGES SEQUENTIALLY"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""
echo "Building images one at a time to prevent system overload..."
echo "This is slower but more stable for systems with limited resources."
echo ""
echo "⏱  Expected time: 10-15 minutes (first run)"
echo ""

cd complete

# Stop any existing services
echo "Stopping any existing services..."
docker compose -f compose-with-ui.yml down 2>/dev/null || true
echo ""

# Build images sequentially
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  Step 1/4: Building FL application image"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
docker compose -f compose-with-ui.yml build superexec-serverapp
echo "✓ FL application image built"
echo ""

echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  Step 2/4: Building MLflow server image"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
docker compose -f compose-with-ui.yml build mlflow
echo "✓ MLflow server image built"
echo ""

echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  Step 3/4: Building platform UI image"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
docker compose -f compose-with-ui.yml build platform-ui
echo "✓ Platform UI image built"
echo ""

echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  Step 4/4: Starting all services"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
docker compose -f compose-with-ui.yml up -d
echo "✓ All services started"
echo ""

echo "Waiting for services to initialize..."
echo "This may take up to 60 seconds..."
sleep 60

echo ""
echo "Checking platform status..."
docker compose -f compose-with-ui.yml ps

echo ""
echo "✅ Platform is ready!"
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  ACCESS POINTS"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""
echo "  📊 Dashboard:  http://localhost:8050"
echo "  📈 MLflow:     http://localhost:5000"
echo "  🔗 SuperLink:  http://localhost:9093"
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  NEXT STEPS"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""
echo "  1. Open http://localhost:8050 to see the dashboard"
echo "  2. Start federated learning:"
echo ""
echo "     cd complete"
echo "     flwr run fl local-deployment --stream"
echo ""
echo "  3. View experiment results at http://localhost:5000"
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo "  USEFUL COMMANDS"
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""
echo "  View logs:     docker compose -f compose-with-ui.yml logs -f"
echo "  Stop:          docker compose -f compose-with-ui.yml down"
echo "  Restart:       docker compose -f compose-with-ui.yml restart"
echo ""
echo "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
echo ""