Update app.py

app.py

@@ -169,7 +169,20 @@ def generate(prompt, progress=gr.Progress(track_tqdm=True)):
     return image_paths
 
 
-default_image = cfg["cover_path"] if cfg.get("cover_path", None) and os.path.exists(cfg.get("cover_path", None)) else None
+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)]
+        if existing_images:
+            default_image = random.choice(existing_images)
+        else:
+            default_image = None
+    elif not os.path.exists(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):