static/js/network.js

/**
 * Reconcile paths for hf.space resources fetched from hf.co iframe
 */

const pathFor = (path) => {
  const basePath = document.location.origin + document.location.pathname;
  return new URL(path, basePath).href;
};

const generateDetails = async () => {
  const details = await fetch(pathFor('details'));
  return await details.json();
};

const createTask = async (prompt) => {
  const taskResponse = await fetch(pathFor(`task/create?prompt=${prompt}`));
  const task = await taskResponse.json();

  return task;
};

const pollTask = async (task) => {
  const taskResponse = await fetch(pathFor(`task/poll?task_id=${task.task_id}`));

  return await taskResponse.json();
};

const longPollTask = async (task, interval = 10_000, max) => {
  const etaDisplay = document.querySelector('.eta');

  task = await pollTask(task);

  if (task.status === 'completed' || (max && task.poll_count > max)) {
    return task;
  }

  if (task.status === 'failed') {
    return task;
  }

  etaDisplay.textContent = Math.round(task.eta);

  await new Promise((resolve) => setTimeout(resolve, interval));

  return await longPollTask(task, interval, max);
};

export { generateDetails, createTask, longPollTask };