update

main.py

@@ -8,49 +8,68 @@ import json
 import uuid
 
 import torch
-from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler, EulerDiscreteScheduler
+from diffusers import (
+    StableDiffusionPipeline,
+    DPMSolverMultistepScheduler,
+    EulerDiscreteScheduler,
+)
+
+app = FastAPI()
 
-app = FastAPI() 
 
 @app.get("/generate")
 def generate_image(prompt, model):
     torch.cuda.empty_cache()
 
-    pipeline = StableDiffusionPipeline.from_pretrained(str(model), torch_dtype=torch.float16)
+    pipeline = StableDiffusionPipeline.from_pretrained(
+        str(model), torch_dtype=torch.float16
+    )
     pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
     pipeline = pipeline.to("cuda")
- 
+
     image = pipeline(prompt, num_inference_steps=50, height=512, width=512).images[0]
 
     filename = str(uuid.uuid4()) + ".jpg"
     image.save(filename)
-    
+
     assertion = {
-                "assertions": [
-                        {
-                        "label": "com.truepic.custom.ai",
-                        "data": {
-                            "model_name": model,
-                            "model_version": "1.0",
-                            "prompt": prompt
-                        }
-                        }
-                    ]
-                }
+        "assertions": [
+            {
+                "label": "com.truepic.custom.ai",
+                "data": {"model_name": model, "model_version": "1.0", "prompt": prompt},
+            }
+        ]
+    }
 
     json_object = json.dumps(assertion)
-#    with open("assertion.json", "w") as outfile:
-#       outfile.write(json_object)
 
-    subprocess.check_output(['./truepic-sign', 'init', 'file-system', '--api-key', os.environ.get("api_key")])
-#    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions-file', 'assertion.json', '--output', (os.getcwd() + '/static/' + filename)])
-    subprocess.check_output(['./truepic-sign', 'sign', filename, '--assertions', json_object, '--output', (os.getcwd() + '/static/' + filename)])
-    
+    subprocess.check_output(
+        [
+            "./truepic-sign",
+            "init",
+            "file-system",
+            "--api-key",
+            os.environ.get("api_key"),
+        ]
+    )
+    subprocess.check_output(
+        [
+            "./truepic-sign",
+            "sign",
+            filename,
+            "--assertions",
+            json_object,
+            "--output",
+            (os.getcwd() + "/static/" + filename),
+        ]
+    )
+
     return {"response": filename}
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
+
 @app.get("/")
 def index() -> FileResponse:
     return FileResponse(path="/app/static/index.html", media_type="text/html")

static/index.html

@@ -35,6 +35,10 @@
             </div>
             <a href="#" target="_blank" id="download-link"><img src="images/download.png" id="download" /></a>
           </div>
+          <div class="parameters">
+            Image prompt: “<span class="prompt">dog on a swing</span>”<br/>
+            Model: <span class="model">stabilityai/stable-diffusion-xl-base-1.0</span>
+          </div>
         </div>
         <section class="verification-details">
             <nav>

static/script.js

@@ -11,6 +11,10 @@ const certificateNav = document.getElementById('certificate-nav');
 const verificationOutput = document.getElementById('verification-output');
 const certificateOutput = document.getElementById('certificate-output');
 const verificationDetails = document.querySelector('.verification-details');
+const parameters = document.querySelector('.parameters');
+const modelParam = document.querySelector('.parameters .model');
+const promptParam = document.querySelector('.parameters .prompt');
+
 
 
 [textGenInput, model].forEach(item => {
@@ -66,6 +70,10 @@ textGenForm.addEventListener('submit', async (event) => {
       downloadLink.style.display = "block";
       downloadLink.href = path;
       downloadLink.download = resp;
+
+      modelParam.innerHTML = model.value;
+      promptParam.innerHTML = textGenInput.value;
+      parameters.style.display = "block";
     } catch (err) {
       console.error(err);
     }

static/style.css

@@ -52,6 +52,13 @@ textarea {
   border-bottom: 1px solid rgba(227, 234, 240, 1);
 }
 
+#column-one .container .parameters {
+  display: none;
+  color: rgba(86, 104, 122, 1);
+  font-size: 12px;
+  padding-bottom: 2rem;
+}
+
 #logo {
   width: 156px;
   padding: 2rem 10rem;