Update app.py

app.py

@@ -448,8 +448,13 @@ def get_response_from_pdf(query, model, selected_docs, num_calls=3, temperature=
 
     retriever = database.as_retriever()
     logging.info(f"Retrieving relevant documents for query: {query}")
-    relevant_docs = retriever.get_relevant_documents(query)
-    logging.info(f"Number of relevant documents retrieved: {len(relevant_docs)}")
+    try:
+        relevant_docs = retriever.invoke(query)  # Use invoke instead of get_relevant_documents
+        logging.info(f"Number of relevant documents retrieved: {len(relevant_docs)}")
+    except Exception as e:
+        logging.error(f"Error retrieving relevant documents: {str(e)}")
+        yield f"An error occurred while retrieving relevant documents: {str(e)}"
+        return
     
     # Filter relevant_docs based on selected documents
     filtered_docs = [doc for doc in relevant_docs if doc.metadata["source"] in selected_docs]