| import os |
| import requests |
| import cv2 |
| import numpy as np |
|
|
| API_URL = os.environ.get( |
| "LIVENESS_API_URL", |
| "http://127.0.0.1:8092/api/check_liveness" |
| ) |
|
|
|
|
| def _load_image(image): |
| if image is None: |
| return None |
| if isinstance(image, str): |
| img = cv2.imread(image) |
| if img is None: |
| return None |
| return img |
| return image |
|
|
|
|
| def check_liveness(image) -> dict: |
| img = _load_image(image) |
| if img is None: |
| return {"error": "No image provided"} |
|
|
| success, img_encoded = cv2.imencode('.jpg', img, [cv2.IMWRITE_JPEG_QUALITY, 95]) |
| if not success: |
| return {"error": "Failed to encode image"} |
|
|
| try: |
| response = requests.post( |
| API_URL, |
| files={"image": ("image.jpg", img_encoded.tobytes(), "image/jpeg")}, |
| timeout=30 |
| ) |
| response.raise_for_status() |
| return response.json() |
| except requests.exceptions.ConnectionError: |
| return {"error": "Cannot connect to liveness detection server. Make sure the SDK server is running."} |
| except requests.exceptions.Timeout: |
| return {"error": "Request timed out"} |
| except requests.exceptions.RequestException as e: |
| return {"error": f"API request failed: {str(e)}"} |
|
|
