#!/bin/bash

# Build script for RunPod deployment
echo "Building RunPod Docker image with MinerU models..."

# Set variables
IMAGE_NAME="mineru-runpod"
TAG="latest"
DOCKER_REPO="marcosremar2/mineru-runpod"  # Change to your Docker Hub username

# Build the image
echo "Building Docker image..."
docker build -f Dockerfile.runpod -t ${IMAGE_NAME}:${TAG} .

# Tag for Docker Hub
docker tag ${IMAGE_NAME}:${TAG} ${DOCKER_REPO}:${TAG}

echo "Build complete!"
echo ""
echo "To test locally:"
echo "docker run --rm -p 8000:8000 ${IMAGE_NAME}:${TAG}"
echo ""
echo "To push to Docker Hub:"
echo "docker login"
echo "docker push ${DOCKER_REPO}:${TAG}"
echo ""
echo "Docker image size:"
docker images ${IMAGE_NAME}:${TAG} --format "table {{.Repository}}\t{{.Tag}}\t{{.Size}}"