src/agentic_multiwriter/cli/main.py

from __future__ import annotations

import argparse
import textwrap

from ..graph.router import run_pipeline
from ..tools import get_logger

logger = get_logger()


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(
        description="Agentic Multiwriter - multi-agent research & writing pipeline."
    )
    parser.add_argument(
        "--topic",
        required=True,
        help="Topic to research and write about.",
    )
    parser.add_argument(
        "--mode",
        default="blog",
        choices=["blog", "research_summary", "linkedin_post"],
        help="Output mode (default: blog).",
    )
    return parser.parse_args()


def main() -> None:
    args = parse_args()
    topic = args.topic
    mode = args.mode

    logger.info("CLI: running pipeline for topic='%s', mode='%s'", topic, mode)
    state = run_pipeline(topic=topic, mode=mode)

    final_output = state.get("final_output", "") or ""
    meta = state.get("meta", {}) or {}

    print("\n" + "=" * 80)
    print(f"TOPIC: {topic}")
    print(f"MODE:  {mode}")
    print("=" * 80 + "\n")
    print(textwrap.dedent(final_output).strip())
    print("\n" + "-" * 80)
    print("Meta:", meta)


if __name__ == "__main__":
    main()