import { ComputeJSON, ComputeText, GenerateImage, sb, Substrate } from "npm:substrate"; import { z } from "npm:zod"; import { zodToJsonSchema } from "npm:zod-to-json-schema"; export default async function(req: Request): Promise { const { subject = "ball lightning" } = Object.fromEntries(new URL(req.url).searchParams); const substrate = new Substrate({ apiKey: Deno.env.get("SUBSTRATE_API_KEY") }); const Topic = z .object({ concepts: z.array(z.string()).min(2).max(2).describe("List of deeper concepts or ideas."), }) .describe("Deeper concepts or ideas related to a topic."); const c1 = new ComputeJSON({ prompt: `List deeper concepts or ideas related to: ${subject}`, json_schema: zodToJsonSchema(Topic), }); const prompt1 = new ComputeText({ prompt: sb.interpolate`generate a description of an image of ${ c1.future.json_object.get("concepts").at(0) }. Be concise and terse. Include details on the background, angle & framing, and style.`, }); const prompt2 = new ComputeText({ prompt: sb.interpolate`generate a description of an image of ${ c1.future.json_object.get("concepts").at(1) }. Be concise and terse. Include details on the background, angle & framing, and style.`, }); const caption1 = new ComputeText({ prompt: sb.interpolate`write a short paragraph relating ${ c1.future.json_object.get("concepts").at(0) } to ${subject}`, }); const caption2 = new ComputeText({ prompt: sb.interpolate`write a short paragraph relating ${ c1.future.json_object.get("concepts").at(1) } to ${subject}. It should be meaningfully different from the following content: ${caption1.future.text}`, }); const image1 = new GenerateImage({ prompt: prompt1.future.text }); const image2 = new GenerateImage({ prompt: prompt2.future.text }); const c2 = new ComputeJSON({ prompt: sb.interpolate`List deeper concepts or ideas related to: ${c1.future.json_object.get("concepts").at(0)} The items must NOT be in this list:[${c1.future.json_object.get("concepts").at(0)}, ${ c1.future.json_object.get("concepts").at(1) }]`, json_schema: zodToJsonSchema(Topic), }); const prompt3 = new ComputeText({ prompt: sb.interpolate`generate a description of an image of ${ c2.future.json_object.get("concepts").at(0) }. Be creative depicting abstract topics with extra detail. Include details on the background, angle & framing, and style.`, }); const prompt4 = new ComputeText({ prompt: sb.interpolate`generate a description of an image of ${ c2.future.json_object.get("concepts").at(1) }. Be creative depicting abstract topics with extra detail. Include details on the background, angle & framing, and style.`, }); const caption3 = new ComputeText({ prompt: sb.interpolate`write a short paragraph relating ${ c2.future.json_object.get("concepts").at(0) } to ${subject}. It should be meaningfully different from the following content: ${caption1.future.text} ${caption2.future.text}`, }); const caption4 = new ComputeText({ prompt: sb.interpolate`write a short paragraph relating ${ c2.future.json_object.get("concepts").at(1) } to ${subject}. It should be meaninfully different from the following content: ${caption1.future.text} ${caption2.future.text} ${caption3.future.text}`, }); const image3 = new GenerateImage({ prompt: prompt3.future.text }); const image4 = new GenerateImage({ prompt: prompt4.future.text }); const nodes = [image1, caption1, image2, caption2, image3, caption3, image4, caption4]; const stream = await substrate.stream(...nodes); return new Response( new ReadableStream({ async start(controller) { const pairs = [ { image: null, caption: null }, { image: null, caption: null }, { image: null, caption: null }, { image: null, caption: null }, ]; const outputPair = (pair, index) => { if (pair.image && pair.caption) { controller.enqueue(new TextEncoder().encode( `

${pair.image}

${pair.caption}

`, )); pairs[index] = { image: null, caption: null }; // Reset after output } }; for await (const event of stream) { if (event.object === "node.result") { const index = nodes.findIndex(node => node.id === event.node_id); if (index !== -1) { const pairIndex = Math.floor(index / 2); const isImage = index % 2 === 0; const content = event.data.image_uri ? `

` : `

${event.data.text}

`; if (isImage) { pairs[pairIndex].image = content; } else { pairs[pairIndex].caption = content; } outputPair(pairs[pairIndex], pairIndex); } } } // Output any remaining complete pairs pairs.forEach((pair, index) => outputPair(pair, index)); controller.close(); }, }), { headers: { "Content-Type": "text/html" } }, ); }

vprtwn