import { uploadFiles } from "./../../node_modules/@huggingface/hub/dist";
import type { RepoDesignation, Credentials } from "./../../node_modules/@huggingface/hub/dist";

export const UploaderDataset = async (blob: Blob, name: string) => {
  const repo: RepoDesignation = { type: "dataset", name: "enzostvs/stable-diffusion-tpu-generations" };
  const credentials: Credentials = { accessToken: process.env.NEXT_APP_HF_TOKEN as string };

  const res: any = await uploadFiles({
    repo,
    credentials,
    files: [
      {
        path: `images/${name}.png`,
        content: blob,
      },
    ],
  });

  if (res?.error) return {
    status: 500,
    ok: false,
    message: res?.error
  };

  return {
    status: 200,
    ok: true,
  };

}