updates

main.py

@@ -51,7 +51,7 @@ def generate_image(prompt, inference_steps, model):
 
     image.save(filename)
     
-    print(filename")
+    print(filename)
     #print(f"after save")
 
     
@@ -75,7 +75,7 @@ def generate_image(prompt, inference_steps, model):
     
     subprocess.check_output(['./truepic-sign', 'sign', filename, '--profile', 'demo', '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
 
-    return {"response": "success"}
+    return {"response": filename}
 
 
 @app.get("/generate-picsum")

static/index.html

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

static/script.js

@@ -27,7 +27,7 @@ const generateImage = async (text, inference_steps, model) => {
     const inferResponse = await fetch(`generate?prompt=${text}&inference_steps=${inference_steps}&model=${model}`);
     const inferJson = await inferResponse.json();
     
-    return inferJson.output;
+    return inferJson.response;
 };
 
 
@@ -40,7 +40,9 @@ textGenForm.addEventListener('submit', async (event) => {
     
     try {
       const resp = await generateImage(textGenInput.value, inferenceSteps.value, model.value);
-      document.getElementById("redirect-form").submit();
+      alert(resp);
+      document.getElementById("ouptput").src = resp;
+      // document.getElementById("redirect-form").submit();
     } catch (err) {
       console.error(err);
     }