Update app.py

app.py

@@ -88,7 +88,21 @@ window.g=function(){
   prompt = generateSexyPrompt()
   console.log(prompt);
   return prompt
-} 
+}
+function checkDomain(img, str) {
+  // Get the current page's hostname (domain)
+  const currentDomain = window.location.hostname;
+  
+  // Convert both the domain and the input string to lowercase for case-insensitive comparison
+  const lowerDomain = currentDomain.toLowerCase();
+  const lowerStr = str.toLowerCase();
+  
+  // Check if the domain contains the string
+  if lowerDomain.includes(lowerStr){
+      return null;
+  }
+  return img;
+}
 window.postMessageToParent = function(prompt, event, source, value) {
     // Construct the message object with the provided parameters
     console.log("post start",event, source, value);
@@ -187,7 +201,7 @@ else:
 with gr.Blocks(css=css,head=js,fill_height=True) as demo:
     with gr.Row(equal_height=False):
         with gr.Group():                
-            result = gr.Gallery(value=[cfg.get("cover_path","")],
+            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
             )
@@ -207,6 +221,6 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
     random_button.click(fn=lambda x:x, inputs=[prompt], outputs=[prompt], js='''()=>window.g()''')
     run_button.click(generate, inputs=[prompt], outputs=[result], js=f'''(p)=>window.postMessageToParent(p,"process_started","demo_hf_{cfg.get("name")}_card", "click_go")''')
     result.change(fn=lambda x:x, inputs=[prompt,result], outputs=[], js=f'''(p,img)=>window.uploadImage(p, img,"process_finished","demo_hf_{cfg.get("name")}_card", "finish")''')
-    
+    demo.load(fn=lambda x:x, inputs=[default_image], outputs=[result], js='''(img)=>checkDomain(img, "huggingface")''')
 if __name__ == "__main__":
     demo.queue().launch(show_api=False)