import argparse, sys
from . import Engine
ap = argparse.ArgumentParser()
ap.add_argument("--model", required=True); ap.add_argument("--table", required=True)
ap.add_argument("--archive"); ap.add_argument("--ask"); ap.add_argument("--chat", action="store_true")
a = ap.parse_args()
eng = Engine(a.model, a.table, archive=a.archive)
if a.ask: print(eng.answer(a.ask))
elif a.chat:
    while True:
        try: q = input("you> ")
        except EOFError: break
        if not q.strip(): continue
        print("shadow>", eng.chat(q))