Patient, document, note

import os
import time

import requests

BASE = "https://api.healos.ai/ext-api/v1"
http = requests.Session()
http.headers["X-API-Key"] = os.environ["HEALOS_API_KEY"]


def check(res, allowed=()):
    if not res.ok and res.status_code not in allowed:
        raise RuntimeError(f"{res.request.method} {res.url} -> {res.status_code}: {res.json().get('error')}")
    return res.json()


# 1. Create the patient (a 409 means it already exists: reuse it).
res = http.post(f"{BASE}/patients", json={
    "first_name": "Testy",
    "last_name": "Recipe",
    "date_of_birth": "1970-01-01",
    "external_id": "test-pt-001",
}, timeout=30)
body = check(res, allowed=(409,))
patient_id = body["existing"]["id"] if res.status_code == 409 else body["data"]["id"]

# 2. Upload the document (a 409 means this external_id was already uploaded).
with open("test-intake-form.pdf", "rb") as f:
    res = http.post(
        f"{BASE}/patients/{patient_id}/documents",
        files={"file": ("test-intake-form.pdf", f, "application/pdf")},
        data={"external_id": "test-doc-001"},
        timeout=60,
    )
check(res, allowed=(409,))
if res.status_code == 201 and "text extraction started" not in res.json().get("message", ""):
    document_id = res.json()["data"]["id"]
    check(http.post(f"{BASE}/patients/{patient_id}/documents/{document_id}/extract", timeout=60))

# 3. Wait for text extraction.
for _ in range(60):
    docs = check(http.get(
        f"{BASE}/patients/{patient_id}/documents",
        params={"external_id": "test-doc-001"},
        timeout=30,
    ))["data"]
    if docs[0]["status"] == "failed":
        raise RuntimeError(f"Extraction failed: {docs[0].get('extraction_error')}")
    if docs[0]["status"] == "completed":
        break
    time.sleep(5)
else:
    raise RuntimeError("Extraction is still running; check the document later")

# 4. Generate the note. Sent once: each call creates a new session and note.
queued = check(http.post(
    f"{BASE}/patients/{patient_id}/notes/from-documents",
    json={"model": "balanced"},
    timeout=60,
))
session_id = queued["data"]["session_id"]

# 5. Read the note.
notes = []
for _ in range(30):
    time.sleep(10)
    notes = check(http.get(f"{BASE}/sessions/{session_id}/notes", timeout=30))["data"]["notes"]
    if notes:
        break
print(notes[0]["content"] if notes else f"Note not ready yet; check session {session_id} later")