Update app.py

app.py

@@ -44,6 +44,12 @@ def home():
         # Redirect to your PDF viewer route when ready
         return redirect(url_for("view_pdf", **request.args))
 PROCESSED_PDF_PATH = "static/processed.pdf"
+from flask import Flask, request, send_file
+from io import BytesIO
+import requests
+from urllib.parse import unquote
+
+app = Flask(__name__)
 
 def getpdfcontent(pdf_path):
     # Handle Dropbox URLs
@@ -54,39 +60,39 @@ def getpdfcontent(pdf_path):
     response = requests.get(pdf_path)
     pdf_bytes = response.content
 
-    if not pdf_bytes:
+    if not pdf_bytes or not pdf_bytes.startswith(b"%PDF"):
         raise ValueError("No valid PDF content found.")
 
-    # Convert to BytesIO only after validation
+    # Return a BytesIO stream
     return BytesIO(pdf_bytes)
 
 @app.route('/view-pdf', methods=['GET'])
 def view_pdf():
-    # supports /view-pdf?pdfLink=<encoded link>
     encoded_pdf_link = request.args.get('pdfLink')
     if not encoded_pdf_link:
         return "Missing pdfLink parameter.", 400
 
     pdf_link = unquote(encoded_pdf_link)
     print("Extracted PDF Link:", pdf_link)
-    
+
     try:
-        pdf_content=getpdfcontent(pdf_link)
-        # pdf_content = InitialMarkups.extract_section_under_header(pdf_link)[0]
+        pdf_content = getpdfcontent(pdf_link)
     except Exception as e:
         print("Error during PDF extraction:", e)
         return "PDF could not be processed.", 500
 
-    if pdf_content is None :
+    if pdf_content is None:
         return "PDF content not found or broken.", 404
 
+    # ✅ Do NOT wrap again in BytesIO
     return send_file(
-        BytesIO(pdf_content),
+        pdf_content,
         mimetype='application/pdf',
         as_attachment=False,
-        download_name=f"annotated_page_{pageNumTextFound}.pdf"
+        download_name="annotated_page.pdf"
     )
 
+
 @app.route('/get-pdf')
 def get_pdf():
     """Serve the processed PDF."""