Spaces:
Running
Running
| /** | |
| * 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 }; | |