Delete client_example.py

client_example.py

@@ -1,177 +0,0 @@
-"""
-Example client script for Invoice Information Extractor API
-Shows how to integrate the API into your application
-"""
-
-import requests
-from pathlib import Path
-import json
-from typing import List, Dict
-
-
-class InvoiceExtractorClient:
-    """Client for Invoice Information Extractor API"""
-    
-    def __init__(self, base_url: str = "http://localhost:7860"):
-        """
-        Initialize client
-        
-        Args:
-            base_url: Base URL of the API (default: http://localhost:7860)
-        """
-        self.base_url = base_url.rstrip('/')
-        self.session = requests.Session()
-    
-    def health_check(self) -> Dict:
-        """Check API health status"""
-        response = self.session.get(f"{self.base_url}/health")
-        response.raise_for_status()
-        return response.json()
-    
-    def extract_invoice(self, image_path: str, doc_id: str = None) -> Dict:
-        """
-        Extract information from a single invoice
-        
-        Args:
-            image_path: Path to invoice image
-            doc_id: Optional document identifier
-        
-        Returns:
-            Extraction results as dictionary
-        """
-        with open(image_path, 'rb') as f:
-            files = {'file': f}
-            data = {'doc_id': doc_id} if doc_id else {}
-            
-            response = self.session.post(
-                f"{self.base_url}/extract",
-                files=files,
-                data=data,
-                timeout=60
-            )
-            response.raise_for_status()
-            return response.json()
-    
-    def extract_batch(self, image_paths: List[str]) -> List[Dict]:
-        """
-        Extract information from multiple invoices
-        
-        Args:
-            image_paths: List of paths to invoice images
-        
-        Returns:
-            List of extraction results
-        """
-        files = [('files', open(path, 'rb')) for path in image_paths]
-        
-        try:
-            response = self.session.post(
-                f"{self.base_url}/extract_batch",
-                files=files,
-                timeout=120
-            )
-            response.raise_for_status()
-            return response.json()['results']
-        finally:
-            # Close all file handles
-            for _, file_handle in files:
-                file_handle.close()
-
-
-# Example usage
-if __name__ == "__main__":
-    # Initialize client
-    client = InvoiceExtractorClient("http://localhost:7860")
-    
-    # Check health
-    print("Checking API health...")
-    health = client.health_check()
-    print(f"Status: {health['status']}")
-    print(f"Models loaded: {health['models_loaded']}\n")
-    
-    # Example 1: Single invoice extraction
-    print("=" * 60)
-    print("Example 1: Single Invoice Extraction")
-    print("=" * 60)
-    
-    # Replace with your invoice path
-    invoice_path = "sample_invoice.png"
-    
-    if Path(invoice_path).exists():
-        try:
-            result = client.extract_invoice(invoice_path, doc_id="demo_001")
-            
-            print(f"\n📄 Document ID: {result['doc_id']}")
-            print(f"✅ Confidence: {result['confidence']}")
-            print(f"⏱️  Processing Time: {result['processing_time_sec']}s")
-            print(f"💰 Cost Estimate: ${result['cost_estimate_usd']}")
-            
-            print("\n📋 Extracted Fields:")
-            fields = result['fields']
-            print(f"  Dealer Name: {fields['dealer_name']}")
-            print(f"  Model Name: {fields['model_name']}")
-            print(f"  Horse Power: {fields['horse_power']} HP")
-            print(f"  Asset Cost: ₹{fields['asset_cost']:,}")
-            print(f"  Signature: {'✓ Detected' if fields['signature']['present'] else '✗ Not found'}")
-            print(f"  Stamp: {'✓ Detected' if fields['stamp']['present'] else '✗ Not found'}")
-            
-            if result.get('warnings'):
-                print(f"\n⚠️  Warnings: {', '.join(result['warnings'])}")
-            
-        except requests.exceptions.RequestException as e:
-            print(f"❌ Error: {e}")
-    else:
-        print(f"⚠️  Sample invoice not found at: {invoice_path}")
-        print("   Please provide a valid invoice image path.")
-    
-    # Example 2: Batch processing
-    print("\n" + "=" * 60)
-    print("Example 2: Batch Invoice Processing")
-    print("=" * 60)
-    
-    # Replace with your invoice paths
-    batch_paths = ["invoice_001.png", "invoice_002.png"]
-    
-    existing_paths = [p for p in batch_paths if Path(p).exists()]
-    
-    if existing_paths:
-        try:
-            results = client.extract_batch(existing_paths)
-            
-            print(f"\n📦 Processed {len(results)} invoices")
-            
-            for i, result in enumerate(results, 1):
-                if 'error' in result:
-                    print(f"\n  {i}. ❌ {result.get('filename', 'Unknown')}: {result['error']}")
-                else:
-                    print(f"\n  {i}. ✅ {result['doc_id']}")
-                    print(f"     Confidence: {result['confidence']}")
-                    print(f"     Dealer: {result['fields']['dealer_name']}")
-                    print(f"     Cost: ₹{result['fields']['asset_cost']:,}")
-            
-        except requests.exceptions.RequestException as e:
-            print(f"❌ Error: {e}")
-    else:
-        print("⚠️  No valid invoice images found for batch processing")
-    
-    # Example 3: Save results to JSON
-    print("\n" + "=" * 60)
-    print("Example 3: Save Results to JSON")
-    print("=" * 60)
-    
-    if Path(invoice_path).exists():
-        try:
-            result = client.extract_invoice(invoice_path)
-            
-            output_file = "extraction_result.json"
-            with open(output_file, 'w', encoding='utf-8') as f:
-                json.dump(result, f, indent=2, ensure_ascii=False)
-            
-            print(f"\n✅ Results saved to: {output_file}")
-            
-        except Exception as e:
-            print(f"❌ Error: {e}")
-    
-    print("\n" + "=" * 60)
-    print("Examples complete!")
-    print("=" * 60)