Classification

from unsiloed_sdk import UnsiloedClient

# Define categories with optional descriptions
categories = [
    {"name": "Medical Record", "description": "Patient medical records and history"},
    {"name": "Lab Report", "description": "Laboratory test results"},
    {"name": "Prescription"}  # Description is optional
]

with UnsiloedClient(api_key="your-api-key") as client:
    # Classify and wait for completion
    result = client.classify_and_wait(
        file="document.pdf",
        categories=categories
    )

    print(f"Classification: {result.result['classification']}")
    print(f"Confidence: {result.result['confidence']:.2%}")

    # Check page-by-page results if available
    if 'page_results' in result.result:
        for page_result in result.result['page_results']:
            print(f"Page {page_result['page']}: {page_result['classification']}")

from unsiloed_sdk import UnsiloedClient

def check_classification_status(job_id: str, api_key: str):
    with UnsiloedClient(api_key=api_key) as client:
        # Get classification result
        job = client.get_classify_result(job_id)

        print(f"Status: {job.status}")
        print(f"Progress: {job.progress}")

        if job.status == "completed" and job.result:
            result = job.result
            print(f"\nClassification: {result['classification']}")
            print(f"Confidence: {result['confidence']:.2%}")
            print(f"Total Pages: {result.get('total_pages', 'N/A')}")

            if 'page_results' in result:
                for page_result in result['page_results']:
                    print(f"Page {page_result['page']}: {page_result['classification']}")
        
        return job

from unsiloed_sdk import UnsiloedClient

categories = [
    {"name": "Invoice"},
    {"name": "Receipt"},
    {"name": "Contract"}
]

with UnsiloedClient(api_key="your-api-key") as client:
    # Use classify_and_wait for automatic polling
    result = client.classify_and_wait(
        file="document.pdf",
        categories=categories,
        poll_interval=5,  # Check every 5 seconds
    )

    print(f"Classification: {result.result['classification']}")
    print(f"Confidence: {result.result['confidence']:.2%}")