#!/usr/bin/env python3
"""AION unified local assistant entrypoint."""
from __future__ import annotations
import os
from pathlib import Path

BASE = Path(__file__).resolve().parent


def generate(prompt: str) -> str:
    """Generate an AION answer for a prompt."""
    cwd = os.getcwd()
    try:
        os.chdir(BASE)
        from unified_learning_ai import answer
        return answer(Path("outputs/unified_learning_ai"), prompt)
    finally:
        os.chdir(cwd)


def main():
    import argparse
    ap = argparse.ArgumentParser(description="AION unified assistant")
    ap.add_argument("prompt", nargs="*", default=["hola"])
    args = ap.parse_args()
    print(generate(" ".join(args.prompt)))

if __name__ == "__main__":
    main()