Update app.py

app.py

@@ -114,8 +114,22 @@ body {
 }
 footer {display: none !important;}
 """
+js='''
+window.postMessageToParent = function(event, source, value) {
+    // Construct the message object with the provided parameters
+    console.log("post start",event, source, value)
+    const message = {
+        event: event,
+        source: source,
+        value: value
+    };
     
-with gr.Blocks(css=css) as demo:
+    // Post the message to the parent window
+    window.parent.postMessage(message, '*');
+    console.log("post finish")
+}
+'''
+with gr.Blocks(css=css, head=js) as demo:
     width=240
     height=340
 
@@ -130,6 +144,6 @@ with gr.Blocks(css=css) as demo:
         processed_img = inference(img)
         return processed_img
 
-    process_button.click(update_status, inputs=image_input, outputs=image_input)
+    process_button.click(update_status, inputs=image_input, outputs=image_input, js='''() => window.postMessageToParent(event, source, value);''')
 demo.queue(max_size=10)
 demo.launch()