import requests

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

INPUT_KEY = "my-project/photo.jpg"  # AI Cut requires JPEG input

# Step 1 — Upload a JPEG image (AI Cut requires JPEG input and output)
with open("photo.jpg", "rb") as f:
    upload_response = requests.post(
        f"{API_BASE}/v3/file-ops/upload/",
        headers=HEADERS,
        files={"file": f},
        data={"key": INPUT_KEY},
        timeout=60
    )

upload_response.raise_for_status()
print("Uploaded:", INPUT_KEY)

# Step 2 — Segment with point hints (application/json)
# foreground_points = areas to include, background_points = areas to exclude
segment_response = requests.post(
    f"{API_BASE}/v3/tools/ai/ai-cut/segment/",
    headers={**HEADERS, "Content-Type": "application/json"},
    json={
        "input_key": INPUT_KEY,
        "output_key": "my-project/segment-v1.jpg",  # must be JPEG
        "foreground_points": [{"x": 250, "y": 300}],
        "background_points": []
    },
    timeout=30
)

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

# Refine — send updated points in a new request
# Always include ALL accumulated foreground and background points.
refine_response = requests.post(
    f"{API_BASE}/v3/tools/ai/ai-cut/segment/",
    headers={**HEADERS, "Content-Type": "application/json"},
    json={
        "input_key": INPUT_KEY,
        "output_key": "my-project/segment-v2.jpg",
        "foreground_points": [{"x": 250, "y": 300}, {"x": 270, "y": 320}],
        "background_points": [{"x": 50, "y": 50}]
    },
    timeout=30
)

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

# Step 3 — Mint a public URL to access the result
pub = requests.post(
    f"{API_BASE}/v3/file-ops/get-public-url/",
    headers={**HEADERS, "Content-Type": "application/json"},
    json={"key": refined["output_key"]},
    timeout=10
)
pub.raise_for_status()
print("Download URL:", pub.json()["url"])