Upscale a video

import { readFile } from "node:fs/promises";
import { basename } from "node:path";

const SOURCE = "old-clip.mp4";


// Files live inside a workspace, so the upload needs one. An account
// always has at least one, and the first is the default.
async function firstWorkspaceId() {
  const response = await fetch(`${API}/workspaces`, { headers });

  if (!response.ok) {
    throw new Error("Could not list workspaces: " + response.status);
  }

  const body = await response.json();
  return body.data[0].id;
}


// Push the raw bytes into the media library. The response carries the
// file_hash the AI endpoints take as input.
async function upload(path, workspaceId) {
  const bytes = await readFile(path);

  const query = new URLSearchParams({
    file_name: basename(path),
    mime_type: "video/mp4",
  });

  const url = `${API}/workspaces/${workspaceId}/library?${query}`;

  const response = await fetch(url, {
    method: "POST",
    headers: {
      ...headers,
      "Content-Type": "application/octet-stream",
    },
    body: bytes,
  });

  if (!response.ok) {
    throw new Error("Upload failed: " + response.status);
  }

  const body = await response.json();
  return body.data;
}


// Start the upscale job. This returns straight away with a job id.
async function startUpscale(fileHash, workspaceId) {
  const response = await fetch(`${API}/ai/upscale-video`, {
    method: "POST",
    headers: {
      ...headers,
      "Content-Type": "application/json",
    },
    // To skip the upload, drop file_hash and send a public https URL
    // as file_url at the top level of the body, next to params.
    body: JSON.stringify({
      workspace_id: workspaceId,
      params: {
        file_hash: fileHash,
        target_resolution: "4k",
        target_fps: 30,
      },
    }),
  });

  if (!response.ok) {
    throw new Error("Could not start the job: " + response.status);
  }

  const body = await response.json();
  return body.data.job_id;
}


const workspaceId = await firstWorkspaceId();
const asset = await upload(SOURCE, workspaceId);

const jobId = await startUpscale(asset.file_hash, workspaceId);
const job = await waitForJob(jobId);

// The signed URL is the result. Fetch it, pipe it to your storage,
// or hand it to the browser.
console.log(job.output.url);