main.py

from __future__ import annotations

import argparse
import sys

from graph.runner import run_task
from utils.logger import get_logger

logger = get_logger()


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Supervisor multi-agent team")
    parser.add_argument("task", nargs="*", help="Question or task for the team")
    parser.add_argument("--quiet", action="store_true", help="Hide node progress logs")
    return parser.parse_args()


def ask_once(task: str) -> None:
    result = run_task(task)
    print("\n=== Final answer ===\n")
    print(result.get("final_answer") or "No answer was produced.")


def interactive() -> None:
    print("Multi-agent team ready. Type a task, or 'exit' to quit.\n")
    while True:
        try:
            task = input("You> ").strip()
        except (EOFError, KeyboardInterrupt):
            print()
            return
        if not task or task.lower() in {"exit", "quit"}:
            return
        ask_once(task)
        print()


def main() -> int:
    args = parse_args()
    if args.quiet:
        logger.setLevel(40)

    task = " ".join(args.task).strip()
    if task:
        ask_once(task)
        return 0

    interactive()
    return 0


if __name__ == "__main__":
    sys.exit(main())