only reveal specific fields

main.py

@@ -103,24 +103,20 @@ def generate(request: GenerationRequest):
         else:
             images = model.sample(request)
 
-        prompt = request.prompt
-        del request.prompt
-        if request.image is not None:
-            del request.image
-
         seed = request.seed
+        comment = json.dumps({"steps":request.steps,"sampler":request.sampler,"seed":request.seed,"strength":request.strength,"noise":request.noise,"scale":request.scale})
+
         images_encoded = []
         for x in range(len(images)):
             metadata = PngInfo()
             metadata.add_text("Title", "AI generated image")
-            metadata.add_text("Description", prompt)
+            metadata.add_text("Description", request.prompt)
             metadata.add_text("Software", "NovelAI")
             metadata.add_text("Source", "Stable Diffusion "+model_hash)
             if seed is not None:
                 request.seed = seed
                 seed += 1
-            print(request, json.dumps(request.toDict()))
-            metadata.add_text("Comment", json.dumps(request.toDict()))
+            metadata.add_text("Comment", comment)
             #image = simplejpeg.encode_jpeg(images[x], quality=95) 
             image = Image.fromarray(images[x])
             #save pillow image with bytesIO