Update app.py

app.py

@@ -133,7 +133,19 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
 
     print(nsfw_prediction)
 
-    return image_paths, {item['label']: round(item['score'], 3) for item in nsfw_prediction}
+    image = images[0].convert("RGBA")
+    image = image.resize((width, height))
+    json_data = []
+
+    for x in range(image.width):
+        json_data.append([])
+        for y in range(image.height):
+            r, g, b, a = image.getpixel((x, y))
+            json_data[x].append([r, g, b, a])
+
+    json_string = json.dumps(json_data)
+
+    return image_paths, {item['label']: round(item['score'], 3) for item in nsfw_prediction}, json_string
 
 def cloud():
     print("[CLOUD] | Space maintained.")
@@ -160,6 +172,7 @@ with gr.Blocks(css=css) as main:
     with gr.Column():
         output = gr.Gallery(columns=1, label="Image")
         output_2 = gr.Label()
+        output_3 = gr.Textbox(lines=1, value="", label="Buffer")
             
     submit.click(generate, inputs=[input, filter_input, negative_input, model, height, width, steps, guidance, number, seed], outputs=[output, output_2], queue=False)
     maintain.click(cloud, inputs=[], outputs=[], queue=False)