Every Listing Visual. One API. Pay Per Image.

import hmac, hashlib, json
from fastapi import FastAPI, Request, HTTPException

app = FastAPI()

# Store pending jobs when you submit them:
# pending_jobs[job_id] = {"listing_id": "42", "room": "kitchen"}

@app.post("/webhooks/creativeai")
async def handle_video_callback(request: Request):
    body = await request.body()

    # 1. Verify HMAC-SHA256 signature (secret = your API key)
    sig = request.headers.get("x-creativeai-signature", "")
    expected = "sha256=" + hmac.new(
        API_KEY.encode(), body, hashlib.sha256
    ).hexdigest()
    if not hmac.compare_digest(sig, expected):
        raise HTTPException(401, "Invalid signature")

    event = json.loads(body)
    job_id = event["id"]
    status = event["status"]            # "completed" or "failed"
    video_url = event.get("video_url")  # CDN URL when completed

    # 2. Match callback to listing + room
    job = pending_jobs.get(job_id)
    if not job:
        return {"ok": True}  # idempotent — already processed

    # 3. Update your listing record
    if status == "completed":
        await save_video_to_listing(
            listing_id=job["listing_id"],
            room=job["room"],
            video_url=video_url,
        )
        await mark_room_ready(job["listing_id"], job["room"])

    # 4. Check if all rooms are done -> notify agent / publish MLS
    if await all_rooms_ready(job["listing_id"]):
        await publish_listing_video_package(job["listing_id"])

    del pending_jobs[job_id]
    return {"ok": True}

# Webhook payload you receive:
# {
#   "id": "vg_abc123",
#   "status": "completed",
#   "video_url": "https://cdn.creativeai.run/videos/vg_abc123.mp4",
#   "model": "kling-v3",
#   "duration": "5s",
#   "created_at": "2026-03-12T10:00:00Z",
#   "completed_at": "2026-03-12T10:01:12Z"
# }

import httpx, asyncio, time

API_KEY = "your_api_key"
BASE    = "https://api.creativeai.run/v1/video/generations"

# After submitting all rooms (see batch example above), poll for status:
async def track_listing_render(jobs: list[dict], poll_interval: int = 5):
    """
    Poll all room jobs and yield status updates until every room is done.
    Perfect for feeding a real-time dashboard or SSE stream to your frontend.
    """
    pending = {j["id"]: j["room"] for j in jobs}
    results = {}

    async with httpx.AsyncClient() as client:
        while pending:
            for job_id, room in list(pending.items()):
                resp = await client.get(
                    f"{BASE}/{job_id}",
                    headers={"Authorization": f"Bearer {API_KEY}"},
                    timeout=15,
                )
                data = resp.json()
                status = data["status"]

                # Yield per-room status update (push to your UI via SSE / WS)
                yield {
                    "room": room,
                    "job_id": job_id,
                    "status": status,         # pending | processing | completed | failed
                    "video_url": data.get("video_url"),
                    "elapsed": time.time(),
                }

                if status in ("completed", "failed"):
                    results[room] = data
                    del pending[job_id]

            if pending:
                await asyncio.sleep(poll_interval)

    # All rooms done — assemble the listing package
    yield {
        "event": "listing_complete",
        "rooms_completed": len([r for r in results.values() if r["status"] == "completed"]),
        "rooms_failed":    len([r for r in results.values() if r["status"] == "failed"]),
        "results": results,
    }

# Usage in your API handler (FastAPI + SSE example):
#
# @app.get("/listings/{id}/render-status")
# async def stream_render_status(id: str):
#     jobs = await get_pending_jobs(id)
#     async def generate():
#         async for update in track_listing_render(jobs):
#             yield f"data: {json.dumps(update)}\n\n"
#     return StreamingResponse(generate(), media_type="text/event-stream")
#
# Your frontend renders:
#   Kitchen ........... ✅ Completed (62s)
#   Living Room ....... ⏳ Processing
#   Master Bedroom .... ⏳ Processing
#   Backyard .......... 🕐 Pending