Hugging Face's logo

Spaces:
enzostvs
/
lora-studio
Running on CPU Upgrade

App Files Community
17
lora-studio
File size: 4,151 Bytes 
c9a7b70
 
 
a1f821c
c9a7b70
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a1f821c
08ea78f
 
 
 
 
 
549506b
a1f821c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c9a7b70
 
549506b
c9a7b70
549506b
 
 
 
 
 
 
 
 
 
c9a7b70
 
 
 
 
a1f821c
 
 
 
 
 
 
 
 
 
 
 
 
08ea78f
 
 
 
a1f821c
 
 
08ea78f
a1f821c
 
 
 
 
 
 
 
 
 
08ea78f
 
 
 
 
 
a1f821c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
 
// latent-consistency/lcm-lora-sdv1-5
/** @type {import('./$types').RequestHandler} */

import { json, type RequestEvent } 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();

  if (model?.cardData?.thumbnail) {
    const filename = model?.cardData?.thumbnail?.split("/").pop()
    const hasImage = model?.siblings?.find((sibling: Record<string, string>) => sibling?.rfilename.includes(filename))

    if (hasImage?.rfilename) {
      model.image = `https://huggingface.co/${model.id}/resolve/main/${hasImage?.rfilename}`
    }
  } else {
    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("./", "")}`
      }
    }
  }

  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}`,
  })
}
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: {
        startsWith: slug
      }
    },
  })

  for (const model of models) {
    const hf_model = await fetch(`https://huggingface.co/api/models/${model.id}`)
    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) {
      const filename = new_model?.cardData?.thumbnail?.split("/").pop()
      const hasImage = new_model?.siblings?.find((sibling: Record<string, string>) => sibling?.rfilename.includes(filename))

      if (hasImage?.rfilename) {
        update_data.image = `https://huggingface.co/${model.id}/resolve/main/${hasImage?.rfilename}`
      }
    }

    await prisma.model.update({
      where: {
        id: model.id
      },
      data: {
        ...update_data,
        likes: new_model.likes,
        downloads: new_model.downloads,
      }
    })
  }

  return json({
    message: true
  })
}