#!/bin/bash
set -e

MODELS_PATH="./models"
OUTPUT_DIR="./compiled-models"
IMAGE_NAME="rkllm-builder"
CONTAINER_NAME="rkllm-compile-$$"

mkdir -p "$OUTPUT_DIR"

echo "🏗️ Сборка образа для компиляции..."
docker build -f Dockerfile -t "$IMAGE_NAME" .

mkdir -p "$MODELS_PATH"
cd "$MODELS_PATH"
git clone https://huggingface.co/simaai/Qwen3-4B-Instruct-2507-a16w4 && \
    cd Qwen3-4B-Instruct-2507-a16w4 && git lfs pull

echo "⚙️ Запуск компиляции (может занять 30-60 минут)..."
docker run --rm \
    --name "$CONTAINER_NAME" \
    -v "$(pwd)/"$MODELS_PATH":/models" \
    -v "$(pwd)/$OUTPUT_DIR:/output" \
    "$IMAGE_NAME"

echo "✅ Модель сохранена в: $OUTPUT_DIR"