Update app.py

app.py

@@ -116,7 +116,7 @@ footer {display: none !important;}
 """
 js='''
 <script>
-window.postMessageToParent = function(event, source, value) {
+window.postMessageToParent = function(img, event, source, value) {
     // Construct the message object with the provided parameters
     console.log("post start",event, source, value);
     const message = {
@@ -128,6 +128,7 @@ window.postMessageToParent = function(event, source, value) {
     // Post the message to the parent window
     window.parent.postMessage(message, '*');
     console.log("post finish");
+    return img;
 }
 </script>
 '''
@@ -146,6 +147,6 @@ with gr.Blocks(css=css, head=js) as demo:
         processed_img = inference(img)
         return processed_img
 
-    process_button.click(update_status, inputs=image_input, outputs=image_input)#, js='''() => window.postMessageToParent("process_started", "deepnude_gan", "click_nude")'''
+    process_button.click(update_status, inputs=image_input, outputs=image_input)#, js='''(i) => window.postMessageToParent(i, "process_started", "deepnude_gan", "click_nude")'''
 demo.queue(max_size=10)
 demo.launch()