updates

static/index.html

@@ -12,7 +12,12 @@
     <main>
       <section id="text-gen">
         <h1>Text to Image with Signed Attribution</h1>
-        <truepic-display><img src="/output.jpg" id="output" /></truepic-display>
+        <truepic-display><img src="/output.jpg" id="outputImg" /></truepic-display>
+
+        <div id="outputClass">
+
+
+        </div>
         <form class="text-gen-form" style="padding:40px 0;">
           <label for="text-gen-input">Enter prompt: </label>
           <input

static/script.js

@@ -41,8 +41,16 @@ textGenForm.addEventListener('submit', async (event) => {
     try {
       const resp = await generateImage(textGenInput.value, inferenceSteps.value, model.value);
       const path = "/" + resp;
-      alert(path);
-      document.getElementById("outputImg").src = path;
+      
+      var resultsContainer = document.getElementById('outputClass');
+      var truepicDisplay = document.createElement('truepic-display');
+      var truepic = document.createElement('img');
+      truepic.src = path;
+
+      truepicDisplay.appendChild(truepic);
+
+      resultsContainer.appendChild(truepicDisplay);
+
       // document.getElementById("redirect-form").submit();
     } catch (err) {
       console.error(err);