app/routes/panApi.py

# import re
# from flask import request, jsonify, current_app
# from PIL import Image, UnidentifiedImageError
# from io import BytesIO
# import base64
# import requests
# from  ..services.panServices.panOcr import process_results
# import io
# def ocrPan(mode, session):
#     try:
#         print("API HIT ************* PANOCR")
#         data = request.get_json()
#         if not data:
#             return jsonify({"error": "Invalid request payload"}), 400
        
#         if mode == "prod":
#             if not data.get('image'):
#                 return jsonify({"error": "Image data/buffer is required"}), 400
            
#             #removing 'data:image/png,base64 ' from buffer.
#             imgBuffer = data.get('image')
#             imgBuffer = re.sub("^data:image/.+;base64,", "", imgBuffer)
#             # Adjust base64 string padding
#             if len(imgBuffer) % 4:
#                 imgBuffer += '=' * (4 - len(imgBuffer) % 4)
                
#             try:
#                 img_data = base64.b64decode(imgBuffer)
#                 img = Image.open(BytesIO(img_data))
#                 img.verify()  # Verify image format
#                 print(img, "img")
#                 img = Image.open(io.BytesIO(img_data))  # Re-open image after verification
#             except (base64.binascii.Error, ValueError) as decode_err:
#                 return jsonify({"error": f"Image decoding failed: {str(decode_err)}"}), 400
#             except UnidentifiedImageError:
#                 return jsonify({"error": "Unable to identify image format."}), 400

#         elif mode == "dev":
#             if not data.get('imgUrl'):
#                 return jsonify({"error": "Image URL is required"}), 400
            
#             img_url = data.get('imgUrl')
#             response = session.get(img_url)
#             response.raise_for_status()
#             img = Image.open(BytesIO(response.content))
#             print(img, "img")
#             img.verify()  # Verify image format
#             img = Image.open(BytesIO(response.content))  # Re-open image after verification

#         else:
#             return jsonify({"error": "Invalid mode configuration"}), 500

#         # Check image format
#         if img.format not in ['JPEG', 'JPG', 'PNG']:
#             return jsonify({"error": "Invalid image format. Only JPG and PNG are supported."}), 400
        
#         # Run detection
#         model = current_app.models.get('panModel')
#         results = model.predict(source=img, imgsz=680, iou=0.7, augment=True)
#         # print(results,"model result")
#         extracted_data = process_results(results, img)
#         # print(extracted_data, "extracted data")
        
#         if extracted_data.get('statusCode') == 400:
#             return jsonify(extracted_data), 400
        
#         return jsonify(extracted_data), 200
#     except requests.RequestException as req_err:
#         return jsonify({"error": f"Image download failed: {str(req_err)}"}), 500
#     except UnidentifiedImageError:
#         return jsonify({"error": "Unable to identify image format."}), 400
#     except Exception as e:
#         current_app.logger.error(f"Unexpected error: {str(e)}")
#         return jsonify({"error": "An unexpected error occurred on api call."}), 500