not conviced with the editor, confusing and slow to process on low bitrate connection

app.py

@@ -51,12 +51,13 @@ class BZHStableSignatureDemo(object):
         self.pipe.vae.load_state_dict(sd, strict=False)
 
         output = self.pipe(prompt, num_inference_steps=4, guidance_scale=0.0, output_type="pil")
-        return { "background": output.images[0], "layers": [], "composite": None }
+        return output.images[0] #{ "background": output.images[0], "layers": [], "composite": None }
 
-    def attack_detect(self, img_edit, jpeg_compression, downscale, saturation):
+    def attack_detect(self, img, jpeg_compression, downscale, saturation):
 
-        img = img_edit["composite"]
+        #img = img_edit["composite"]
         img = img.convert("RGB")
+        
         # attack
         if downscale != 1:
             size = img.size
@@ -116,7 +117,7 @@ def interface():
         with gr.Row():
             btn1 = gr.Button("Generate")
         with gr.Row():
-            watermarked_image = gr.ImageEditor(type="pil", width=512, height=512)
+            watermarked_image = gr.Image(type="pil", width=512, height=512)           
             with gr.Column():
                 downscale = gr.Slider(1, 3, value=1, step=0.1, label="Downscale ratio")
                 saturation = gr.Slider(0, 2, value=1, step=0.1, label="Color saturation")