From 8211705debf9d1335223c606275f46c43c78d8a2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 8 Apr 2024 00:23:52 +0200 Subject: Updates --- rag/cli.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 rag/cli.py (limited to 'rag/cli.py') diff --git a/rag/cli.py b/rag/cli.py new file mode 100644 index 0000000..5ea1a47 --- /dev/null +++ b/rag/cli.py @@ -0,0 +1,28 @@ +from pathlib import Path + + +try: + from rag.rag import RAG +except ModuleNotFoundError: + from rag import RAG + +if __name__ == "__main__": + rag = RAG() + + while True: + print("Retrieval Augmented Generation") + choice = input("1. add pdf from path\n2. Enter a query\n") + match choice: + case "1": + path = input("Enter the path to the pdf: ") + path = Path(path) + rag.add_pdf_from_path(path) + case "2": + query = input("Enter your query: ") + if query: + result = rag.retrive(query) + print("Answer: \n") + print(result.answer) + case _: + print("Invalid option!") + -- cgit v1.2.3-70-g09d2