Update app.py

app.py

@@ -1,51 +1,61 @@
-from flask import Flask, request, jsonify, abort
-import tsadropboxretrieval
-import pdftotext
+from flask import Flask, send_from_directory, jsonify, redirect, url_for
+import os
 import requests
+from io import BytesIO
+
 app = Flask(__name__)
-# API_KEY = "adrpdftotext"  # Replace with your actual API key
 
-# def check_api_key():
-#     api_key = request.headers.get("x-api-key")
-#     if api_key != API_KEY:
-#         abort(403)  # Forbidden if API key is missing or incorrect
-@app.route("/",methods=["GET", "POST"])
-def thismain():
-    print('ayhaga')
-    return jsonify('done')
-    
-@app.route('/api/process-data', methods=['POST'])
-def process():
-    # check_api_key()
-    try:
-        
-        print('In process [Try]')
-        data = request.form
-        # data = json.loads(request.data)
-        print('here',data)
-        # data = request.get_json()  # Use get_json() to parse JSON data
-        # print('Received data:', data)
-        # Extracting values
-        filePath = data.get('filePath')
-        groupName = data.get('groupName')
+# Directory for storing temporary PDFs
+PDF_DIRECTORY = './temp_pdfs'
+TEMP_PDF_NAME = 'temp_downloaded.pdf'
 
+# Ensure the directory exists
+os.makedirs(PDF_DIRECTORY, exist_ok=True)
 
-        # dbxTeam = tsadropboxretrieval.ADR_Access_DropboxTeam('user')
-        # print('3')
-        # md, res = dbxTeam.files_download(path=filePath)
-        # pdf_data = res.content
-        print('4')
-        
+@app.route('/download-pdf', methods=['GET'])
+def download_pdf():
+    """
+    Download PDF from a Dropbox link and save it temporarily.
+    """
+    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'
+    
+    # Modify Dropbox link for direct download
+    if 'dl=0' in link:
+        link = link.replace('dl=0', 'dl=1')
 
-        pdftext = pdftotext.texts_from_pdf(filePath,groupName)
-        print('5')
+    try:
+        response = requests.get(link)
+        response.raise_for_status()  # Raise exception for HTTP errors
+        pdf_content = BytesIO(response.content)  # Store PDF in memory
+        local_pdf_path = os.path.join(PDF_DIRECTORY, TEMP_PDF_NAME)
 
-        return jsonify(pdftext)
-    
+        # Save the PDF to the temporary directory
+        with open(local_pdf_path, 'wb') as f:
+            f.write(pdf_content.getvalue())
+        
+        return jsonify({"message": "PDF downloaded successfully."})
     except Exception as e:
-        print(f"Error: {e}")
         return jsonify({"error": str(e)}), 500
 
+@app.route('/view-pdf', methods=['GET'])
+def view_pdf():
+    """
+    Serve the PDF for viewing in an iframe or direct link.
+    """
+    local_pdf_path = os.path.join(PDF_DIRECTORY, TEMP_PDF_NAME)
+    if os.path.exists(local_pdf_path):
+        return send_from_directory(PDF_DIRECTORY, TEMP_PDF_NAME)
+    else:
+        return jsonify({"message": "PDF not found. Please download it first via /download-pdf"}), 404
+
+@app.route('/pdf-page/<int:page>', methods=['GET'])
+def pdf_page(page):
+    """
+    Redirect to the PDF page with navigation.
+    """
+    return redirect(url_for('view_pdf') + f"#page={page}")
+
+
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)
 # from flask import Flask, request, jsonify