Update app.py

app.py

@@ -1,39 +1,45 @@
-from flask import Flask, send_file, render_template, request, session
+from flask import Flask, send_file, render_template, request
 import requests
 from io import BytesIO
 import fitz  # PyMuPDF
 
+# Define local variables to retain the PDF content across function calls
+pdf_content = None
+pageNumTextFound = 0
+
 app = Flask(__name__)
 
-# Store the result in session to reuse it across routes
 @app.route("/", methods=["GET", "POST"])
 def getInfotoMeasure():
-    if 'pageNumTextFound' not in session:
-        pdf_link = ['https://www.dropbox.com/scl/fi/fjykwhhn9gu9t3kqrflxd/LA002-NOR-ZZ-ZZ-T-A-2403_Architectural-Specification-F10-Brick-and-Block-Walling_A4-_C01.pdf?rlkey=ek9i66i79m0hwp8z5yjs6rp5p&st=jh05a6qs&dl=0']
-        keyword = "To be read with preliminaries/ general conditions"
+    global pdf_content, pageNumTextFound
+
+    pdf_link = ['https://www.dropbox.com/scl/fi/fjykwhhn9gu9t3kqrflxd/LA002-NOR-ZZ-ZZ-T-A-2403_Architectural-Specification-F10-Brick-and-Block-Walling_A4-_C01.pdf?rlkey=ek9i66i79m0hwp8z5yjs6rp5p&st=jh05a6qs&dl=0']
+    keyword = "To be read with preliminaries/ general conditions"
 
-        # Call the function once and store the result in session
-        pdf_content, pageNumTextFound = highlight_text_from_pdf(pdf_link, keyword)
-        session['pageNumTextFound'] = pageNumTextFound
-        session['pdf_content'] = pdf_content.getvalue()
+    # Call the function to process the PDF
+    pdf_content, pageNumTextFound = highlight_text_from_pdf(pdf_link, keyword)
 
-    return render_template("gui.html", page=session['pageNumTextFound'])
+    # Render the GUI with the current page number
+    return render_template("gui.html", page=pageNumTextFound)
 
 @app.route('/view-pdf', methods=['GET'])
 def download_pdf():
-    if 'pdf_content' not in session:
-        return "PDF content not found in session.", 404
+    global pdf_content, pageNumTextFound
+
+    if pdf_content is None:
+        return "PDF content not found.", 404
 
-    pdf_bytes = BytesIO(session['pdf_content'])
+    pdf_bytes = BytesIO(pdf_content)
     return send_file(
         pdf_bytes,
         mimetype='application/pdf',
         as_attachment=False,
-        download_name=f"highlighted_page_{session['pageNumTextFound']}.pdf"
+        download_name=f"highlighted_page_{pageNumTextFound}.pdf"
     )
 
 def highlight_text_from_pdf(pdfshareablelinks, keyword):
     print('PDF Links:', pdfshareablelinks)
+
     for link in pdfshareablelinks:
         pdf_content = None
 
@@ -42,6 +48,7 @@ def highlight_text_from_pdf(pdfshareablelinks, keyword):
                 link = link.replace('dl=0', 'dl=1')
 
             response = requests.get(link)
+
             if response.status_code == 200:
                 pdf_content = BytesIO(response.content)
 
@@ -58,9 +65,15 @@ def highlight_text_from_pdf(pdfshareablelinks, keyword):
             if matched:
                 for word in matched:
                     page.add_highlight_annot(word)
+                
                 pageNumTextFound = page_num + 1
 
-        return pdf_document, pageNumTextFound
+        # Save PDF content to memory and return it along with the page number
+        pdf_bytes = BytesIO()
+        pdf_document.save(pdf_bytes)
+        pdf_document.close()
+
+        return pdf_bytes.getvalue(), pageNumTextFound
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)