Update app.py

app.py

@@ -119,7 +119,7 @@ def main_ui():
     gen_btn.click(generate, paras + [others[0], others[5]], [image, info])
     others[2].click(lambda o, s: o if len(s) == 0 else s['parameters']['seed'], inputs=[paras[3], info], outputs=paras[3])
     others[3].click(lambda i: i, inputs=image, outputs=paras[14])
-    others[4].click(lambda i: i, inputs=image, outputs=paras[17])
+    others[4].click(lambda i, d: {'background': i, 'layers':d['layers'], 'composite': i}, inputs=[image, paras[17]], outputs=paras[17])
     return page, paras[:14]