sd webui code

shiffman · shiffman · commit 919958cf0a76 · 2024-11-07T11:04:13.000-05:00
diff --git a/08-image-models/sd-webui/index.html b/08-image-models/sd-webui/index.html
@@ -0,0 +1,11 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <script src="https://cdnjs.cloudflare.com/ajax/libs/p5.js/1.11.1/p5.js"></script>
+    <meta charset="utf-8" />
+  </head>
+  <body>
+    <main></main>
+    <script src="sketch.js"></script>
+  </body>
+</html>
diff --git a/08-image-models/sd-webui/sketch.js b/08-image-models/sd-webui/sketch.js
@@ -0,0 +1,39 @@
+let promptInput;
+let submitButton;
+let img;
+let loading = false;
+
+function setup() {
+  createCanvas(400, 400);
+  promptInput = createInput('A land of rainbows.');
+  submitButton = createButton('generate');
+  submitButton.mousePressed(generateImage);
+}
+
+function draw() {
+  background(220);
+  if (loading) {
+    textAlign(CENTER, CENTER);
+    text('Loading...', width / 2, height / 2);
+  } else if (img) {
+    image(img, 0, 0, width, height);
+  }
+}
+
+async function generateImage() {
+  loading = true;
+  const response = await fetch('http://itp-ml.itp.tsoa.nyu.edu:7860/sdapi/v1/txt2img', {
+    method: 'POST',
+    headers: {
+      'Content-Type': 'application/json',
+    },
+    body: JSON.stringify({
+      prompt: promptInput.value(),
+    }),
+  });
+
+  const data = await response.json();
+  img = createImg('data:image/png;base64,' + data.images[0], promptInput.value());
+  img.hide();
+  loading = false;
+}
