Update app.py

app.py

@@ -18,6 +18,7 @@ import requests
 import torch
 from diffusers import DiffusionPipeline
 import importlib
+from urllib.parse import urlparse
 
 random.seed(time.time())
 MAX_SEED = 12211231
@@ -211,7 +212,8 @@ def generate(prompt, progress=gr.Progress(track_tqdm=True)):
 
 with gr.Blocks(css=css,head=js,fill_height=True) as demo:
     with gr.Row(equal_height=False):
-        with gr.Group():                
+        with gr.Group():
+            
             result = gr.Gallery(
               label="Result",  show_label=False, columns=1, rows=1, show_share_button=True,
               show_download_button=True,allow_preview=True,interactive=False, min_width=cfg.get("window_min_width", 340),height=360
@@ -230,6 +232,19 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
             run_button = gr.Button( "GO!", scale=1, min_width=20, variant="primary",icon="https://huggingface.co/spaces/nsfwalex/sd_card/resolve/main/hot.svg")
             
     def on_demo_load(request: gr.Request):
+        current_domain = request.request.headers.get("Host", "")
+    
+        # Get the potential iframe parent domain from the Referer header
+        referer = request.request.headers.get("Referer", "")
+        iframe_parent_domain = ""
+    
+        if referer:
+            try:
+                parsed_referer = urlparse(referer)
+                iframe_parent_domain = parsed_referer.netloc
+            except:
+                iframe_parent_domain = "Unable to parse referer"
+
         params = dict(request.query_params)
         default_image = cfg.get("cover_path", None)
         
@@ -247,7 +262,7 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
                 default_image = None
         else:
             default_image = None
-        print("load_demo, url params", params, "image", default_image)#, "domain", domain, "iframe", iframe_domain)
+        print("load_demo, url params", params, "image", default_image, "domain", current_domain, "iframe", iframe_parent_domain)
         if params.get("e", "0") == "1":
             #update the image
             #bind events