Mistri / createSpace.mts
acecalisto3's picture
Upload 20 files
e9674a6 verified
raw
history blame
1.69 kB
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 })
}