#!/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 MiniCPM-V 4.5 GPU variant
echo -e "${GREEN}Building MiniCPM-V 4.5 GPU variant...${NC}"
docker build \
    -f Dockerfile_minicpm45v_ollama_gpu_VRAM9GB \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v-gpu \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:latest \
    .

# Build Qwen3-VL GPU variant
echo -e "${GREEN}Building Qwen3-VL-30B-A3B GPU variant...${NC}"
docker build \
    -f Dockerfile_qwen3vl_ollama_gpu_VRAM20GB \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:qwen3vl \
    .

# Build Nanonets-OCR GPU variant
echo -e "${GREEN}Building Nanonets-OCR-s GPU variant...${NC}"
docker build \
    -f Dockerfile_nanonets_vllm_gpu_VRAM10GB \
    -t ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:nanonets-ocr \
    .

echo -e "${GREEN}All images built successfully!${NC}"
echo ""
echo "Available images:"
echo "  MiniCPM-V 4.5 (Ollama, ~9GB VRAM):"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:minicpm45v"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:latest"
echo ""
echo "  Qwen3-VL-30B-A3B (Ollama, ~20GB VRAM):"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:qwen3vl"
echo ""
echo "  Nanonets-OCR-s (vLLM, ~10GB VRAM):"
echo "  - ${REGISTRY}/${NAMESPACE}/${IMAGE_NAME}:nanonets-ocr"