main.py

import argparse
import json
import asyncio
from dotenv import load_dotenv
from src.processor import process_document

if __name__ == "__main__":
    load_dotenv()
    parser = argparse.ArgumentParser()
    parser.add_argument("input")
    parser.add_argument("-o", "--output", required=True)
    parser.add_argument("-p", "--page", type=int)
    parser.add_argument("-nm", "--no-multi-step", dest="multi_step", action="store_false", default=True)
    args = parser.parse_args()

    data = asyncio.run(process_document(
        args.input, 
        args.page, 
        m_cls="cometapi/gpt-5-nano", 
        m_ext="cometapi/gemini-3-flash",
        multi_step=args.multi_step
    ))

    with open(args.output, "w", encoding="utf-8") as f:
        json.dump(data, f, ensure_ascii=False, indent=2)