import requests

API_KEY = "YOUR_API_KEY"
API_BASE = "https://api.backgroundcut.co"
HEADERS = {"Authorization": f"Bearer {API_KEY}"}

# Step 1 — Initialize session (multipart/form-data)
# Uploads the image and prepares it for interactive segmentation.
with open("photo.jpg", "rb") as f:
    init_response = requests.post(
        f"{API_BASE}/v3/ai-cut/initialize/",
        headers=HEADERS,
        files={"image": f},
        timeout=300
    )

init_response.raise_for_status()
unique_filename = init_response.json()["unique_filename"]
print("Session:", unique_filename)

# Step 2 — Segment with points (application/json, FREE, unlimited)
# Green dots mark areas to include. Red dots mark areas to exclude.
# Always send ALL points accumulated so far, not just new ones.
segment_response = requests.post(
    f"{API_BASE}/v3/ai-cut/segment/",
    headers={**HEADERS, "Content-Type": "application/json"},
    json={
        "unique_filename": unique_filename,
        "greenDots": [{"x": 250, "y": 300}],
        "redDots": [],
        "request_number": 1
    },
    timeout=300
)

segment_response.raise_for_status()
result = segment_response.json()
print("Segmented:", result["output_url"])

# Refine — add more points (still FREE, still unlimited)
# Include ALL previous points plus new ones.
refine_response = requests.post(
    f"{API_BASE}/v3/ai-cut/segment/",
    headers={**HEADERS, "Content-Type": "application/json"},
    json={
        "unique_filename": unique_filename,
        "greenDots": [{"x": 250, "y": 300}, {"x": 270, "y": 320}],
        "redDots": [{"x": 50, "y": 50}],
        "request_number": 2       # increment with each call
    },
    timeout=300
)

refine_response.raise_for_status()
refined = refine_response.json()
print("Refined:", refined["output_url"])