Update app.py

app.py

@@ -1,40 +1,42 @@
-from flask import Flask, send_file, render_template, request, session, jsonify
+from flask import Flask, send_file, render_template, request
 import requests
 from io import BytesIO
 import fitz  # PyMuPDF
 
-# Flask App
+# Define local variables to retain the PDF content across function calls
+pdf_content = None
+pageNumTextFound = 0
+
 app = Flask(__name__)
-app.secret_key = 'your_secret_key'  # Flask session requires a secret key
 
 @app.route("/", methods=["GET", "POST"])
 def getInfotoMeasure():
-    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'
-    ]
+    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 to process the PDF
-    session['pdf_content'], session['pageNumTextFound'] = highlight_text_from_pdf(pdf_link, keyword)
+    pdf_content, pageNumTextFound = highlight_text_from_pdf(pdf_link, keyword)
 
     # Render the GUI with the current page number
-    return render_template("gui.html", page=session.get('pageNumTextFound', 0))
-
+    return render_template("gui.html", page=pageNumTextFound)
 
 @app.route('/view-pdf', methods=['GET'])
 def download_pdf():
-    if 'pdf_content' not in session or session['pdf_content'] is None:
+    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.get('pageNumTextFound', 0)}.pdf"
+        download_name=f"highlighted_page_{pageNumTextFound}.pdf"
     )
 
-
 def highlight_text_from_pdf(pdfshareablelinks, keyword):
     print('PDF Links:', pdfshareablelinks)
 
@@ -63,16 +65,15 @@ def highlight_text_from_pdf(pdfshareablelinks, keyword):
             if matched:
                 for word in matched:
                     page.add_highlight_annot(word)
-
+                
                 pageNumTextFound = page_num + 1
 
+        # Save PDF content to memory and return it along with the page number
         pdf_bytes = BytesIO()
         pdf_document.save(pdf_bytes)
         pdf_document.close()
 
-        # Save PDF content and page number to Flask session
         return pdf_bytes.getvalue(), pageNumTextFound
 
-
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)