generate_spec_agent.py

#!/usr/bin/env python3
"""Generate collaborative spec draft + Cursor Agent prompt (no Ollama)."""

from __future__ import annotations

import argparse
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parent.parent
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

from src.agent_spec import (  # noqa: E402
    agent_prompt_path,
    build_agent_prompt_document,
)
from src.spec_generator import load_extraction_report  # noqa: E402
from src.spec_skeleton import build_draft_spec, draft_spec_path  # noqa: E402


def main() -> int:
    parser = argparse.ArgumentParser(
        description=(
            "Build a collaborative DRAFT_SPEC and Cursor AGENT_PROMPT "
            "from extraction JSON — no LLM call."
        )
    )
    parser.add_argument(
        "extraction",
        type=Path,
        help="Path to extractions/<name>_extraction.json",
    )
    parser.add_argument(
        "--draft-only",
        action="store_true",
        help="Write only specs/<name>_DRAFT_SPEC.md",
    )
    parser.add_argument(
        "--prompt-only",
        action="store_true",
        help="Write only specs/<name>_AGENT_PROMPT.md",
    )
    parser.add_argument(
        "--lang",
        default="en",
        choices=["fr", "en"],
        help="Language hint for the agent task (default: en)",
    )
    args = parser.parse_args()

    if not args.extraction.is_file():
        print(f"Error: file not found — {args.extraction}", file=sys.stderr)
        return 1

    report = load_extraction_report(args.extraction)
    draft_path = draft_spec_path(args.extraction)
    prompt_path = agent_prompt_path(args.extraction)

    write_draft = not args.prompt_only
    write_prompt = not args.draft_only

    if write_draft:
        draft_path.parent.mkdir(parents=True, exist_ok=True)
        draft_path.write_text(
            build_draft_spec(report, source_json=args.extraction),
            encoding="utf-8",
        )
        print(f"  -> {draft_path}")
        print("     Open, complete « Your answer / Your correction », then validate.")

    if write_prompt:
        prompt_path.parent.mkdir(parents=True, exist_ok=True)
        prompt_path.write_text(
            build_agent_prompt_document(
                report,
                extraction_path=args.extraction,
                draft_path=draft_path if draft_path.exists() else None,
                language=args.lang,
            ),
            encoding="utf-8",
        )
        print(f"  -> {prompt_path}")
        print("     Paste into Cursor Agent with the extraction JSON attached.")

    if write_draft and write_prompt:
        print("\nWorkflow:")
        print("  1. Fill user sections in DRAFT_SPEC (your Excel knowledge)")
        print("  2. Run Cursor Agent with AGENT_PROMPT + extraction JSON")
        print("  3. Review FUNCTIONAL_SPEC, merge corrections, check validation boxes")

    return 0


if __name__ == "__main__":
    raise SystemExit(main())