Spaces:
Sleeping
Sleeping
Julian Bilcke commited on
Commit ·
b58c38b
1
Parent(s): 1cea837
update
Browse files
src/components/interface/latent-engine/core/prompts/getCharacterReferencePrompt.ts
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import { ClapEntity } from "@aitube/clap";
|
| 2 |
+
|
| 3 |
+
import { getCharacterPrompt } from "./getCharacterPrompt";
|
| 4 |
+
|
| 5 |
+
/**
|
| 6 |
+
* Return a prompt for a "formal" picture, centered, neutral etc
|
| 7 |
+
* @param entity
|
| 8 |
+
* @returns
|
| 9 |
+
*/
|
| 10 |
+
export function getCharacterReferencePrompt(entity: ClapEntity) {
|
| 11 |
+
const characterPrompt = [
|
| 12 |
+
`beautiful`,
|
| 13 |
+
`close-up`,
|
| 14 |
+
`photo portrait`,
|
| 15 |
+
`id photo`,
|
| 16 |
+
getCharacterPrompt(entity),
|
| 17 |
+
`neutral expression`,
|
| 18 |
+
`neutral background`,
|
| 19 |
+
`frontal`,
|
| 20 |
+
`photo studio`,
|
| 21 |
+
`crisp`,
|
| 22 |
+
`sharp`,
|
| 23 |
+
`intricate details`,
|
| 24 |
+
`centered`,
|
| 25 |
+
// `aligned`
|
| 26 |
+
].map(i => i.trim()).filter(i => i).join(", ")
|
| 27 |
+
|
| 28 |
+
return characterPrompt
|
| 29 |
+
}
|
src/components/interface/latent-engine/core/prompts/getVideoPrompt.ts
CHANGED
|
@@ -26,35 +26,21 @@ export function getVideoPrompt(
|
|
| 26 |
return false
|
| 27 |
}
|
| 28 |
|
| 29 |
-
if (
|
| 30 |
-
|
| 31 |
-
|
| 32 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
if (category === "splat" ||
|
| 42 |
-
category === "mesh" ||
|
| 43 |
-
category === "depth"
|
| 44 |
) {
|
| 45 |
-
return
|
| 46 |
}
|
| 47 |
-
|
| 48 |
-
if (category === "storyboard" ||
|
| 49 |
-
category === "video") {
|
| 50 |
-
return false
|
| 51 |
-
}
|
| 52 |
-
|
| 53 |
-
if (category === "transition") {
|
| 54 |
-
return false
|
| 55 |
-
}
|
| 56 |
-
|
| 57 |
-
return true
|
| 58 |
})
|
| 59 |
|
| 60 |
tmp.sort((a, b) => b.label.localeCompare(a.label))
|
|
|
|
| 26 |
return false
|
| 27 |
}
|
| 28 |
|
| 29 |
+
if (
|
| 30 |
+
category === "character" ||
|
| 31 |
+
category === "location" ||
|
| 32 |
+
category === "time" ||
|
| 33 |
+
category === "era" ||
|
| 34 |
+
category === "lighting" ||
|
| 35 |
+
category === "weather" ||
|
| 36 |
+
category === "action" ||
|
| 37 |
+
category === "style" ||
|
| 38 |
+
category === "camera" ||
|
| 39 |
+
category === "generic"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
) {
|
| 41 |
+
return true
|
| 42 |
}
|
| 43 |
+
return false
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
})
|
| 45 |
|
| 46 |
tmp.sort((a, b) => b.label.localeCompare(a.label))
|