comments

app.py

@@ -291,8 +291,9 @@ def set_up_chat_ui():
     ):
         prompt_text = prompt.text or ''
         if prompt['files']:
+            # store uploaded pdf in session state
             uploaded_pdf = prompt['files'][0]
-            st.session_state["pdf_file"] = uploaded_pdf
+            st.session_state["pdf_file"] = uploaded_pdf  
             # Apparently, Streamlit stores uploaded files in memory and clears on browser close
             # https://docs.streamlit.io/knowledge-base/using-streamlit/where-file-uploader-store-when-deleted
 
@@ -300,7 +301,7 @@ def set_up_chat_ui():
             st.session_state["start_page"], st.session_state["end_page"] = filem.validate_page_range(uploaded_pdf, 
                                                                                                      st.session_state["start_page"],
                                                                                                      st.session_state["end_page"])
-            # update sidebar text
+            # show sidebar text for page selection and file name
             with st.sidebar:
                 st.text(f"Extracting pages {st.session_state["start_page"]} to {st.session_state["end_page"]} in {uploaded_pdf.name}")
 
@@ -315,7 +316,7 @@ def set_up_chat_ui():
                 st.session_state["start_page"], st.session_state["end_page"] = filem.validate_page_range(st.session_state["pdf_file"], 
                                                                                                     st.session_state["start_page"],
                                                                                                     st.session_state["end_page"])
-                # update sidebar text
+                # update sidebar text for name and page selection
                 with st.sidebar:
                     st.text(f"Extracting pages {st.session_state["start_page"]} to {st.session_state["end_page"]} in {st.session_state["pdf_file"].name}")
 

helpers/file_manager.py

@@ -46,13 +46,23 @@ def get_pdf_contents(
 
 def validate_page_range(pdf_file: st.runtime.uploaded_file_manager.UploadedFile,
                         start:int, end:int) -> tuple[int, int]:
-    
+    """
+    Validate the page range.
+
+    :param pdf_file: The uploaded PDF file.
+    :param start: The start page 
+    :param max_pages: The end page
+    :return: The validated page range tuple
+    """
     n_pages = len(PdfReader(pdf_file).pages)
-    #start, end = st.session_state["page_range_slider"]
+
+    # set start to max of 1 or specified start (whichever's higher)
     start = max(1, start)
+
+    # set end to min of pdf length or specified end (whichever's lower)
     end = min(n_pages, end)
 
-    if start >= end:
+    if start >= end:  # if the start is higher than the end, make it 1
         start = 1
 
     return (start, end)