ocr api

2025-07-17 16:00:42 +02:00
parent 7860aa3579
commit fed33b7623
9 changed files with 893 additions and 2 deletions
@@ -0,0 +1,109 @@
+#!/usr/bin/env python3
+"""
+Test client for the OCR API
+This script demonstrates how to use the OCR API endpoints
+"""
+
+import requests
+import json
+import sys
+from pathlib import Path
+
+# API base URL
+BASE_URL = "http://localhost:8000"
+
+def test_health():
+    """Test the health endpoint"""
+    try:
+        response = requests.get(f"{BASE_URL}/health")
+        print(f"Health check: {response.status_code} - {response.json()}")
+        return response.status_code == 200
+    except requests.exceptions.ConnectionError:
+        print("❌ Could not connect to the API. Make sure it's running on localhost:8000")
+        return False
+
+def test_pdf_ocr(file_path):
+    """Test PDF OCR endpoint"""
+    if not Path(file_path).exists():
+        print(f"❌ File not found: {file_path}")
+        return False
+
+    try:
+        with open(file_path, 'rb') as f:
+            files = {'file': f}
+            print(f"📄 Processing PDF: {file_path}")
+            response = requests.post(f"{BASE_URL}/ocr/pdf", files=files)
+
+        if response.status_code == 200:
+            result = response.json()
+            print(f"✅ PDF OCR successful!")
+            print(f"   Pages: {result['pages']}")
+            print(f"   Processing time: {result['processing_time']}s")
+            print(f"   Confidence: {result['confidence']}%")
+            print(f"   Text preview: {result['text'][:200]}...")
+            return True
+        else:
+            print(f"❌ PDF OCR failed: {response.status_code} - {response.text}")
+            return False
+
+    except Exception as e:
+        print(f"❌ Error processing PDF: {str(e)}")
+        return False
+
+def test_image_ocr(file_path):
+    """Test image OCR endpoint"""
+    if not Path(file_path).exists():
+        print(f"❌ File not found: {file_path}")
+        return False
+
+    try:
+        with open(file_path, 'rb') as f:
+            files = {'file': f}
+            print(f"🖼️  Processing image: {file_path}")
+            response = requests.post(f"{BASE_URL}/ocr/image", files=files)
+
+        if response.status_code == 200:
+            result = response.json()
+            print(f"✅ Image OCR successful!")
+            print(f"   Processing time: {result['processing_time']}s")
+            print(f"   Confidence: {result['confidence']}%")
+            print(f"   Text: {result['text']}")
+            return True
+        else:
+            print(f"❌ Image OCR failed: {response.status_code} - {response.text}")
+            return False
+
+    except Exception as e:
+        print(f"❌ Error processing image: {str(e)}")
+        return False
+
+def main():
+    """Main test function"""
+    print("🧪 OCR API Test Client")
+    print("=" * 50)
+
+    # Test health endpoint
+    if not test_health():
+        sys.exit(1)
+
+    print("\n" + "=" * 50)
+
+    # Test PDF OCR if file provided
+    if len(sys.argv) > 1:
+        file_path = sys.argv[1]
+        file_ext = Path(file_path).suffix.lower()
+
+        if file_ext == '.pdf':
+            test_pdf_ocr(file_path)
+        elif file_ext in ['.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif']:
+            test_image_ocr(file_path)
+        else:
+            print(f"❌ Unsupported file type: {file_ext}")
+            print("Supported types: .pdf, .png, .jpg, .jpeg, .bmp, .tiff, .tif")
+    else:
+        print("📝 Usage: python test_client.py <file_path>")
+        print("   Example: python test_client.py document.pdf")
+        print("   Example: python test_client.py image.png")
+
+if __name__ == "__main__":
+    main()