Spaces:
Sleeping
Sleeping
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 }) | |
} |