Update app.py

app.py

@@ -27,6 +27,13 @@ def question_answering(question, pdf_path):
 
     return answer, pdf_text_with_pages
 
+def get_context_text(pdf_text_with_pages, context_page_num, context_window=3):
+    context_start = max(0, context_page_num - context_window - 1)
+    context_end = min(len(pdf_text_with_pages), context_page_num + context_window)
+    context_lines = [text for _, text in pdf_text_with_pages[context_start:context_end]]
+    context_text = "\n".join(context_lines)
+    return context_text
+
 def main():
     st.title("Question Answering on a PDF File")
 
@@ -49,9 +56,9 @@ def main():
 
         # Display context where the answer came from
         context_page_num = answer['start']
-        context_text = pdf_text_with_pages[context_page_num][1]  # Use the page number directly (no need to adjust)
+        context_text = get_context_text(pdf_text_with_pages, context_page_num)
         st.write("Context:")
         st.write(context_text)
 
 if __name__ == "__main__":
-    main()
+    main()