#!/bin/bash
# PM-Sample-Bundle: erzeugt fuer 5 hochbewertete Anträge je eine PM
# (Klassisch + Thread) zur manuellen Qualitätssichtung.
#
# Manueller Aufruf:
#   ./scripts/pm-sample-bundle.sh gwoe-antragspruefer-dev
#
# Bei mehrfachem Aufruf werden bestehende PMs nicht regeneriert
# (Idempotenz pro drucksache+news_url+style).

set -euo pipefail

CONTAINER="${1:-gwoe-antragspruefer-dev}"
N_SAMPLES="${N_SAMPLES:-5}"

if ! docker ps --format '{{.Names}}' | grep -q "^${CONTAINER}$"; then
    echo "$(date -Iseconds) SKIP — ${CONTAINER} is not running"
    exit 0
fi

echo "$(date -Iseconds) START pm-sample-bundle (n=${N_SAMPLES})"

docker exec -i -e N_SAMPLES="$N_SAMPLES" "$CONTAINER" python <<'EOF'
import os
import asyncio
import sqlite3
from app.presse_generator import generate_draft


N_SAMPLES = int(os.environ.get("N_SAMPLES", "5"))


def find_candidates(n: int) -> list:
    """Top-N hochbewertete Anträge ohne existierende PM (style=pm)."""
    conn = sqlite3.connect("/app/data/gwoe-antraege.db")
    try:
        cur = conn.execute("""
            SELECT a.drucksache, a.bundesland, a.gwoe_score
            FROM assessments a
            WHERE a.gwoe_score >= 7.0
            ORDER BY a.gwoe_score DESC, a.datum DESC
        """)
        out = []
        for ds, bl, score in cur.fetchall():
            existing = conn.execute(
                "SELECT id FROM presse_drafts WHERE drucksache=? AND style='pm' LIMIT 1",
                (ds,),
            ).fetchone()
            if existing:
                continue
            news_match = conn.execute("""
                SELECT n.url FROM news_articles n
                ORDER BY n.datum DESC LIMIT 1
            """).fetchone()
            if not news_match:
                continue
            out.append({"drucksache": ds, "bundesland": bl,
                        "score": score, "news_url": news_match[0]})
            if len(out) >= n:
                break
        return out
    finally:
        conn.close()


async def main() -> None:
    cands = find_candidates(N_SAMPLES)
    if not cands:
        print("  keine Kandidaten gefunden")
        return
    print(f"  {len(cands)} Kandidaten:")
    for i, c in enumerate(cands, 1):
        print(f"  [{i}/{len(cands)}] {c['bundesland']}/{c['drucksache']} (GWÖ {c['score']})")
        try:
            pm = await generate_draft(
                drucksache=c["drucksache"], news_url=c["news_url"], style="pm",
            )
            print(f"     PM:     id={pm.get('id')}  was_existing={pm.get('_was_existing')}")
            thread = await generate_draft(
                drucksache=c["drucksache"], news_url=c["news_url"], style="thread",
            )
            print(f"     Thread: id={thread.get('id')}  was_existing={thread.get('_was_existing')}")
        except Exception as e:
            print(f"     ERROR:  {str(e)[:120]}")


asyncio.run(main())
EOF

echo "$(date -Iseconds) END pm-sample-bundle"