Update app.py

app.py

@@ -8,65 +8,58 @@ app = Flask(__name__)
 def thismain():
     print('ayhaga')
     return jsonify('done')
-    
+
+pdf_content = None
+pageNumTextFound = 0
+BASE_URL = "hhttps://huggingface.co/spaces/Marthee/NBSLink"  # Localhost URL for testing
+
+@app.route('/view-pdf', methods=['GET'])
+def download_pdf():
+    global pdf_content, pageNumTextFound
+
+    if pdf_content is None:
+        return "PDF content not found.", 404
+
+    pdf_bytes = BytesIO(pdf_content)
+    return send_file(
+        pdf_bytes,
+        mimetype='application/pdf',
+        as_attachment=False,
+        download_name=f"annotated_page_{pageNumTextFound}.pdf"
+    )
+
 @app.route('/api/process-data', methods=['POST'])
-def process():
-    # check_api_key()
-    try:
-        
-        print('In process [Try]')
-        data = request.form
-        # Extracting values
-        filePath = data.get('filePath')
-        groupName = data.get('groupName')
+def receive_pdf_data():
+    global pdf_content, pageNumTextFound
 
-        pdftext = pdftotext.texts_from_pdf(filePath,groupName)
-        return jsonify(pdftext)
-    
-    except Exception as e:
-        print(f"Error: {e}")
-        return jsonify({"error": str(e)}), 500
+    # Get PDF link and keyword from the request body
+    pdf_link = request.form.get('pdf_link')
+    keyword = request.form.get('keyword')
 
+    if not pdf_link or not keyword:
+        return jsonify({"error": "Both 'pdf_link' and 'keyword' must be provided."}), 400
 
-@app.route("/apiMC",methods=["GET", "POST"])
-def apifunction():
     try:
-        print('In process [Try]')
-        # data = request.form
-        data = request.get_json()
-        print('here',data)
-        # Extracting values
-        if data:
-            # Pass the raw JSON data to the filtering function
-            jsonOutput = pdftotext.apiFiltering(data)
-            return jsonify(jsonOutput)
+        # Call the function to process the PDF
+        keyword = json.loads(keyword)
 
+        global pdf_content
+        pdf_content, pageNumTextFound,highlight_rect  = Find_Hyperlinking_text.annotate_text_from_pdf([pdf_link], keyword)
+
+        if pdf_content is None:
+            return jsonify({"error": "No valid PDF content found."}), 404
+
+        # Construct the URL in the desired format with the rectangle coordinates
+        download_link = f"{BASE_URL}/view-pdf#page={pageNumTextFound}&zoom={highlight_rect}"
+
+        return jsonify({
+            "message": "PDF processed successfully.",
+            "download_link": download_link  # Return the formatted URL
+        })
 
-    
     except Exception as e:
-        print(f"Error: {e}")
         return jsonify({"error": str(e)}), 500
     
     
 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)