import requests

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

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

upload_response.raise_for_status()
input_file_path = upload_response.json()["file_paths"][0]
print("Uploaded:", input_file_path)

# 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_file_path": input_file_path,
        "output_file_path": "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_file_path"])

# 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_file_path": input_file_path,
        "output_file_path": "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_file_path"])