ux updates

main.py

@@ -44,10 +44,30 @@ def generate_image(prompt):
     #image = pipe(prompt, num_inference_steps=5).images[0]
     #image = pipe(prompt).images[0]
         
-    print(image)    
-    image.save("static/ai.jpg")
-    image.save("static/ai.png")
+    image.save("ai.jpg")
+    
+
+    # Data to be written
+    assertion = {
+                "assertions": [
+                    {
+                    "label": "com.truepic.custom.ai",
+                    "data": {
+                        "model_name": model_id,
+                        "model_version": "1.0",
+                        "prompt": prompt
+                    }
+                    }
+                ]
+                }
+    
+    json_object = json.dumps(assertion, indent=4)
+    with open("assertion.json", "w") as outfile:
+        outfile.write(json_object)
+    
+    subprocess.check_output(['./truepic-sign', 'sign', 'ai.jpg', '--profile', 'demo', '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/output.jpg')])
 
+    return {"response": "success"}
 
 
 @app.get("/generate-picsum")

static/index.html

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