Hugging Face's logo

Spaces:
jbilcke-hf
/
ai-tube
Running

App Files Community
10
ai-tube
File size: 1,488 Bytes 
8101ed0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
// should we really have default prompts in here?
// I think they should probably be defined at the applicative software layer (ie. in the client)

export function addWordsIfNotPartOfThePrompt(prompt: string = "", words: string[] = []): string {
  const promptWords = prompt.split(",").map(w => w.trim().toLocaleLowerCase())

  return [
    prompt,
    // we add our keywords, unless they are already part of the prompt
    ...words.filter(w => !promptWords.includes(w.toLocaleLowerCase()))
  ].join(", ")
}

 export function getPositivePrompt(prompt: string = "", triggerWord = "") {
  return addWordsIfNotPartOfThePrompt(prompt, [
    triggerWord,
    "sublime",
    "pro quality",
    "sharp",
    "crisp",
    "beautiful",
    "impressive",
    "amazing",
    "4K",
    "hd"
  ])
}

export function getNegativePrompt(prompt: string = "") {
  return addWordsIfNotPartOfThePrompt(prompt, [
    "cropped",
    // "underexposed", // <-- can be a desired style
    // "overexposed", // <-- can be a desired style
    "logo",
    "hud",
    "ui",
    "censored",
    "blurry",
    "watermark",
    "watermarked",
    "copyright",
    "extra digit",
    "fewer digits",
    "bad fingers",
    "bad quality",
    "worst quality",
    "low quality",
    "low resolution",
    "glitch", // <-- keep or not? could be a desired style?
    // "deformed",
    // "mutated",
    // "ugly",
    // "disfigured",
    // "3D render", // <-- keep or not? could be a desired style?
    "signature"
  ])
}