jbilcke-hf's picture
jbilcke-hf HF staff
various improvements
bf6568f
import { pick } from "./pick"
import { Agent, Scene } from "./types"
const actions = [
"idling",
"making bubbles",
"making circles",
"opening and closing its mouth",
// "with an octopus",
"playing with another fish",
"eating fishfood",
"eating a crab",
"attacked by a jellyfish"
]
const positions = [
"at the top of the coral",
"at the bottom of the coral",
"centered in the middle",
"burrowing in the sand",
"hiding in the coral"
]
const lights = [
"during the day",
]
const actionnables = [
"fish",
"coral",
"sand"
]
export const agent: Agent = {
title: "Fish",
type: "fish",
simulate: (): Scene => {
const action = pick(actions)
const position = pick(positions)
const light = pick(lights)
const prompt = [
`medium shot of a clownfish`,
action,
position,
light,
`in front of yellow coral`,
`high res`,
`underwater footage`,
].join(", ")
return {
action,
position,
light,
actionnables,
prompt
}
}
}