feat: add progressive EPUB chapter reader

app.py

@@ -120,7 +120,13 @@ def decode_reader_assets(raw: bytes) -> dict:
             mode = entry.get("m")
             if mode not in ("p", "e", "d", "h", "a", "v") or not isinstance(path, str) or not re.fullmatch(r"objects/[0-9a-f]{2}/[0-9a-f]{64}/(?:[a-z0-9-]+/)?(?:document\.pdf|book\.epub|document\.docx|document\.html|audio\.mp3|video\.mp4)", path):
                 continue
-            clean[key] = {"s": 2, "m": mode, "p": path}
+            chapter_manifest = entry.get("c")
+            fallback = entry.get("f")
+            if chapter_manifest is not None and (mode != "e" or not isinstance(chapter_manifest, str) or not re.fullmatch(r"objects/[0-9a-f]{2}/[0-9a-f]{64}/[a-z0-9-]+/chapter-manifest\.json", chapter_manifest)):
+                continue
+            if fallback is not None and (not isinstance(fallback, str) or not re.fullmatch(r"objects/[0-9a-f]{2}/[0-9a-f]{64}/[a-z0-9-]+/document\.pdf", fallback)):
+                continue
+            clean[key] = {"s": 2, "m": mode, "p": path, **({"c": chapter_manifest} if chapter_manifest else {}), **({"f": fallback} if fallback else {})}
         else:
             clean[key] = {"s": 4}
     return {"v": 1, "f": clean}
@@ -391,7 +397,7 @@ def validate_voiceofml_source_url(url: str) -> str:
 
 
 READER_ASSET_SOURCE_RE = re.compile(
-    r"^/datasets/vomebook/Reader-Assets/resolve/main/objects/[0-9a-f]{2}/[0-9a-f]{64}/(?:[a-z0-9-]+/)?(?:document\.pdf|book\.epub|document\.docx|document\.html|audio\.mp3|video\.mp4)$"
+    r"^/datasets/vomebook/Reader-Assets/resolve/main/objects/[0-9a-f]{2}/[0-9a-f]{64}/(?:[a-z0-9-]+/)?(?:chapter-manifest\.json|document\.pdf|book\.epub|document\.docx|document\.html|audio\.mp3|video\.mp4)$"
 )
 VOICEOFML_READER_SOURCE_RE = re.compile(r"^/datasets/VoiceOfML/[A-Za-z0-9._-]+/(?:resolve|raw)/main/.+$")
 
@@ -1418,8 +1424,10 @@ async def api_random_reader(repo: Optional[str] = Query(default=None)):
     rec["Path"] = rec.get("Path") or build_record_path_url(rec)
     asset = converted.get(index)
     if asset:
-        rec["ReaderLink"] = f"https://huggingface.co/datasets/vomebook/Reader-Assets/resolve/main/{asset['p']}"
-        rec["ReaderExtension"] = {"p": "pdf", "e": "epub", "d": "docx", "h": "html", "a": "audio", "v": "video"}[asset["m"]]
+        rec["ReaderLink"] = f"https://huggingface.co/datasets/vomebook/Reader-Assets/resolve/main/{asset.get('c') or asset['p']}"
+        rec["ReaderExtension"] = "epub-chapters" if asset.get("c") else {"p": "pdf", "e": "epub", "d": "docx", "h": "html", "a": "audio", "v": "video"}[asset["m"]]
+        if asset.get("f"):
+            rec["ReaderFallback"] = f"https://huggingface.co/datasets/vomebook/Reader-Assets/resolve/main/{asset['f']}"
         rec["DownloadLink"] = rec["Link"]
     return JSONResponse(rec)