Try to fix from (previous) working version to load images

backend/runner/app.py

@@ -191,14 +191,33 @@ def list_work_images(work_id: str):
     full_work_id = f"W{work_id}"
     print(f"🔍 list_work_images called with work_id: {full_work_id}")
     work_dir = get_markdown_dir() / full_work_id
+    print(f"🔍 Work directory: {work_dir}")
+    print(f"🔍 Work directory exists: {work_dir.exists()}")
+    if work_dir.exists():
+        work_contents = list(work_dir.iterdir())
+        print(f"🔍 Work directory contents: {[f.name for f in work_contents]}")
     img_dir = work_dir / "images"
     if not img_dir.is_dir():
         print(f"❌ Images directory not found: {img_dir}")
-        return jsonify([])
-
-    files = sorted(
-        f for f in img_dir.iterdir() if f.suffix.lower() in (".jpg", ".jpeg", ".png")
-    )
+        # Fallback: look for images directly in the work directory
+        print(f"🔍 Checking for images directly in work directory...")
+        files = sorted(
+            f for f in work_dir.iterdir() if f.suffix.lower() in (".jpg", ".jpeg", ".png")
+        )
+        if files:
+            print(f"✅ Found {len(files)} images directly in work directory")
+            host = request.host_url.rstrip("/")
+            urls = [f"{host}/marker/{full_work_id}/{f.name}" for f in files]
+            return jsonify(urls)
+        else:
+            print(f"❌ No images found in work directory either")
+            return jsonify([])
+    else:
+        print(f"✅ Images directory found: {img_dir}")
+        files = sorted(
+            f for f in img_dir.iterdir() if f.suffix.lower() in (".jpg", ".jpeg", ".png")
+        )
+        print(f"🔍 Found {len(files)} images in images directory")
     host = request.host_url.rstrip("/")
     urls = [f"{host}/marker/{full_work_id}/{f.name}" for f in files]
     return jsonify(urls)
@@ -608,14 +627,24 @@ def heatmap():
 def serve_marker_image(work_id: str, filename: str):
     """
     Static file server for data/marker_output/<work_id>/images/<filename>
+    Falls back to work_id/<filename> if images directory doesn't exist
     """
     work_dir = get_markdown_dir() / work_id
     img_dir = work_dir / "images"
     img_path = img_dir / filename
-    mime, _ = guess_type(filename)
-    if not img_path.exists():
-        return jsonify({"error": "not-found"}), 404
-    return send_from_directory(img_dir, filename, mimetype=mime)
+    
+    # Try images directory first
+    if img_path.exists():
+        mime, _ = guess_type(filename)
+        return send_from_directory(img_dir, filename, mimetype=mime)
+    
+    # Fallback: try work directory directly
+    work_img_path = work_dir / filename
+    if work_img_path.exists():
+        mime, _ = guess_type(filename)
+        return send_from_directory(work_dir, filename, mimetype=mime)
+    
+    return jsonify({"error": "not-found"}), 404
 
 
 # --------------------------------------------------------------------------- #

backend/runner/config.py

@@ -443,6 +443,7 @@ def _download_images_robust(works_dir: Path, work_dirs: set, files: list, progre
             
             # Get list of image files for this work
             work_files = [f for f in files if f.startswith(f"works/{work_id}/images/")]
+            print(f"🔍 Work {work_id}: Found {len(work_files)} image files to download")
             
             downloaded = 0
             failed = 0