main.py

from query import ask_question


def run():

    print("Ready. Ask your questions (type 'exit' to quit)\n")

    while True:
        question = input(">> ").strip()

        if question.lower() == "exit":
            break

        if not question:
            continue

        answer, docs = ask_question(question)

        print("\nAnswer:\n", answer)

        # Optional: show sources (very useful for debugging)
        print("\nSources:")
        for doc in docs:
            print("-", doc.metadata.get("book", "Unknown"))


if __name__ == "__main__":
    run()