01_ocr_extract.py

"""Extract text from a PDF or image using DeepRead OCR."""

import os
import sys
import time
import requests

API_KEY = os.environ.get("DEEPREAD_API_KEY")
BASE = "https://api.deepread.tech"

if not API_KEY:
    print("Set DEEPREAD_API_KEY in your environment or .env file")
    sys.exit(1)

file_path = sys.argv[1] if len(sys.argv) > 1 else None
if not file_path:
    print("Usage: python 01_ocr_extract.py <file.pdf>")
    sys.exit(1)

headers = {"X-API-Key": API_KEY}

# Submit document
with open(file_path, "rb") as f:
    resp = requests.post(
        f"{BASE}/v1/process",
        headers=headers,
        files={"file": f},
        data={"preview": "true"},  # the shareable preview link is opt-in (the link is public)
    )
    resp.raise_for_status()

job = resp.json()
job_id = job["id"]
print(f"Submitted: {job_id} (status: {job['status']})")

# Poll with backoff
delay = 5
while True:
    time.sleep(delay)
    resp = requests.get(f"{BASE}/v1/jobs/{job_id}", headers=headers)
    if resp.status_code in (429, 503):  # rate limited or busy: no status, wait and retry
        delay = float(resp.headers.get("Retry-After", 5))
        continue
    result = resp.json()
    status = result["status"]
    print(f"  Status: {status}")

    if status == "completed":
        content = result["document"]["content"]
        print("\n--- Extracted Text ---")
        print(content.get("text_preview") or content.get("text", "")[:2000])
        artifacts = result.get("artifacts", {})
        if artifacts.get("preview_url"):
            print(f"\nShareable preview: {artifacts['preview_url']}")
        break
    elif status == "failed":
        print(f"Failed: {result.get('error', 'Unknown error')}")
        break

    delay = min(delay * 1.5, 15)