import requests

API_KEY = "sk_YOUR_API_KEY"
HEADERS = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
BASE = "https://api.superdocs.app/v1/chat"

# Your AI agent's workflow
tasks = [
    "Create a project proposal with 4 sections: Overview, Goals, Timeline, Budget",
    "Add specific milestones to the Timeline section",
    "Add cost estimates to the Budget section",
    "Review the entire document and fix any inconsistencies"
]

session_id = "ai-agent-proposal"
document_html = ""

for task in tasks:
    response = requests.post(BASE, headers=HEADERS, json={
        "message": task,
        "session_id": session_id,
        "document_html": document_html
    })

    data = response.json()
    print(f"Task: {task}")
    print(f"AI: {data['response'][:100]}...")

    # Use the updated document for the next task
    if data.get("document_changes"):
        document_html = data["document_changes"]["updated_html"]

print("\nFinal document ready.")

# First call
requests.post(BASE, headers=HEADERS, json={
    "message": "Draft a sales contract",
    "session_id": "agent-contract",
    "document_html": ""
})

# Later call — SuperDocs remembers the contract it drafted
requests.post(BASE, headers=HEADERS, json={
    "message": "Add a payment terms section based on net-30",
    "session_id": "agent-contract",
    "document_html": current_html
})

revert = requests.post(
    f"https://api.superdocs.app/v1/sessions/agent-contract/revert",
    headers=HEADERS,
    json={"turn_index": 4},  # the user message you want to undo
)
result = revert.json()
# result["compose_text"]      → the original message text
# result["document_state"]    → the document HTML at the rewind point
# result["reverted_to_turn"]  → the turn the conversation now ends at