add route to manually update models from specific author

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

@@ -1,7 +1,7 @@
 // latent-consistency/lcm-lora-sdv1-5
 /** @type {import('./$types').RequestHandler} */
 
-import { json } from '@sveltejs/kit';
+import { json, type RequestEvent } from '@sveltejs/kit';
 // import moment from 'moment';
 import prisma from '$lib/prisma';
 
@@ -18,30 +18,34 @@ export async function POST({ request, params }) {
   const response = await fetch(`https://huggingface.co/api/models/${slug}`)
   const model = await response.json();
 
-  const hasImages = model?.siblings?.filter((sibling: Record<string, string>) => sibling?.rfilename.endsWith(".png") || sibling?.rfilename.endsWith(".jpeg") || sibling?.rfilename.endsWith(".jpg"))
-  if (hasImages.length > 0) {
-    model.image = hasImages[1]?.rfilename ? `https://huggingface.co/${model.id}/resolve/main/${hasImages[1]?.rfilename}` : `https://huggingface.co/${model.id}/resolve/main/${hasImages[0]?.rfilename}`
+  if (model?.cardData?.thumbnail) {
+    model.image = `https://huggingface.co/${model.id}/resolve/main/${model?.cardData?.thumbnail}`
   } else {
-    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 hasImages = model?.siblings?.filter((sibling: Record<string, string>) => sibling?.rfilename.endsWith(".png") || sibling?.rfilename.endsWith(".jpeg") || sibling?.rfilename.endsWith(".jpg"))
+    if (hasImages.length > 0) {
+      model.image = hasImages[1]?.rfilename ? `https://huggingface.co/${model.id}/resolve/main/${hasImages[1]?.rfilename}` : `https://huggingface.co/${model.id}/resolve/main/${hasImages[0]?.rfilename}`
+    } else {
+      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 image found"
+          }, { 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("./", "")}`
       }
-      const imageRegex = /!\[.*\]\((.*)\)/
-      let image = readme.match(imageRegex)?.[1]
-  
-      if (!image) {
-        return json({
-          message: "No image found"
-        }, { 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("./", "")}`
     }
   }
 
@@ -60,3 +64,51 @@ export async function POST({ request, params }) {
     message: `Successfully added the model ${model.id}`,
   })
 }
+export async function PATCH({ request, params } : RequestEvent) {
+  const headers = Object.fromEntries(request.headers.entries());
+
+  if (headers["x-hf-token"] !== process.env.SECRET_HF_TOKEN) {
+    return Response.json({
+      message: "Wrong castle fam :^)"
+    }, { status: 401 });
+  }
+
+  const slug = params?.slug?.replace("@", "/")
+
+  const models = await prisma.model.findMany({
+    where: {
+      id: slug
+    }
+  })
+
+  for (const model of models) {
+    const hf_model = await fetch(`https://huggingface.co/api/models/${slug}`)
+    const new_model = await hf_model.json();
+    const update_data = {
+      instance_prompt: model.instance_prompt,
+      image: model.image,
+    }
+    
+    if (new_model?.cardData?.instance_prompt) {
+      update_data.instance_prompt = new_model?.cardData?.instance_prompt
+    }
+    if (new_model?.cardData?.thumbnail) {
+      update_data.image = `https://huggingface.co/${model.id}/resolve/main/${new_model?.cardData?.thumbnail}`
+    }
+
+    await prisma.model.update({
+      where: {
+        id: model.id
+      },
+      data: {
+        ...update_data,
+        likes: new_model.likes,
+        downloads: new_model.downloads,
+      }
+    })
+  }
+
+  return json({
+    message: true
+  })
+}