const API_BASE = "https://api.exactly.ai/public-api";
const headers = { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" };

async function createModel(name, customData = null) {
  const res = await fetch(`${API_BASE}/v1/models/`, {
    method: "POST", headers, body: JSON.stringify({ name, custom_data: customData })
  });
  return (await res.json()).uid;
}

async function uploadTrainingImage(modelUid, imageFile) {
  const formData = new FormData();
  formData.append("image", imageFile);
  const res = await fetch(`${API_BASE}/v1/models/${modelUid}/train_images/`, {
    method: "POST", headers: { "Authorization": `Bearer ${API_KEY}` }, body: formData
  });
  return await res.json();
}

async function trainModel(modelUid) {
  await fetch(`${API_BASE}/v1/models/${modelUid}/train/`, { method: "POST", headers });
}

async function getTrainingProgress(modelUid) {
  const res = await fetch(`${API_BASE}/v1/models/${modelUid}/train/progress/`, { headers });
  return (await res.json()).progress;
}

async function generateImages(modelUid, prompt, numImages = 4, size = [1024, 1024]) {
  const res = await fetch(`${API_BASE}/v1/images/`, {
    method: "POST", headers, body: JSON.stringify({ model_uid: modelUid, prompt, num_images: numImages, size })
  });
  return (await res.json()).images;
}

async function upscaleImage(imageUid, scale = "4k") {
  await fetch(`${API_BASE}/v1/images/${imageUid}/upscales/`, {
    method: "POST", headers, body: JSON.stringify({ scale })
  });
  await sleep(3000);
  const res = await fetch(`${API_BASE}/v1/images/${imageUid}/upscales/${scale}/`, { headers });
  return await res.json();
}

async function generateVideo(modelUid, prompt, size, fps = 24, duration = 5) {
  const res = await fetch(`${API_BASE}/v1/videos/`, {
    method: "POST", headers, body: JSON.stringify({ model_uid: modelUid, prompt, num_videos: 1, size, fps, duration })
  });
  return await res.json();
}

async function listModels(activeOnly = false) {
  const params = activeOnly ? "?active=true" : "";
  const res = await fetch(`${API_BASE}/v1/models/${params}`, { headers });
  return (await res.json()).items;
}