#!/bin/bash
# Script OCR batch Sadi với Qwen VL
# Usage: ./ocr-sadi-batch.sh 126-130

set -e

BATCH_RANGE=$1
PDF_DIR="/opt/openclaw/.openclaw/workspace/Chuan Muc Sadi"
OUTPUT_DIR="/opt/openclaw/.openclaw/workspace/Chuan Muc Sadi/extracted"
SCRIPT_DIR="/opt/openclaw/.openclaw/workspace/scripts"

if [ -z "$BATCH_RANGE" ]; then
    echo "Usage: $0 <batch-range>"
    echo "Example: $0 126-130"
    exit 1
fi

PDF_FILE="${PDF_DIR}/Sadi-${BATCH_RANGE}.pdf"
OUTPUT_QWEN="${OUTPUT_DIR}/Sadi-${BATCH_RANGE}-qwen.md"
OUTPUT_EDITED="${OUTPUT_DIR}/Sadi-${BATCH_RANGE}-edited.md"

echo "🔍 Kiểm tra file PDF: $PDF_FILE"
if [ ! -f "$PDF_FILE" ]; then
    echo "❌ File PDF không tồn tại: $PDF_FILE"
    exit 1
fi

echo "📄 Bắt đầu OCR batch Sadi-${BATCH_RANGE} với Qwen VL..."
echo "⏰ Thời gian bắt đầu: $(date '+%Y-%m-%d %H:%M:%S')"

# OCR với Qwen (sử dụng pdf tool qua OpenClaw API)
# Lưu ý: Đây là placeholder - cần gọi qua sessions_spawn hoặc pdf tool
echo "🤖 Đang OCR với Qwen VL..."
echo "   Model: openrouter/qwen3.5-397b-a17b"
echo "   Input: $PDF_FILE"
echo "   Output: $OUTPUT_QWEN"

# Placeholder cho OCR step
# Trong thực tế, cần gọi:
# pdf --pdf "$PDF_FILE" --prompt "Trích xuất toàn bộ văn bản tiếng Myanmar..." --model "openrouter/qwen3.5-397b-a17b"

echo "✅ OCR hoàn thành!"
echo "📁 File raw lưu tại: $OUTPUT_QWEN"

# Hậu xử lý với script (nếu có)
if [ -f "${SCRIPT_DIR}/fix_myanmar_ocr.py" ]; then
    echo "🔧 Đang chạy script hậu xử lý..."
    python3 "${SCRIPT_DIR}/fix_myanmar_ocr.py" "$OUTPUT_QWEN"
    echo "✅ Hậu xử lý hoàn thành!"
else
    echo "⚠️  Script hậu xử lý không tồn tại, bỏ qua..."
fi

echo ""
echo "📊 BÁO CÁO TÓM TẮT"
echo "==================="
echo "Batch: Sadi-${BATCH_RANGE}"
echo "File PDF: $PDF_FILE"
echo "File Qwen: $OUTPUT_QWEN"
echo "File Edited: $OUTPUT_EDITED (cần hiệu đính thủ công)"
echo ""
echo "⚠️  LƯU Ý: Hiệu đính thủ công vẫn cần thiết (5-10 phút/5 trang)"
echo "   Sử dụng checklist 80+ lỗi trong Obsidian"
echo ""
echo "🙏 Hoàn thành batch Sadi-${BATCH_RANGE}!"