Update app.py

app.py

@@ -279,7 +279,7 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
         with gr.Group():
             gr.HTML(value=desc_html, elem_id='desc_html_code')
             result = gr.Gallery(
-              label="Result",  show_label=False, columns=1, rows=1, show_share_button=True,
+              label="Result",  show_label=False, columns=1, rows=1, show_share_button=True,elem_id=cfg["model_id"].replace("/", "-"),
               show_download_button=True,allow_preview=False,interactive=False, min_width=cfg.get("window_min_width", 340),height=360
             )
             with gr.Row(): 
@@ -328,16 +328,24 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
         else:
             default_image = None
         print(f"load_demo, urlparams={params},cover={default_image},domain={current_domain},iframe={iframe_parent_domain}")
+        session_data = {
+            "params": params,
+            "client_ip": request.client.host,
+            "refer": referer,
+            "host": current_domain
+        }
         if params.get("e", "0") == "1" or "nsfwais.io" in current_domain or "nsfwais.io" in iframe_parent_domain or "127.0.0.1" in current_domain or "127.0.0.1" in iframe_parent_domain:
             #update the image
             #bind events
-            return [Image.open(default_image)]
-        return []
-            
+            return [Image.open(default_image)], session_data
 
+
+        return [], session_data
+            
+    session_state = gr.State()
     result.change(fn=lambda x,y:None , inputs=[prompt,result], outputs=[], js=f'''(p,img)=>window.uploadImage(p, img,"process_finished","demo_hf_{cfg.get("name")}_card", "{cfg["model_id"]}")''')    
     run_button.click(generate, inputs=[prompt], outputs=[result],trigger_mode="once",js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
     random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''(p)=>window.g(p)''')
-    demo.load(fn=on_demo_load, inputs=[], outputs=[result], js='''()=>onDemoLoad()''')
+    demo.load(fn=on_demo_load, inputs=[], outputs=[result, session_state], js='''()=>onDemoLoad()''')
 if __name__ == "__main__":
     demo.queue(max_size=100).launch(show_api=False,show_error=False)