Update app.py

app.py

@@ -273,12 +273,27 @@ def generate(p, progress=gr.Progress(track_tqdm=True)):
     image_paths = [i[1] for i in images]
     print(prompt_str, image_paths)
     return [i[0] for i in images]
+default_image = cfg.get("cover_path", None)
 
+if default_image:
+    if isinstance(default_image, list):
+        # Filter out non-existent paths
+        existing_images = [img for img in default_image if os.path.exists(img)]
+        #print(f"found cover files: {existing_images}")
+        if existing_images:
+            default_image = existing_images[int(time.time()*1000)%len(existing_images)]
+        else:
+            default_image = None
+    elif not os.path.exists(default_image):
+        print(f"cover file not existed, {default_image}")
+        default_image = None
+else:
+    default_image = None
 with gr.Blocks(css=css,head=js,fill_height=True) as demo:
     with gr.Row(equal_height=False):
         with gr.Group():
             gr.HTML(value=desc_html, elem_id='desc_html_code')
-            result = gr.Gallery(
+            result = gr.Gallery(value=default_image,
               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
             )
@@ -311,22 +326,7 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
                 iframe_parent_domain = "Unable to parse referer"
 
         params = dict(request.query_params)
-        default_image = cfg.get("cover_path", None)
-        
-        if default_image:
-            if isinstance(default_image, list):
-                # Filter out non-existent paths
-                existing_images = [img for img in default_image if os.path.exists(img)]
-                #print(f"found cover files: {existing_images}")
-                if existing_images:
-                    default_image = existing_images[int(time.time()*1000)%len(existing_images)]
-                else:
-                    default_image = None
-            elif not os.path.exists(default_image):
-                print(f"cover file not existed, {default_image}")
-                default_image = None
-        else:
-            default_image = None
+
         print(f"load_demo, urlparams={params},cover={default_image},domain={current_domain},iframe={iframe_parent_domain}")
         session_data = {
             "params": params,
@@ -337,15 +337,16 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
         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)], session_data
-
+            #return [Image.open(default_image)], session_data
+            return session_data
 
-        return [], session_data
+        return 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, session_state], js='''()=>onDemoLoad()''')
+    demo.load(fn=on_demo_load, inputs=[], outputs=[session_state], js='''()=>onDemoLoad()''')
 if __name__ == "__main__":
     demo.queue(max_size=100).launch(show_api=False,show_error=False)