Update app.py

app.py

@@ -99,7 +99,7 @@ class SemanticSearch:
 #    return 'Corpus Loaded.'
 
 # The modified function generates embeddings based on PDF file name and page number and checks if the embeddings file exists before loading or generating it.	
-def load_recommender(path, start_page=1):
+def load_recommender(path, start_page=1, end_page=None):
     global recommender
     pdf_file = os.path.basename(path)
     embeddings_file = f"{pdf_file}_{start_page}.npy"
@@ -110,7 +110,7 @@ def load_recommender(path, start_page=1):
         recommender.fitted = True
         return "Embeddings loaded from file"
     
-    texts = pdf_to_text(path, start_page=start_page)
+    texts = pdf_to_text(path, start_page=start_page, end_page=end_page)
     chunks = text_to_chunks(texts, start_page=start_page)
     recommender.fit(chunks)
     np.save(embeddings_file, recommender.embeddings)