#!/usr/bin/env bash
# Refresh model weights + minimal MSAI inference runtime in this repo.
set -euo pipefail

ROOT="$(cd "$(dirname "$0")/../.." && pwd)"
HF="$ROOT/kiosk_vanilla"
MSAI="$ROOT/MSAI_Text_Generation"

RUNTIME_FILES=(
  src/__init__.py
  src/paths.py
  src/inference/__init__.py
  src/inference/generate.py
  src/inference/answer_quality.py
  src/inference/types.py
  src/data/__init__.py
  src/data/format.py
  src/data/kiosk_answer.py
  src/data/kiosk_actions.py
  src/data/kiosk_schemas.py
  src/data/kiosk_tool_schemas.json
  src/executor/serialize.py
  src/model/__init__.py
  src/model/config.py
  src/model/transformer.py
)

if [[ ! -f "$MSAI/checkpoints/best.pt" ]]; then
  echo "Expected MSAI_Text_Generation/checkpoints/best.pt." >&2
  exit 1
fi

mkdir -p "$HF/models" "$HF/MSAI_Text_Generation" "$HF/backend/storage"
cp "$MSAI/checkpoints/best.pt" "$HF/models/best.pt"
rm -rf "$HF/models/tokenizer"
cp -R "$MSAI/tokenizer" "$HF/models/tokenizer"

rm -rf "$HF/MSAI_Text_Generation/src"
mkdir -p "$HF/MSAI_Text_Generation/src"
for rel in "${RUNTIME_FILES[@]}"; do
  dest_dir="$HF/MSAI_Text_Generation/$(dirname "$rel")"
  mkdir -p "$dest_dir"
  cp "$MSAI/$rel" "$HF/MSAI_Text_Generation/$rel"
done

touch "$HF/backend/storage/.gitkeep"

echo "Refreshed weights + MSAI runtime in $HF"
echo "Checkpoint: $(ls -lh "$HF/models/best.pt" | awk '{print $5, $6, $7, $8}')"
echo "Tokenizer:  $(ls -lh "$HF/models/tokenizer/tokenizer.json" | awk '{print $5, $6, $7, $8}')"