import inputHeader from "https://esm.town/v/substrate/inputHeader"; 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 handler(req: Request): Promise { if (new URL(req.url).pathname === "/robots.txt") { return new Response("User-agent: *\nDisallow: /"); } const input = new URL(req.url).searchParams.get("input") || "modernism"; // Substrate graph const substrate = new Substrate({ apiKey: Deno.env.get("SUBSTRATE_API_KEY_R") }); const Topic = z .object({ concepts: z.array(z.string()).min(2).max(2).describe( "List of advanced concepts related to a topic. Only one concept per item.", ), }) .describe("Advanced concepts related to a topic."); const c1 = new ComputeJSON({ prompt: `List advanced concepts related to: ${input}`, 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 creative depicting abstract topics using detailed abstract imagery. 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 creative depicting abstract topics using detailed abstract imagery. 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 ${input}`, }); const caption2 = new ComputeText({ prompt: sb.interpolate`write a short paragraph relating ${ c1.future.json_object.get("concepts").at(1) } to ${input}. It should be meaningfully different from the following content: ${caption1.future.text}`, }, { cache_age: 800 }); const image1 = new GenerateImage({ prompt: prompt1.future.text }); const image2 = new GenerateImage({ prompt: prompt2.future.text }); const c2 = new ComputeJSON({ prompt: sb.interpolate`List advanced concepts related to: ${c1.future.json_object.get("concepts").at(0)} They must be meaningfully different from those 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 using detailed abstract imagery. 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 using detailed abstract imagery. 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 ${input}. 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 ${input}. 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); // UI const capitalize = s => s.replace(/^\w/, c => c.toUpperCase()); return new Response( new ReadableStream({ async start(controller) { const pairs = [ { image: null, caption: null, title: null }, { image: null, caption: null, title: null }, { image: null, caption: null, title: null }, { image: null, caption: null, title: null }, ]; const outputPair = (pair, index) => { if (pair.image && pair.caption) { controller.enqueue(new TextEncoder().encode( `

${pair.image}

${capitalize(pair.title)}
${pair.caption}

`, )); pairs[index] = { image: null, caption: null, title: null }; // Reset after output } }; controller.enqueue(new TextEncoder().encode(inputHeader(input))); for await (const event of stream) { if (event.object === "node.result") { if (event.node_id === c1.id) { pairs[0].title = event.data.json_object.concepts[0]; pairs[1].title = event.data.json_object.concepts[1]; } else if (event.node_id === c2.id) { pairs[2].title = event.data.json_object.concepts[0]; pairs[3].title = event.data.json_object.concepts[1]; } else { 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); } } } } pairs.forEach((pair, index) => outputPair(pair, index)); controller.close(); }, }), { headers: { "Content-Type": "text/html" } }, ); }

liam