added page range slider by modifying get_pdf_content and app.py

app.py

@@ -222,6 +222,12 @@ with st.sidebar:
                 value='2024-05-01-preview',
             )
 
+        page_range_slider = st.slider(label=('4: Specify a page range to examine:\n\n'
+                                            '(min=1, max=50)'),
+                                    min_value=1, max_value=50,
+                                    value=(1, 50))
+        st.session_state['page_range'] = page_range_slider
+
 
 def build_ui():
     """
@@ -284,7 +290,10 @@ def set_up_chat_ui():
         if prompt['files']:
             # 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
-            st.session_state[ADDITIONAL_INFO] = filem.get_pdf_contents(prompt['files'][0])
+            page_range = st.session_state.get('page_range', (1, 50))  # fallback default
+            st.session_state[ADDITIONAL_INFO] = filem.get_pdf_contents(
+                prompt['files'][0], page_range
+            )
             print(f'{prompt["files"]=}')
 
         provider, llm_name = llm_helper.get_provider_model(

helpers/file_manager.py

@@ -19,6 +19,7 @@ logger = logging.getLogger(__name__)
 
 def get_pdf_contents(
         pdf_file: st.runtime.uploaded_file_manager.UploadedFile,
+        page_range: tuple[int, int],
         max_pages: int = GlobalConfig.MAX_PAGE_COUNT
 ) -> str:
     """
@@ -30,11 +31,17 @@ def get_pdf_contents(
     """
 
     reader = PdfReader(pdf_file)
-    n_pages = min(max_pages, len(reader.pages))
-    text = ''
 
-    for page in range(n_pages):
-        page = reader.pages[page]
+    total_pages = len(reader.pages)
+    n_pages = min(max_pages, total_pages)
+
+    start, end = page_range
+    start = max(1, start)
+    end = min(n_pages, end)
+
+    text = ''
+    for page_num in range(start - 1, end):
+        page = reader.pages[page_num]
         text += page.extract_text()
 
     return text