Add util function to increment page number +1 (as indexing starts at 0)

app.py

@@ -67,7 +67,7 @@ def get_document_filename(document) -> str:
     Extracts the filename from a document path.
 
     Args:
-        document: Once document retrieved from querying the vector store.
+        document: One document retrieved from querying the vector store.
 
     Returns:
         str: The extracted document filename.
@@ -77,6 +77,21 @@ def get_document_filename(document) -> str:
 
     return document_filename
 
+def get_chunk_page_number(document) -> str:
+    """
+    Retrieves the correct page number from where the chunk was extracted from a certain document. Info: The page number returned in the metadata object starts with page 0, so it has to be increased by 1. 
+
+    Args:
+        document: One document retrieved from querying the vector store.
+
+    Returns:
+        str: The extracted page number as string.
+    """
+    document_path = int(document.metadata["page"])
+    correct_page_number = document_path + 1
+
+    return str(correct_page_number)
+
 def pretty_source_history_md():
     unpretty_history = get_metadata()
     if not unpretty_history:
@@ -106,7 +121,7 @@ def retrieve(query: str):
     document_metadata_to_display_in_ui.append({
         "query": query,
         "filename": [get_document_filename(doc) for doc in retrieved_docs],
-        "number_of_pages": [doc.metadata["page"] for doc in retrieved_docs],
+        "number_of_pages": [get_chunk_page_number(doc) for doc in retrieved_docs],
         "source_text_chunk": [doc.page_content for doc in retrieved_docs]
     })