Update app.py

app.py

@@ -9,46 +9,28 @@ def getInfotoMeasure():
     return render_template("gui.html")
 # Route to serve the PDF in iframe
 
-@app.route('/view-pdf')
-def view_pdf():
+@app.route('/pdf', methods=['GET'])
+def serve_pdf():
+    # Dropbox link
     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'
 
+    # Convert Dropbox link for direct download
     if 'dl=0' in link:
         link = link.replace('dl=0', 'dl=1')
 
+    # Download the PDF content
     response = requests.get(link)
+    pdf_content = BytesIO(response.content)
 
-    if response.status_code == 200:
-        pdf_path = 'temp_view.pdf'
-        with open(pdf_path, 'wb') as f:
-            f.write(response.content)
-
-        # Serve PDF with CSP header
-        response = send_file(pdf_path, mimetype='application/pdf')
-        response.headers['Content-Security-Policy'] = "frame-src *;"
-        return response
-    else:
-        return "Failed to download PDF content", 404
+    # Serve the PDF
+    return send_file(
+        pdf_content,
+        download_name="document.pdf",  # Name the file
+        mimetype="application/pdf"
+    )
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)
-# from flask import Flask, request, jsonify
-
-# app = Flask(__name__)
-
-# @app.route('/api/process-data', methods=['POST'])
-# def process_data():
-#     # Get JSON data from the request
-#     data = request.get_json()
-
-#     # Perform some processing on the data
-#     # For example, let's just return the data back with a message
-#     processed_data = {
-#         'message': 'Data received and processed successfully!',
-#         'received_data': data
-#     } 
-
-#     return jsonify(processed_data)
 
 # if __name__ == '__main__':
 #     app.run(host='0.0.0.0', port=7860, debug=True)