ai-toolkit/build_and_push_docker

#!/usr/bin/env bash

# Extract version from version.py
if [ -f "version.py" ]; then
    VERSION=$(python3 -c "from version import VERSION; print(VERSION)")
    echo "Building version: $VERSION"
else
    echo "Error: version.py not found. Please create a version.py file with VERSION defined."
    exit 1
fi

echo "Docker builds from the repo, not this dir. Make sure changes are pushed to the repo."
echo "Building version: $VERSION and latest"
# wait 2 seconds
sleep 2

# Build the image with cache busting
docker build --build-arg CACHEBUST=$(date +%s) -t aitoolkit:$VERSION -f docker/Dockerfile .

# Tag with version and latest
docker tag aitoolkit:$VERSION ostris/aitoolkit:$VERSION
docker tag aitoolkit:$VERSION ostris/aitoolkit:latest

# Push both tags
echo "Pushing images to Docker Hub..."
docker push ostris/aitoolkit:$VERSION
docker push ostris/aitoolkit:latest

echo "Successfully built and pushed ostris/aitoolkit:$VERSION and ostris/aitoolkit:latest"