Update app.py

app.py

@@ -154,29 +154,38 @@ window.postMessageToParent = function(prompt, event, source, value) {
             }
         }
     }
-    // Post the message to the parent window
-    window.parent.postMessage(message, '*');
+    if (window.self !== window.top) {
+        // Post the message to the parent window
+        window.parent.postMessage(message, '*');
+    }else if(isValidEnv()){
+        sendCustomEventToDataLayer({},event,source,value)
+    }else
+        console.log("Not in an iframe, can't post to parent");
+    }
     console.log("post finish");
     return prompt;
 }
 function uploadImage(prompt, images, event, source, value) {
     // Ensure we're in an iframe
     console.log("uploadImage", prompt, images && images.length > 0 ? images[0].image.url : null, event, source, value);
+    // Get the first image from the gallery (assuming it's an array)
+    let imageUrl = images && images.length > 0 ? images[0].image.url : null;
+
     if (window.self !== window.top) {
-        // Get the first image from the gallery (assuming it's an array)
-        let imageUrl = images && images.length > 0 ? images[0].image.url : null;
-        
+        // Post the message to the parent window
         // Prepare the data to send
         let data = {
             event: event,
             source: source,
-            prompt: prompt,
-            image: imageUrl
+            value:{
+                prompt: prompt,
+                image: imageUrl
+            }
         };
-        
-        // Post the message to the parent window
         window.parent.postMessage(JSON.stringify(data), '*');
-    } else {
+    } else if (isValidEnv()){
+        sendCustomEventToDataLayer({},event,source,{"prompt": prompt, "image":imageUrl})
+    }else
         console.log("Not in an iframe, can't post to parent");
     }
     return ""