Update app.py

app.py

@@ -206,27 +206,9 @@ def generate(prompt, progress=gr.Progress(track_tqdm=True)):
     ).images
     
     image_paths = [save_image(img) for img in images]
-    print(prompt_str, image_paths)
+    print(prompt_str, image_paths, [img.url for img in images])
     return image_paths
 
-
-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 = random.choice(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():                
@@ -249,6 +231,22 @@ with gr.Blocks(css=css,head=js,fill_height=True) as demo:
             
     def on_demo_load(request: gr.Request):
         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 = random.choice(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("load_demo, url params", params, "image", default_image)#, "domain", domain, "iframe", iframe_domain)
         if params.get("e", "0") == "1":
             #update the image