#!/bin/bash
set -e

# Configuration
REGISTRY="code.foss.global"
NAMESPACE="host.today"
IMAGE_NAME="ht-docker-ai"

# Colors for output
GREEN='\033[0;32m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

echo -e "${BLUE}Building ht-docker-ai images...${NC}"

# Build GPU variant
echo -e "${GREEN}Building MiniCPM-V 4.5 GPU variant...${NC}"
docker build \
    -f Dockerfile_minicpm45v \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-gpu \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:latest \
    .

# Build CPU variant
echo -e "${GREEN}Building MiniCPM-V 4.5 CPU variant...${NC}"
docker build \
    -f Dockerfile_minicpm45v_cpu \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-cpu \
    .

# Build PaddleOCR GPU variant
echo -e "${GREEN}Building PaddleOCR GPU variant...${NC}"
docker build \
    -f Dockerfile_paddleocr \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-gpu \
    .

# Build PaddleOCR CPU variant
echo -e "${GREEN}Building PaddleOCR CPU variant...${NC}"
docker build \
    -f Dockerfile_paddleocr_cpu \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-cpu \
    .

echo -e "${GREEN}All images built successfully!${NC}"
echo ""
echo "Available images:"
echo "  MiniCPM-V 4.5:"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v (GPU)"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-cpu (CPU)"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:latest (GPU)"
echo ""
echo "  PaddleOCR:"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr (GPU)"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-gpu (GPU)"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:paddleocr-cpu (CPU)"