createSpace.mts

import { v4 as uuidv4 } from "uuid"
import { createRepo, uploadFiles, whoAmI } from "@huggingface/hub"
import type { RepoDesignation, Credentials } from "@huggingface/hub"
import slugify from "slugify"

import { RepoFile } from "./types.mts"

export const createSpace = async (files: RepoFile[], token: string) => {

  const credentials: Credentials = { accessToken: token }

  const { name: username } = await whoAmI({ credentials })

  let slug = ``
  let title = ``
  const readme = files.find(p => p.path === "README.md")
  try {
    const matches = readme.content.match(/title: ([^\n]+)\n/)
    title = matches?.[1] || ""
    slug = (slugify as any)(title) as string
    if (!slug.length) {
      throw new Error("sluggification failed")
    }
  } catch (err) {
    slug = `sf-${uuidv4().slice(0, 3)}`
  }

  const repoName = `${username}/${slug}`

  const repo: RepoDesignation = { type: "space", name: repoName }
  console.log(`Creating space at ${repoName}${title ? ` (${title})` : ''}`)

  await createRepo({
    repo,
    credentials,
    license: "mit",
    sdk:
      files.some(file => file.path.includes("Dockerfile"))
        ? "docker"
      : files.some(file => file.path.includes("app.py"))
        ? "streamlit"
        : "static" // "streamlit" | "gradio" | "docker" | "static";
  });

  console.log("uploading files..")
  await uploadFiles({
    repo,
    credentials,
    files: files.map(file => ({
      path: file.path,
      content: new Blob([ file.content ])
    })),
  });

  console.log("upload done!")

  // TODO we should keep track of the repo and delete it after 30 min
  // or delete it if we reached 20 repos
  // await deleteRepo({ repo, credentials })
}