1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
export default function renderHTML(stream: any) {
let { readable, writable } = new TransformStream();
let writer = writable.getWriter();
const textEncoder = new TextEncoder();
let htmlStarted = false;
let logLines = [];
new Promise(async () => {
for await (const event of stream) {
if (event.object === "node.result") {
const logLine = `<pre>Generated ${event.node_id}</pre>`;
logLines.push(logLine);
if (!htmlStarted) {
writer.write(textEncoder.encode(logLine));
}
}
if (event.object === "node.delta") {
if (!htmlStarted) {
htmlStarted = true;
writer.write(textEncoder.encode("<script>document.body.innerHTML = \"\";</script>"));
}
writer.write(textEncoder.encode(event.data.text || ""));
}
}
writer.close();
});
return new Response(readable, {
headers: {
"Content-Type": "text/html",
},
});
}