Update app.py

app.py

@@ -189,6 +189,10 @@ def query_embeddings(query_embedding, n_results=5):
         print(f"Error in query_embeddings: {e}")
         return []
 
+query_embedding = embed_query_text(query_text)  # Embed the query text
+initial_results = query_embeddings(query_embedding, embeddings_data, n_results=5)
+document_ids = [doc_id for doc_id, _ in initial_results]
+
 def retrieve_document_text(doc_id):
     """Retrieve document text from HTML file"""
     try:
@@ -204,6 +208,8 @@ def retrieve_document_text(doc_id):
         print(f"Error retrieving document {doc_id}: {e}")
         return ""
 
+document_texts = retrieve_document_texts(document_ids, folder_path)
+
 def rerank_documents(query, doc_texts):
     """Rerank documents using cross-encoder"""
     try: