Update main.py

main.py

@@ -242,6 +242,13 @@ def ensure_case_generated(case_id: str) -> Dict[str, Any]:
     if existing_public:
         return existing_public
 
+    # Make sure we have a minimum pool of cached images ready
+    try:
+        ensure_minimum_ia_pool()
+    except Exception as e:
+        print("Bootstrap warning:", e)
+
+
     # Ensure we have at least some IA records; if not, auto-ingest a default set (one page)
     pool = ia_pool_ref().get() or {}
     if not pool:
@@ -272,7 +279,9 @@ def ensure_case_generated(case_id: str) -> Dict[str, Any]:
     style_period = "sourced from Internet Archive; museum catalog reproduction"
 
     # Download authentic image
-    auth_img = download_image_to_pil(ia_item["download_url"])
+    source_url = ia_item.get("storage_url") or ia_item["download_url"]
+    auth_img = download_image_to_pil(source_url)
+
 
     images_urls: List[str] = []
     signature_crops: List[str] = []
@@ -712,4 +721,11 @@ def leaderboard_daily():
 # 6) MAIN
 # -----------------------------------------------------------------------------
 if __name__ == "__main__":
+    # Optional: pre-warm pool on boot so you’re ready before first request
+    if os.environ.get("BOOTSTRAP_IA", "1") == "1":
+        print("Bootstrapping Internet Archive pool...")
+        stats = ensure_minimum_ia_pool()
+        print("Bootstrap complete:", stats)
+
     app.run(host="0.0.0.0", port=int(os.environ.get("PORT", "7860")), debug=True)
+