trending sort

prisma/schema.prisma

@@ -18,6 +18,7 @@ model Model {
   downloads       Int?
   instance_prompt String?
   isPublic        Boolean   @default(false)
+  likes7d         Int?      @default(0)
   user            User?     @relation(fields: [userId], references: [sub])
   userId          String?
   gallery         Gallery[]

server.js

@@ -19,7 +19,7 @@ app.listen(3000, () => {
 });
 
 cron.schedule("0 * * * *", async () => {
-    console.log('Scraping models', process.env.SPACE_HOST)
+    console.log('Scraping models')
     const request = await fetch(`https://${process.env.SPACE_HOST}/api/scrap-models`, {
         method: 'POST',
         headers: {
@@ -31,8 +31,9 @@ cron.schedule("0 * * * *", async () => {
     console.info(response?.message)
 });
 
-cron.schedule("0 0 * * *", async () => {
-    const request = await fetch(`https://${process.env.SPACE_HOST}/api/models`, {
+cron.schedule("0 0 */12 * *", async () => {
+    console.log('Updating models')
+    const request = await fetch(`http://${process.env.SPACE_HOST}/api/models`, {
         method: 'PATCH',
         headers: {
             'Content-Type': 'application/json',

src/lib/utils/index.ts

@@ -16,12 +16,6 @@ export const COMMUNITY_FILTER_OPTIONS = [
 ];
 
 export const MODELS_FILTER_OPTIONS = [
-  {
-    label: "Most Liked",
-    value: "likes",
-    icon: "lucide:heart",
-    iconColor: "text-red-500"
-  },
   {
     label: "Hotest",
     value: "hotest",
@@ -33,7 +27,13 @@ export const MODELS_FILTER_OPTIONS = [
     value: "newest",
     icon: "ph:clock-bold",
     iconColor: "text-blue-500"
-  }
+  },
+  {
+    label: "Most Liked",
+    value: "likes",
+    icon: "lucide:heart",
+    iconColor: "text-red-500"
+  },
 ];
 
 export const SIDEBAR_MENUS = [ {

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

@@ -21,6 +21,15 @@ export async function GET(request : RequestEvent) {
   const search = request.url.searchParams.get('search') || ''
   const limit = parseInt(request.url.searchParams.get('limit') || '20')
 
+  const orderBy: Record<string, string> = {}
+  if (filter === 'hotest') {
+    orderBy['likes7d'] = 'desc'
+  } else if (filter === 'likes') {
+    orderBy['likes'] = 'desc'
+  } else {
+    orderBy['createdAt'] = 'desc'
+  }
+
   const cards = await prisma.model.findMany({
     where: {
       ...(IS_ADMIN ? {} : { isPublic: true }),
@@ -28,9 +37,7 @@ export async function GET(request : RequestEvent) {
         { id: { contains: search } },
       ]
     },
-    orderBy: {
-      ...(filter === 'hotest' ? { downloads: 'desc' } : filter === 'likes' ? { likes: 'desc' } : { createdAt: 'desc' })
-    },
+    orderBy: orderBy,
     skip: page * limit,
     take: limit,
   })
@@ -59,28 +66,26 @@ export async function PATCH({ request } : RequestEvent) {
     }, { status: 401 });
   }
 
-  const models = await prisma.model.findMany({
-    where: {
-      isPublic: true,
-    }
-  })
+  const hf_request = await fetch(`https://huggingface.co/api/models?limit=10000&filter=lora%2Cdiffusers&sort=likes7d`)
+  const hf_models = await hf_request.json();
 
   let total_updates = 0;
 
-  for (const model of models) {
-    const response = await fetch(`https://huggingface.co/api/models/${model.id}`)
-    const hf_model = await response.json();
-
+  for (const model of hf_models) {
     await prisma.model.update({
       where: {
         id: model.id
       },
       data: {
-        likes: hf_model.likes,
-        downloads: hf_model.downloads,
+        likes: model.likes,
+        downloads: model.downloads,
+        likes7d: model.likes7d,
       }
     })
-    total_updates++
+    .then(() => {
+      total_updates++
+    })
+    .catch(() => {})
   }
 
   return json({