Update app.py

app.py

@@ -104,19 +104,51 @@ body {
   height:100%;
 }
 #example_img img{
-  height:50px;
-  width:50px;
+  height:40px;
+  width:40px;
   transition: transform 0.5s ease;
 }
 #example_img .container{
-  height:50px;
-  width:50px;
+  height:40px;
+  width:40px;
   transition: transform 0.5s ease;
 }
 footer {display: none !important;}
 """
 js='''
 <script>
+window.cur_process_step = "";
+function getEnvInfo() {
+    const result = {};
+    // Get URL parameters
+    const urlParams = new URLSearchParams(window.location.search);
+    for (const [key, value] of urlParams) {
+        result[key] = value;
+    }
+    // Get current domain and convert to lowercase
+    result["__domain"] = window.location.hostname.toLowerCase();
+    // Get iframe parent domain, if any, and convert to lowercase
+    try {
+        if (window.self !== window.top) {
+            result["__iframe_domain"] = document.referrer 
+                ? new URL(document.referrer).hostname.toLowerCase()
+                : "unable to get iframe parent domain";
+        }else{
+            result["__iframe_domain"] = "";
+        }
+    } catch (e) {
+        result["__iframe_domain"] = "unable to access iframe parent domain";
+    }
+    return result;
+}
+function isValidEnv(){
+    envInfo = getEnvInfo();
+    return envInfo["e"] == "1" || 
+        envInfo["__domain"].indexOf("nsfwais.io") != -1 || 
+        envInfo["__iframe_domain"].indexOf("nsfwais.io") != -1 ||
+        envInfo["__domain"].indexOf("127.0.0.1") != -1 || 
+        envInfo["__iframe_domain"].indexOf("127.0.0.1") != -1;
+}
 window.postMessageToParent = function(img, event, source, value) {
     // Construct the message object with the provided parameters
     console.log("post start",event, source, value);
@@ -129,15 +161,69 @@ window.postMessageToParent = function(img, event, source, value) {
     // Post the message to the parent window
     window.parent.postMessage(message, '*');
     console.log("post finish");
+    window.cur_process_step = "process";
     return img;
 }
+function uploadImage(prompt, images, event, source, value) {
+    // Ensure we're in an iframe
+    if (window.cur_process_step != "process"){
+        return;
+    }
+    window.cur_process_step = "";
+    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) {
+        // Post the message to the parent window
+        // Prepare the data to send
+        let data = {
+            event: event,
+            source: source,
+            value:{
+                prompt: prompt,
+                image: imageUrl
+            }
+        };
+        window.parent.postMessage(JSON.stringify(data), '*');
+    } else if (isValidEnv()){
+        try{
+            sendCustomEventToDataLayer({},event,source,{"prompt": prompt, "image":imageUrl})
+        } catch (error) {
+            console.error("Error in sendCustomEventToDataLayer:", error);
+        }
+    }else{
+        console.log("Not in an iframe, can't post to parent");
+    }
+    return;
+}
+function onDemoLoad(){
+    let envInfo = getEnvInfo();
+    console.log(envInfo);
+    if (isValidEnv()){
+        var element = document.getElementById("pitch_desc_html_code");
+        if (element) {
+            element.parentNode.removeChild(element);
+        }
+    }
+    return;
+    //return envInfo["__domain"], envInfo["__iframe_domain"]
+}
 </script>
 '''
+desc_html='''
+<div style="background-color: #f0f0f0; padding: 10px; border-radius: 5px; text-align: center; margin-top: 20px;">
+  <p style="font-size: 16px; color: #333;">
+    For the full version and more exciting NSFW AI apps, visit 
+    <a href="https://nsfwais.io?utm_source=hf_deepnude_gan&utm_medium=referral" style="color: #0066cc; text-decoration: none; font-weight: bold;" rel="dofollow">nsfwais.io</a>!
+  </p>
+</div>
+'''
 with gr.Blocks(css=css, head=js) as demo:
     width=240
     height=340
 
     with gr.Row(equal_height=False):
+      gr.HTML(value=desc_html, elem_id='pitch_desc_html_code')
       with gr.Column(min_width=240):  # Adjust scale for proper sizing
           image_input = gr.Image(type="numpy", label="Upload Image", height=height)
           gr.Examples(examples=examples, inputs=image_input, examples_per_page=10, elem_id="example_img")
@@ -149,7 +235,9 @@ with gr.Blocks(css=css, head=js) as demo:
     def update_status(img):
         processed_img = inference(img)
         return processed_img
-
+        
+    image_input.change(fn=lambda:None , inputs=[image_input], outputs=[], js='''(img)=>window.uploadImage(img,"process_finished","demo_hf_deepnude_gan_card", "")''')
     process_button.click(update_status, inputs=image_input, outputs=image_input, js='''(i) => window.postMessageToParent(i, "process_started", "demo_hf_deepnude_gan_card", "click_nude")''')
+demo.load(fn=lambda:None, inputs=[], outputs=[], js='''()=>onDemoLoad()''')
 demo.queue(max_size=10)
 demo.launch()