Stream AI Media in Real Time

import asyncio
import httpx

API_KEY = "YOUR_CREATIVEAI_KEY"
BASE_URL = "https://api.creativeai.run/v1"

async def generate_image(client: httpx.AsyncClient, prompt: str) -> dict:
    """Fire a single generation request."""
    resp = await client.post(
        f"{BASE_URL}/images/generations",
        json={"model": "seedream-3", "prompt": prompt, "size": "1024x1024"},
        headers={"Authorization": f"Bearer {API_KEY}"},
        timeout=30,
    )
    resp.raise_for_status()
    return resp.json()

async def batch_generate(prompts: list[str]) -> list[dict]:
    """Run up to 50 concurrent image generations."""
    async with httpx.AsyncClient() as client:
        tasks = [generate_image(client, p) for p in prompts]
        results = await asyncio.gather(*tasks)
    return results

# Generate 50 images in parallel — all hit the API simultaneously
prompts = [f"product photo variant {i}, white background" for i in range(50)]
results = asyncio.run(batch_generate(prompts))
print(f"Generated {len(results)} images concurrently")

import { useCallback, useState } from "react";

export function useStreamGeneration() {
  const [progress, setProgress] = useState(0);
  const [result, setResult] = useState<string | null>(null);

  const generate = useCallback(async (prompt: string) => {
    setProgress(0);
    setResult(null);

    const resp = await fetch(
      "https://api.creativeai.run/v1/images/generations/stream",
      {
        method: "POST",
        headers: {
          Authorization: `Bearer ${process.env.NEXT_PUBLIC_API_KEY}`,
          "Content-Type": "application/json",
        },
        body: JSON.stringify({
          model: "seedream-3",
          prompt,
          stream: true,
        }),
      }
    );

    const reader = resp.body!.getReader();
    const decoder = new TextDecoder();
    let buffer = "";

    while (true) {
      const { done, value } = await reader.read();
      if (done) break;

      buffer += decoder.decode(value, { stream: true });
      const lines = buffer.split("\n");
      buffer = lines.pop() || "";

      for (const line of lines) {
        if (!line.startsWith("data: ")) continue;
        const data = JSON.parse(line.slice(6));
        if (data.status === "progress") setProgress(data.percent);
        if (data.status === "complete") setResult(data.url);
      }
    }
  }, []);

  return { generate, progress, result };
}