run.sh

#!/usr/bin/env bash
set -e

# ─────────────────────────────────────────────────────────────
#  ModelFleet — One-command launcher
#  Builds and runs the full platform with Docker Compose.
#  Usage: ./run.sh
# ─────────────────────────────────────────────────────────────

BOLD='\033[1m'
GREEN='\033[0;32m'
BLUE='\033[0;34m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
DIM='\033[2m'
NC='\033[0m'

banner() {
  echo ""
  echo -e "${BLUE}${BOLD}"
  echo "  ╔══════════════════════════════════════════════╗"
  echo "  ║                                              ║"
  echo "  ║   ModelFleet                                 ║"
  echo "  ║   Kubernetes ML Model Serving Platform       ║"
  echo "  ║                                              ║"
  echo "  ╚══════════════════════════════════════════════╝"
  echo -e "${NC}"
}

info()  { echo -e "  ${BLUE}▸${NC} $1"; }
ok()    { echo -e "  ${GREEN}✓${NC} $1"; }
warn()  { echo -e "  ${YELLOW}!${NC} $1"; }
fail()  { echo -e "  ${RED}✗${NC} $1"; exit 1; }

banner

# ── Check Docker ─────────────────────────────────────────────
info "Checking Docker..."
if ! command -v docker &> /dev/null; then
  fail "Docker is not installed. Get it at https://docker.com/products/docker-desktop"
fi

if ! docker info &> /dev/null; then
  warn "Docker daemon is not running. Starting Docker Desktop..."
  if [[ "$OSTYPE" == "darwin"* ]]; then
    open -a Docker
  fi
  echo -ne "  ${DIM}Waiting for Docker"
  for i in $(seq 1 30); do
    if docker info &> /dev/null; then break; fi
    echo -n "."
    sleep 2
  done
  echo -e "${NC}"
  docker info &> /dev/null || fail "Could not start Docker. Please start it manually."
fi
ok "Docker is running"

# ── Check Docker Compose ────────────────────────────────────
if docker compose version &> /dev/null; then
  COMPOSE="docker compose"
elif command -v docker-compose &> /dev/null; then
  COMPOSE="docker-compose"
else
  fail "Docker Compose is not available. Update Docker Desktop or install docker-compose."
fi
ok "Docker Compose available"

# ── Build & Start ───────────────────────────────────────────
echo ""
info "Building containers (this may take a few minutes on first run)..."
echo ""
$COMPOSE build

echo ""
info "Starting ModelFleet..."
$COMPOSE up -d

# ── Wait for services ───────────────────────────────────────
echo ""
info "Waiting for services to be ready..."

# Wait for gateway
for i in $(seq 1 60); do
  if curl -sf http://localhost:8080/health > /dev/null 2>&1; then
    break
  fi
  sleep 2
done

if ! curl -sf http://localhost:8080/health > /dev/null 2>&1; then
  warn "Gateway is taking longer than expected. Check: docker compose logs gateway"
else
  ok "Gateway is healthy"
fi

# Check models
for i in $(seq 1 10); do
  MODELS=$(curl -sf http://localhost:8080/models 2>/dev/null || echo "")
  if echo "$MODELS" | grep -q "sentiment"; then
    break
  fi
  sleep 3
done
ok "Models are registered"

# ── Done ────────────────────────────────────────────────────
echo ""
echo -e "${GREEN}${BOLD}  ══════════════════════════════════════════════"
echo -e "   ModelFleet is running!"
echo -e "  ══════════════════════════════════════════════${NC}"
echo ""
echo -e "  ${BOLD}Dashboard:${NC}  http://localhost:8080"
echo -e "  ${BOLD}API Docs:${NC}   http://localhost:8080/docs"
echo -e "  ${BOLD}Health:${NC}     http://localhost:8080/health"
echo ""
echo -e "  ${DIM}Try a prediction:${NC}"
echo -e "  ${DIM}curl -X POST http://localhost:8080/models/sentiment/v1/predict \\${NC}"
echo -e "  ${DIM}  -H 'Content-Type: application/json' \\${NC}"
echo -e "  ${DIM}  -d '{\"input_data\": {\"text\": \"I love this product!\"}}'${NC}"
echo ""
echo -e "  ${DIM}Stop:  docker compose down${NC}"
echo -e "  ${DIM}Logs:  docker compose logs -f${NC}"
echo ""

# ── Open browser ────────────────────────────────────────────
if [[ "$OSTYPE" == "darwin"* ]]; then
  open http://localhost:8080
elif command -v xdg-open &> /dev/null; then
  xdg-open http://localhost:8080
fi