diff --git a/__init__.py b/__init__.py
index 571d1d5..1474cdd 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,13 +1,15 @@
-from .nodes import OmniVoiceModelLoader, OmniVoiceGenerate
+from .nodes import OmniVoiceModelLoader, OmniVoiceGenerate, OmniVoiceEpubLoader
NODE_CLASS_MAPPINGS = {
"OmniVoiceModelLoader": OmniVoiceModelLoader,
"OmniVoiceGenerate": OmniVoiceGenerate,
+ "OmniVoiceEpubLoader": OmniVoiceEpubLoader,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"OmniVoiceModelLoader": "OmniVoice Model Loader",
"OmniVoiceGenerate": "OmniVoice Generate",
+ "OmniVoiceEpubLoader": "OmniVoice EPUB Loader",
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
diff --git a/install.py b/install.py
index 1f2fa24..de6bdb1 100644
--- a/install.py
+++ b/install.py
@@ -25,6 +25,7 @@ pip(
"pydub",
"soundfile",
"numpy",
+ "beautifulsoup4",
)
print("\n[ComfyUI-Omnivoice] Installation complete.")
diff --git a/nodes/__init__.py b/nodes/__init__.py
index f17a35a..fcc139b 100644
--- a/nodes/__init__.py
+++ b/nodes/__init__.py
@@ -1,4 +1,5 @@
from .loader import OmniVoiceModelLoader
from .generator import OmniVoiceGenerate
+from .epub_loader import OmniVoiceEpubLoader
-__all__ = ["OmniVoiceModelLoader", "OmniVoiceGenerate"]
+__all__ = ["OmniVoiceModelLoader", "OmniVoiceGenerate", "OmniVoiceEpubLoader"]
diff --git a/nodes/epub_loader.py b/nodes/epub_loader.py
new file mode 100644
index 0000000..e9485f1
--- /dev/null
+++ b/nodes/epub_loader.py
@@ -0,0 +1,109 @@
+import zipfile
+import io
+import xml.etree.ElementTree as ET
+from bs4 import BeautifulSoup
+
+
+def _local(tag):
+ """Strip XML namespace prefix, return local tag name."""
+ return tag.split('}')[-1]
+
+
+def _extract_chapters(epub_path):
+ """Parse EPUB and return list of {"title": str|None, "text": str}."""
+ chapters = []
+ with zipfile.ZipFile(epub_path, 'r') as zf:
+ # 1. Find OPF path from container.xml
+ container = ET.fromstring(zf.read('META-INF/container.xml'))
+ rootfile = next(
+ el for el in container.iter()
+ if _local(el.tag) == 'rootfile'
+ )
+ opf_path = rootfile.attrib['full-path']
+ opf_dir = opf_path.rsplit('/', 1)[0] + '/' if '/' in opf_path else ''
+
+ # 2. Parse OPF: build manifest and spine
+ opf = ET.fromstring(zf.read(opf_path))
+ manifest = {
+ el.attrib['id']: el.attrib['href']
+ for el in opf.iter()
+ if _local(el.tag) == 'item'
+ and 'xhtml' in el.attrib.get('media-type', '')
+ }
+ spine = [
+ el.attrib['idref']
+ for el in opf.iter()
+ if _local(el.tag) == 'itemref'
+ ]
+
+ # 3. Extract text from each chapter XHTML
+ for idref in spine:
+ href = manifest.get(idref)
+ if href is None:
+ continue
+ xhtml = zf.read(opf_dir + href).decode('utf-8', errors='replace')
+ soup = BeautifulSoup(xhtml, 'html.parser')
+ for tag in soup(['script', 'style']):
+ tag.decompose()
+ # Title: → <h1/h2/h3> → None
+ title = None
+ if soup.title and soup.title.string:
+ title = soup.title.string.strip()
+ if not title:
+ for hn in ['h1', 'h2', 'h3']:
+ tag = soup.find(hn)
+ if tag:
+ title = tag.get_text(strip=True)
+ break
+ text = soup.get_text(separator=' ', strip=True)
+ chapters.append({"title": title, "text": text})
+
+ return chapters
+
+
+class OmniVoiceEpubLoader:
+ @classmethod
+ def INPUT_TYPES(cls):
+ return {
+ "required": {
+ "epub_path": ("STRING", {
+ "default": "",
+ "tooltip": "Absolute path to the .epub file to load.",
+ }),
+ "chapter_start": ("INT", {
+ "default": 1, "min": 1, "max": 9999, "step": 1,
+ "tooltip": "First chapter to include (1-indexed). Clamped to valid range automatically.",
+ }),
+ "chapter_end": ("INT", {
+ "default": 1, "min": 1, "max": 9999, "step": 1,
+ "tooltip": "Last chapter to include (1-indexed, inclusive). Clamped automatically. If less than chapter_start, set to chapter_start.",
+ }),
+ },
+ }
+
+ RETURN_TYPES = ("STRING", "STRING")
+ RETURN_NAMES = ("text", "chapter_list")
+ FUNCTION = "load_epub"
+ CATEGORY = "OmniVoice"
+
+ def load_epub(self, epub_path, chapter_start, chapter_end):
+ chapters = _extract_chapters(epub_path)
+ n = len(chapters)
+
+ if n == 0:
+ return ("", "")
+
+ start = max(1, min(chapter_start, n))
+ end = max(start, min(chapter_end, n))
+
+ # chapter_list: all chapters regardless of selection
+ chapter_list = "\n".join(
+ f"{i}. {ch['title'] if ch['title'] else f'Chapter {i}'}"
+ for i, ch in enumerate(chapters, 1)
+ )
+
+ # text: selected range joined by delimiter
+ selected = chapters[start - 1 : end]
+ text = "\n\n---\n\n".join(ch["text"] for ch in selected)
+
+ return (text, chapter_list)
diff --git a/nodes/generator.py b/nodes/generator.py
index 013b1f4..a93132a 100644
--- a/nodes/generator.py
+++ b/nodes/generator.py
@@ -1,7 +1,7 @@
import tempfile
import os
import torch
-import torchaudio
+import soundfile as sf
class OmniVoiceGenerate:
@@ -109,7 +109,9 @@ class OmniVoiceGenerate:
tmp.close()
try:
ref_waveform = ref_audio["waveform"].squeeze(0).cpu() # (channels, samples)
- torchaudio.save(tmp_path, ref_waveform, int(ref_audio["sample_rate"]))
+ audio_np = ref_waveform.numpy()
+ # soundfile expects (samples,) for mono or (samples, channels) for multi-channel
+ sf.write(tmp_path, audio_np[0] if audio_np.shape[0] == 1 else audio_np.T, int(ref_audio["sample_rate"]))
kwargs["ref_audio"] = tmp_path
if ref_text:
kwargs["ref_text"] = ref_text
diff --git a/requirements.txt b/requirements.txt
index 59d015d..b505f76 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
# Dependencies are managed by install.py to avoid overwriting ComfyUI's torch.
# omnivoice pins torch==2.8.* (CUDA 12.8) which would break ComfyUI's torch build.
# Do not add omnivoice here — install.py handles it with --no-deps.
+# EPUB parsing (OmniVoiceEpubLoader) requires beautifulsoup4 (installed by install.py).
diff --git a/tests/test_epub_loader.py b/tests/test_epub_loader.py
new file mode 100644
index 0000000..8b1ead5
--- /dev/null
+++ b/tests/test_epub_loader.py
@@ -0,0 +1,132 @@
+import io
+import zipfile
+from unittest.mock import patch
+import pytest
+from nodes.epub_loader import OmniVoiceEpubLoader
+
+# Capture the real ZipFile class BEFORE any patching so epub_opener can use it
+# without hitting the patch and causing infinite recursion.
+_RealZipFile = zipfile.ZipFile
+
+_CONTAINER_XML = """<?xml version="1.0"?>
+<container xmlns="urn:oasis:schemas:container" version="1.0">
+ <rootfiles>
+ <rootfile full-path="OEBPS/content.opf" media-type="application/oebps-package+xml"/>
+ </rootfiles>
+</container>"""
+
+
+def make_fake_epub(chapters):
+ """chapters: list of (title, body_html). Returns EPUB bytes."""
+ buf = io.BytesIO()
+ with zipfile.ZipFile(buf, 'w') as z:
+ z.writestr('mimetype', 'application/epub+zip')
+ z.writestr('META-INF/container.xml', _CONTAINER_XML)
+ items = "\n".join(
+ f'<item id="ch{i}" href="ch{i}.xhtml" media-type="application/xhtml+xml"/>'
+ for i in range(len(chapters))
+ )
+ spine = "\n".join(f'<itemref idref="ch{i}"/>' for i in range(len(chapters)))
+ opf = f"""<?xml version="1.0"?>
+<package xmlns="http://www.idpf.org/2007/opf">
+ <manifest>{items}</manifest>
+ <spine>{spine}</spine>
+</package>"""
+ z.writestr('OEBPS/content.opf', opf)
+ for i, (title, body) in enumerate(chapters):
+ z.writestr(f'OEBPS/ch{i}.xhtml',
+ f'<html><head><title>{title}{title}
{body}