add manually model + catch error

src/routes/api/generate/+server.ts

@@ -49,6 +49,8 @@ export async function POST({ request, cookies } : RequestEvent) {
     }),
   })
   .then((response) => {
+    if (response.status !== 200) throw new Error(response.statusText)
+
     return response.arrayBuffer()
   })
   .then((response) => {
@@ -62,9 +64,7 @@ export async function POST({ request, cookies } : RequestEvent) {
 
   if ("error" in response) {
     return json({
-      error: {
-        token: response.error
-      }
+      error: response.error
     }, { status: 400 })
   }
 
@@ -73,8 +73,8 @@ export async function POST({ request, cookies } : RequestEvent) {
   if (token) {
     const user = await tokenIsAvailable(token)
     if (user?.sub) {      
-      const dir = await promises.opendir(process.env.PUBLIC_FILE_UPLOAD_DIR).catch(() => null)
-      if (!dir) await promises.mkdir(process.env.PUBLIC_FILE_UPLOAD_DIR)
+      const dir = await promises.opendir(process?.env?.PUBLIC_FILE_UPLOAD_DIR as string).catch(() => null)
+      if (!dir) await promises.mkdir(process?.env?.PUBLIC_FILE_UPLOAD_DIR as string)
       const file_name_formatted = randomUUID() + "_" + generation?.inputs?.replaceAll(/[^a-zA-Z0-9]/g, "-") + ".png"
       await promises.writeFile(`${process.env.PUBLIC_FILE_UPLOAD_DIR}/${file_name_formatted}`, response)
 

src/routes/api/scrap-models/+server.ts

@@ -47,7 +47,7 @@ export async function POST({ request }) {
         image: model.image,
         likes: model.likes,
         downloads: model.downloads,
-        isPublic: false,
+        isPublic: true,
       }
     }).catch(() => {})
 

src/routes/api/scrap-models/[slug]/+server.ts

@@ -0,0 +1,57 @@
+// latent-consistency/lcm-lora-sdv1-5
+/** @type {import('./$types').RequestHandler} */
+
+import { json } from '@sveltejs/kit';
+// import moment from 'moment';
+import prisma from '$lib/prisma';
+
+export async function POST({ request, params }) {
+  const headers = Object.fromEntries(request.headers.entries());
+  const slug= params?.slug?.replace("@", "/")
+
+  if (headers["x-hf-token"] !== process.env.SECRET_HF_TOKEN) {
+    return Response.json({
+      message: "Wrong castle fam :^)"
+    }, { status: 401 });
+  }
+
+  const response = await fetch(`https://huggingface.co/api/models/${slug}`)
+  const model = await response.json();
+
+  const hasReadme = model?.siblings?.find((sibling: Record<string, string>) => sibling?.rfilename === "README.md")
+    if (hasReadme) {
+      const readmeRes = await fetch(`https://huggingface.co/${model.id}/raw/main/README.md`)
+      const readme = await readmeRes.text().catch(() => null)
+      if (!readme) {
+        return json({
+          message: "No readme"
+        }, { status: 404 })
+      }
+      const imageRegex = /!\[.*\]\((.*)\)/
+      let image = readme.match(imageRegex)?.[1]
+
+      if (!image) {
+        return json({
+          message: "No readme"
+        }, { status: 404 })
+      }
+      image = image.replace(/&#x2F;/g, "/")
+      if (image.startsWith("http")) model.image = image
+      else model.image = `https://huggingface.co/${model.id}/resolve/main/${image.replace("./", "")}`
+    }
+
+    await prisma.model.create({
+      data: {
+        id: model.id,
+        image: model.image,
+        likes: model.likes,
+        downloads: model.downloads,
+        isPublic: true,
+        instance_prompt: model?.cardData?.instance_prompt,
+      }
+    }).catch(() => {})
+
+  return json({
+    message: `Successfully added the model ${model.id}`,
+  })
+}

src/routes/generate/+page.svelte

@@ -12,6 +12,7 @@
 	import Response from "$lib/components/generate/Response.svelte";
 	import Autocomplete from "$lib/components/models/autocomplete/Autocomplete.svelte";
 	import { generationStore } from "$lib/stores/use-generation";
+	import { error } from "$lib/utils/toaster.js";
 
 	export let data
 	let generation = get(generationStore);
@@ -40,11 +41,15 @@
 
 		const response = await request.json();
 
-		generationStore.set({
-			image: response?.image,
-			gallery: response?.gallery,
-			form: form
-		})
+		if (response?.error) {
+			error(response?.error)
+		} else {
+			generationStore.set({
+				image: response?.image,
+				gallery: response?.gallery,
+				form: form
+			})
+		}
     loading = false
 	}
 </script>