import uuid
from app.session import SessionManager
from app.retrieval import retrieval_and_reranking
from app.ingestion import extract

manager = SessionManager()
filename = "d2l-en.pdf"
path = f"corpus/{filename}"
doc_bytes = open(path, "rb").read()
chunks, _, _ = extract(path)

session = manager.create_session(filename, doc_bytes, chunks)
query, top_chunks, _ = retrieval_and_reranking(session, "What's the difference between LSTM and GRU gating mechanisms?")
print(query)
print("--"*20)
print(top_chunks)