delete prisma document + image in dataset

app/api/collections/[id]/route.ts

@@ -1,14 +1,56 @@
 import { PrismaClient } from '@prisma/client'
 
-// const prisma = new PrismaClient()
+import { RemoverDataset } from '@/utils/remover'
 
-export async function DELETE(request: Request) {
-  const { id } = await request.json()
+const prisma = new PrismaClient()
 
-  console.log(id)
+export async function DELETE(request: Request, { params }: { params: { id: string } }) {
+  const { headers } = request
+
+  if (!headers?.get('Authorization') || headers.get('Authorization') !== `Bearer ${process.env.HF_TOKEN}`) {
+    return Response.json(
+      {
+        status: 401,
+        ok: false,
+        message: 'Wrong castle fam :/'
+      }
+    )
+  }
+
+  const collection = await prisma.collection.findUnique({
+    where: {
+      id: parseInt(params.id)
+    }
+  })
+  if (!collection) {
+    return Response.json(
+      {
+        status: 404,
+        ok: false
+      }
+    )
+  }
+
+  const isImageDeleted = await RemoverDataset(collection?.file_name as string)
+  if (!isImageDeleted.ok) {
+    return Response.json(
+      {
+        status: 500,
+        ok: false,
+        message: isImageDeleted.message
+      }
+    )
+  }
+
+  await prisma.collection.delete({
+    where: {
+      id: parseInt(params.id)
+    }
+  })
 
   return Response.json(
     {
+      message: `Image ${params.id} deleted`,
       status: 200,
       ok: true
     }

app/api/route.ts

@@ -1,7 +1,7 @@
 import { PrismaClient } from '@prisma/client'
 
 import list_styles from "@/assets/list_styles.json"
-import { UploaderDataset } from './uploader'
+import { UploaderDataset } from '../../utils/uploader'
 import { isTextNSFW } from '@/utils/checker/prompt'
 import { isImageNSFW } from '@/utils/checker/image'
 
@@ -11,7 +11,7 @@ export async function POST(
   request: Request,
 ) {
   const global_headers = {
-    Authorization: `Bearer ${process.env.NEXT_APP_HF_TOKEN}`,
+    Authorization: `Bearer ${process.env.HF_TOKEN}`,
     'Content-Type': 'application/json',
     ['x-use-cache']: "0"
   }

prisma/schema.prisma

@@ -4,7 +4,7 @@ generator client {
 
 datasource db {
   provider = "sqlite"
-  url      = "file://data/dev.db"
+  url      = "file:../data/dev.db"
 }
 
 model Collection {

utils/remover.ts

@@ -0,0 +1,25 @@
+import { deleteFile } from "../node_modules/@huggingface/hub/dist";
+import type { RepoDesignation, Credentials } from "../node_modules/@huggingface/hub/dist";
+
+export const RemoverDataset = async (name: string) => {
+  const repo: RepoDesignation = { type: "dataset", name: "enzostvs/stable-diffusion-tpu-generations" };
+  const credentials: Credentials = { accessToken: process.env.HF_TOKEN as string };
+
+  const res: any = await deleteFile({
+    repo,
+    credentials,
+    path: `images/${name}.png`,
+  });
+
+  if (res?.error) return {
+    status: 500,
+    ok: false,
+    message: res?.error
+  };
+
+  return {
+    status: 200,
+    ok: true,
+  };
+
+}

{app/api → utils}/uploader.ts

@@ -1,9 +1,9 @@
-import { uploadFile } from "./../../node_modules/@huggingface/hub/dist";
-import type { RepoDesignation, Credentials } from "./../../node_modules/@huggingface/hub/dist";
+import { uploadFile } 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 credentials: Credentials = { accessToken: process.env.HF_TOKEN as string };
 
   const res: any = await uploadFile({
     repo,