update

main.py

@@ -8,6 +8,7 @@ import logging
 import urllib.request
 import os
 import json
+import uuid
 
 import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
@@ -43,8 +44,11 @@ def generate_image(prompt, inference_steps, model):
     #image = pipe(prompt, num_inference_steps=5).images[0]
     #image = pipe(prompt).images[0]
         
-    image.save("ai.jpg")
+    filename = str(uuid.uuid4()) + ".jpg"
+
+    image.save(filename)
     
+    print(filename)
 
     # Data to be written
     assertion = {
@@ -64,7 +68,7 @@ def generate_image(prompt, inference_steps, model):
     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')])
+    subprocess.check_output(['./truepic-sign', 'sign', filename, '--profile', 'demo', '--assertions', 'assertion.json', '--output', (os.getcwd() + '/static/output.jpg')])
 
     return {"response": "success"}