mirror of
https://github.com/denizsafak/abogen.git
synced 2026-07-18 05:40:26 +02:00
JB
41f56a8491
- Added speaker analysis module to infer speaker identities from text chunks. - Introduced SpeakerGuess and SpeakerAnalysis data classes for managing speaker data. - Developed functions for analyzing speaker occurrences and confidence levels. - Created EPUB 3 exporter to generate EPUB packages with synchronized narration and media overlays. - Implemented configurable chunking options for TTS synthesis and EPUB alignment. - Enhanced JavaScript for speaker preview functionality in the web interface. - Added comprehensive tests for chunking and EPUB exporting features. - Documented upgrade plan for transitioning to EPUB 3 with multi-speaker support.
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import time
|
|
from abogen.web.service import JobStatus, build_service
|
|
|
|
|
|
def test_service_processes_job(tmp_path):
|
|
uploads = tmp_path / "uploads"
|
|
outputs = tmp_path / "outputs"
|
|
uploads.mkdir()
|
|
outputs.mkdir()
|
|
|
|
source = uploads / "sample.txt"
|
|
payload = "hello world"
|
|
source.write_text(payload, encoding="utf-8")
|
|
|
|
runner_invocations: list[str] = []
|
|
|
|
def runner(job):
|
|
runner_invocations.append(job.id)
|
|
job.add_log("processing")
|
|
job.progress = 1.0
|
|
job.processed_characters = job.total_characters or len(payload)
|
|
job.result.audio_path = outputs / f"{job.id}.wav"
|
|
|
|
service = build_service(
|
|
runner=runner,
|
|
output_root=outputs,
|
|
uploads_root=uploads,
|
|
)
|
|
|
|
job = service.enqueue(
|
|
original_filename="sample.txt",
|
|
stored_path=source,
|
|
language="a",
|
|
voice="af_alloy",
|
|
speed=1.0,
|
|
use_gpu=False,
|
|
subtitle_mode="Sentence",
|
|
output_format="wav",
|
|
save_mode="Save next to input file",
|
|
output_folder=outputs,
|
|
replace_single_newlines=False,
|
|
subtitle_format="srt",
|
|
total_characters=len(payload),
|
|
)
|
|
|
|
deadline = time.time() + 5
|
|
while time.time() < deadline and job.status not in {JobStatus.COMPLETED, JobStatus.FAILED, JobStatus.CANCELLED}:
|
|
time.sleep(0.05)
|
|
|
|
service.shutdown()
|
|
|
|
assert runner_invocations, "conversion runner was never called"
|
|
assert job.status is JobStatus.COMPLETED
|
|
assert job.progress == 1.0
|
|
assert job.result.audio_path == outputs / f"{job.id}.wav"
|
|
assert job.chunk_level == "paragraph"
|
|
assert job.speaker_mode == "single"
|
|
assert job.chunks == []
|
|
assert not job.generate_epub3 |