#!/bin/bash
# Second batch of training queries — covers gaps from seed_traces.sh:
#   - Contract law sections we haven't asked yet (3, 4, 6, 7, 9, 10, 12, 14, 15, 17, 18, 21, 22, 23, 25, 28, 29, 30, 31, 32, 39)
#   - Variations on existing questions (rephrased — different surface, same answer)
#   - Applied scenarios (case studies)
# Plus 6 deliberately ambiguous questions to teach "I don't know" responses.
set -e

API="${API:-http://127.0.0.1:8000}"
MAX=""
SLEEP="0.8"
while [[ $# -gt 0 ]]; do
    case "$1" in
        --max)   MAX="$2"; shift 2 ;;
        --slow)  SLEEP="2"; shift ;;
        *)       echo "Unknown arg: $1"; exit 1 ;;
    esac
done

QUESTIONS=$(cat << 'QEOF'
מה אומר סעיף 3 לחוק החוזים?
מה אומר סעיף 4 לחוק החוזים?
מה אומר סעיף 6 לחוק החוזים?
מה אומר סעיף 7 לחוק החוזים?
מה אומר סעיף 9 לחוק החוזים?
מה אומר סעיף 10 לחוק החוזים?
מה אומר סעיף 12 לחוק החוזים?
מה אומר סעיף 14 לחוק החוזים?
מה אומר סעיף 15 לחוק החוזים?
מה אומר סעיף 17 לחוק החוזים?
מה אומר סעיף 18 לחוק החוזים?
מה אומר סעיף 21 לחוק החוזים?
מה אומר סעיף 22 לחוק החוזים?
מה אומר סעיף 23 לחוק החוזים?
מה אומר סעיף 25 לחוק החוזים?
מה אומר סעיף 28 לחוק החוזים?
מה אומר סעיף 29 לחוק החוזים?
מה אומר סעיף 30 לחוק החוזים?
מה אומר סעיף 31 לחוק החוזים?
מה אומר סעיף 32 לחוק החוזים?
מה אומר סעיף 39 לחוק החוזים?
מה אומר סעיף 7 לחוק התרופות?
מה אומר סעיף 8 לחוק התרופות?
מה אומר סעיף 9 לחוק התרופות?
מה אומר סעיף 12 לחוק התרופות?
מה אומר סעיף 13 לחוק התרופות?
הסבר על קיום בתום לב
מה זה גמירת דעת?
מה זה מסויימות?
מה ההבדל בין הצעה להזמנה להציע הצעות?
איך נכרת חוזה לפי דיני החוזים הישראליים?
מתי הצעה פוקעת?
מה זה קיבול בהתנהגות?
מה זה תניה מקפחת?
מה זה ביטול חלקי?
מה זה השלמת פרטים בחוזה?
מה זה חוזה על תנאי?
איך פועל קיזוז חיובים?
מה התרופות במכר פגום?
מה ההבדל בין שכירות למכירה?
מה החובות של המוכר במכר?
מתי קונה רשאי לבטל מכר?
מה אומר חוק הגנת הצרכן?
מה הקשר בין סעיף 39 לסעיף 61?
האם מותר ויתור על תרופות?
מה זה חזקת תקנה לציבור?
איך מוכיחים נזק בהפרת חוזה?
מה זה חוק חוזה הביטוח?
מה זה הסכם פשרה?
מתי בית משפט יסרב לאכוף חוזה?
מה ההבדל בין חוזה אחיד לחוזה צרכני?
איזה ראיות מקבל בית המשפט?
איזה דברים אסורים בחוזה?
איך מסתיים חוזה?
מה זה ערבות אישית?
האם ילד יכול לחתום על חוזה?
מה ההבדל בין חוזה רכישה לחוזה שירות?
מה אומר סעיף 100?
מה ההבדל בין הסכם לזכרון דברים?
מה זה תניית בוררות?
מה זה תניית שיפוט?
QEOF
)

QARRAY=()
while IFS= read -r line; do
    QARRAY+=("$line")
done <<< "$QUESTIONS"
if [ -n "$MAX" ]; then
    QARRAY=("${QARRAY[@]:0:$MAX}")
fi
TOTAL=${#QARRAY[@]}
echo "▸ Sending $TOTAL queries to $API ..."
echo ""

OK=0
FAIL=0
START=$(date +%s)
i=0
for Q in "${QARRAY[@]}"; do
    i=$((i+1))
    [ -z "$Q" ] && continue
    BODY=$(python3 -c "import json,sys; print(json.dumps({'query': sys.argv[1], 'top_k': 3}, ensure_ascii=False))" "$Q")
    HTTP=$(curl -s -o /tmp/seed_resp.json -w "%{http_code}" \
        -X POST "$API/v1/query" \
        -H "Content-Type: application/json; charset=utf-8" \
        --data-binary "$BODY")
    if [ "$HTTP" = "200" ]; then
        OK=$((OK+1))
        SHORT=$(echo "$Q" | python3 -c "import sys; t=sys.stdin.read().strip(); print(t[:60])")
        printf "  ✓ %3d/%d  %s\n" "$i" "$TOTAL" "$SHORT"
    else
        FAIL=$((FAIL+1))
        SHORT=$(echo "$Q" | python3 -c "import sys; t=sys.stdin.read().strip(); print(t[:50])")
        printf "  ✗ %3d/%d  HTTP %s — %s\n" "$i" "$TOTAL" "$HTTP" "$SHORT"
    fi
    sleep "$SLEEP"
done

ELAPSED=$(( $(date +%s) - START ))
echo ""
echo "Done in ${ELAPSED}s — ✓ $OK   ✗ $FAIL"
echo ""
echo "▸ Training data summary:"
curl -s "$API/v1/training_data/summary" | python3 -m json.tool