Spaces:
Paused
Paused
File size: 2,969 Bytes
211e423 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
#!/usr/bin/env python3
"""Production API Test Script
Quick test script specifically for the production Dots.OCR API.
"""
import requests
import json
import sys
from pathlib import Path
def test_production_api():
"""Test the production API endpoint."""
api_url = "https://algoryn-dots-ocr-idcard.hf.space"
print(f"🔍 Testing Production API at {api_url}")
# Health check
try:
print("📡 Checking API health...")
health_response = requests.get(f"{api_url}/health", timeout=10)
health_response.raise_for_status()
health_data = health_response.json()
print(f"✅ Health check passed: {health_data}")
except Exception as e:
print(f"❌ Health check failed: {e}")
return False
# Test with front image
front_image = Path(__file__).parent / "tom_id_card_front.jpg"
if not front_image.exists():
print(f"❌ Test image not found: {front_image}")
return False
print(f"📸 Testing OCR with {front_image.name}")
try:
with open(front_image, 'rb') as f:
files = {'file': f}
response = requests.post(
f"{api_url}/v1/id/ocr",
files=files,
timeout=60 # Longer timeout for production
)
response.raise_for_status()
result = response.json()
print(f"✅ OCR test passed")
print(f" Request ID: {result.get('request_id')}")
print(f" Media type: {result.get('media_type')}")
print(f" Processing time: {result.get('processing_time'):.2f}s")
print(f" Detections: {len(result.get('detections', []))}")
# Show extracted fields
for i, detection in enumerate(result.get('detections', [])):
fields = detection.get('extracted_fields', {})
field_count = len([f for f in fields.values() if f is not None])
print(f" Page {i+1}: {field_count} fields extracted")
# Show some key fields
key_fields = ['document_number', 'surname', 'given_names', 'nationality']
for field in key_fields:
if field in fields and fields[field] is not None:
value = fields[field].get('value', 'N/A') if isinstance(fields[field], dict) else str(fields[field])
confidence = fields[field].get('confidence', 'N/A') if isinstance(fields[field], dict) else 'N/A'
print(f" {field}: {value} (confidence: {confidence})")
return True
except Exception as e:
print(f"❌ OCR test failed: {e}")
if hasattr(e, 'response') and e.response is not None:
print(f" Status code: {e.response.status_code}")
print(f" Response: {e.response.text}")
return False
if __name__ == "__main__":
success = test_production_api()
sys.exit(0 if success else 1)
|