Enrich a lead list

import os
import time
import requests

API_KEY = os.environ["SEAMLESS_API_KEY"]
BASE = "https://api.seamless.ai/api/client/v1"
HEADERS = {"Content-Type": "application/json", "Token": API_KEY}

def search_contacts():
    response = requests.post(
        f"{BASE}/search/contacts",
        headers=HEADERS,
        json={
            "jobTitle": ["Chief Technology Officer"],
            "industry": ["Computer Software"],
            "limit": 10,
        },
    )
    response.raise_for_status()
    return response.json()["data"]

def research_contacts(search_result_ids):
    response = requests.post(
        f"{BASE}/contacts/research",
        headers=HEADERS,
        json={"searchResultIds": search_result_ids},
    )
    response.raise_for_status()
    return response.json()["requestIds"]

def poll_until_done(request_ids):
    pending = set(request_ids)
    results = []
    while pending:
        response = requests.get(
            f"{BASE}/contacts/research/poll",
            headers={"Token": API_KEY},
            params={"requestIds": ",".join(pending)},
        )
        response.raise_for_status()
        for row in response.json().get("data", []):
            status = row.get("status")
            if status in ("done", "error", "missing", "duplicate"):
                pending.discard(row["requestId"])
                results.append(row)
        if pending:
            time.sleep(3)
    return results

if __name__ == "__main__":
    matches = search_contacts()
    ids = [row["searchResultId"] for row in matches[:5]]
    print(f"Researching {len(ids)} contacts...")
    request_ids = research_contacts(ids)
    completed = poll_until_done(request_ids)
    for row in completed:
        contact = row.get("contact") or {}
        print(contact.get("fullName"), contact.get("email"), row["status"])