add agent smith

src/app/agents/index.ts

@@ -3,9 +3,10 @@ import { Agent, AgentType } from "./types"
 import { agent as ant } from "./ant"
 import { agent as fish } from "./fish"
 import { agent as fox } from "./fox"
+import { agent as smith } from "./smith"
 
-export const agents = { ant, fish, fox }
+export const agents = { ant, fish, fox, smith }
 
-export const defaultAgent: AgentType = "fox"
+export const defaultAgent: AgentType = "fish"
 
 export const getAgent = (type?: AgentType) => agents[type || defaultAgent] || agents[defaultAgent]

src/app/agents/smith.ts

@@ -0,0 +1,42 @@
+import { pick } from "./pick"
+import { Agent, Scene } from "./types"
+
+const actions = [
+  "standing and waiting",
+  "standing and looking at camera",
+  "standing and adjusting their tie",
+  "standing and talking on a cellphone",
+  "standing and reading the journal"
+]
+
+
+const positions = [
+  "on the roof of a building",
+  "in the lobby of a building",
+  "in an elevator",
+  "on the sidewalk of a street"
+]
+
+export const agent: Agent = {
+  title: "Smith",
+  type: "smith",
+  simulate: (): Scene => {
+    const action = pick(actions)
+    const position = pick(positions)
+
+    const prompt = [
+      `static medium shot of Agent Smith from the Matrix`,
+      `wearing a black costume with black tie and black sunglasses`,
+      action,
+      position,
+      `high res`,
+      `documentary`,
+    ].join(", ")
+
+    return {
+      action,
+      position,
+      prompt
+    }
+  }
+}

src/app/agents/types.ts

@@ -1,4 +1,4 @@
-export type AgentType = 'ant' | 'fish' | 'fox'
+export type AgentType = 'ant' | 'fish' | 'fox' | 'smith'
 
 export interface Scene {
   action: string

src/app/main.tsx

@@ -19,16 +19,15 @@ import { agents, defaultAgent, getAgent } from "./agents"
 export default function Main() {
   const [url, setUrl] = useState<string>()
   const [isPending, startTransition] = useTransition()
-  const [agent, setAgent] = useState<Agent>()
   const [scene, setScene] = useState<Scene>()
   const ref = useRef<AgentType>(defaultAgent)
    
   useEffect(() => {
     
     const updateView = async () => {
-      console.log(`update view..`)
+      // console.log(`update view..`)
 
-      startTransition(async () => {
+      await startTransition(async () => {
 
         // console.log(`getting agent..`)
         const type = ref?.current
@@ -41,18 +40,20 @@ export default function Main() {
         const newUrl = await render(scene.prompt)
 
         if (type !== ref?.current) {
-          console.log("agent type changed while we were rendering")
+          console.log("agent type changed! reloading scene")
           setTimeout(() => { updateView() }, 0)
           return
         } 
 
-        console.log(`newUrl: ${newUrl}`)
-        if (!newUrl) {
+        if (newUrl) {
+          // console.log(`got a new url: ${newUrl}`)
           setUrl(newUrl)
+          setScene(scene)
+          setTimeout(() => { updateView()}, 1000)
+        } else {
+          // console.log(`going to wait a bit more: ${newUrl}`)
+          setTimeout(() => { updateView()}, 3000)
         }
-        setAgent(agent)
-        setScene(scene)
-        setTimeout(() => { updateView()}, 2000)
       })
     }