import argparse
from src.blu.rag import build_index

def main():
    ap = argparse.ArgumentParser()
    ap.add_argument('--input', type=str, default='data/corpus')
    ap.add_argument('--index', type=str, default='.config/index')
    ap.add_argument('--model', type=str, default=None)
    args = ap.parse_args()
    n = build_index(args.input, args.index, args.model) if args.model else build_index(args.input, args.index)
    print(f"Indexed {n} chunks.")

if __name__ == '__main__':
    main()