mirror of
https://github.com/denizsafak/abogen.git
synced 2026-09-20 11:40:57 +02:00
Artem Akymenko
3857c27aae
- Remove 77 unused imports across domain/application/runner files via ruff - Add # noqa: F401 to re-exports used by tests and debug_tts_runner - Fix build_tts_context: usage_counter uses 'is not None' instead of truthiness - Fix test assertions: compiled rules use 'replacement' key not 'pronunciation' - Add re-exports: _compile_pronunciation_rules, _merge_pronunciation_overrides
142 lines
4.6 KiB
Python
142 lines
4.6 KiB
Python
"""ConversionRequest — normalized input for a conversion job.
|
|
|
|
This is NOT a WebUI Job and NOT a PyQt ConversionThread state.
|
|
It describes the TASK, not the UI.
|
|
|
|
UI adapters are responsible for converting their respective state
|
|
into a ConversionRequest before calling ConversionService.run().
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import dataclasses
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
from typing import Any, Dict, Optional
|
|
|
|
from abogen.application.conversion_config import (
|
|
ChapterChunkConfig,
|
|
Epub3ExportConfig,
|
|
PronunciationConfig,
|
|
SubtitleInputConfig,
|
|
WordSubstitutionConfig,
|
|
)
|
|
from abogen.domain.enums import Language, OutputFormat, SaveMode, SubtitleFormat, SubtitleMode
|
|
|
|
|
|
class ConversionRequestError(ValueError):
|
|
"""Raised when ConversionRequest has invalid field values."""
|
|
|
|
|
|
# Numeric field constraints: attr -> (min, max)
|
|
_NUMERIC_CONSTRAINTS: dict[str, tuple[float, float | None]] = {
|
|
"max_subtitle_words": (1, 500),
|
|
"speed": (0.5, 3.0),
|
|
"supertonic_total_steps": (2, 15),
|
|
"silence_between_chapters": (0.0, None),
|
|
"chapter_intro_delay": (0.0, None),
|
|
}
|
|
|
|
|
|
@dataclass
|
|
class ConversionRequest:
|
|
"""Normalized request for a conversion job.
|
|
|
|
Only contains fields that describe the conversion task itself.
|
|
UI-only fields (display, logging, user prompts) stay in adapters.
|
|
|
|
Feature toggles use config objects: if the object is present,
|
|
the feature is enabled. No boolean flags needed.
|
|
|
|
Validation runs on creation via __post_init__:
|
|
- None values → replaced with field default (from declaration)
|
|
- Numeric fields → clamped to valid range
|
|
"""
|
|
|
|
# --- Source ---
|
|
source_path: Optional[Path] = None
|
|
direct_text: Optional[str] = None
|
|
original_filename: str = ""
|
|
|
|
# --- TTS Settings ---
|
|
language: Language = Language.EN_US
|
|
tts_provider: str = "kokoro"
|
|
voice: str = "M1"
|
|
voice_profile: Optional[str] = None
|
|
speed: float = 1.0
|
|
use_gpu: bool = True
|
|
supertonic_total_steps: int = 5
|
|
|
|
# --- Output Format ---
|
|
output_format: OutputFormat = OutputFormat.WAV
|
|
subtitle_mode: SubtitleMode = SubtitleMode.DISABLED
|
|
subtitle_format: SubtitleFormat = SubtitleFormat.SRT
|
|
max_subtitle_words: int = 50
|
|
|
|
# --- Save Options ---
|
|
save_mode: SaveMode = SaveMode.SAVE_NEXT_TO_INPUT
|
|
output_folder: Optional[Path] = None
|
|
save_chapters_separately: bool = False
|
|
merge_chapters_at_end: bool = True
|
|
separate_chapters_format: OutputFormat = OutputFormat.WAV
|
|
save_as_project: bool = False
|
|
|
|
# --- Timing ---
|
|
silence_between_chapters: float = 2.0
|
|
chapter_intro_delay: float = 0.0
|
|
|
|
# --- Content Processing ---
|
|
replace_single_newlines: bool = False
|
|
read_title_intro: bool = False
|
|
read_closing_outro: bool = True
|
|
auto_prefix_chapter_titles: bool = True
|
|
normalize_chapter_opening_caps: bool = False
|
|
|
|
# --- Metadata ---
|
|
metadata_tags: Dict[str, Any] = field(default_factory=dict)
|
|
|
|
# --- Artifacts ---
|
|
cover_image_path: Optional[Path] = None
|
|
cover_image_mime: Optional[str] = None
|
|
|
|
# --- Feature configs (None = disabled) ---
|
|
word_substitution: Optional[WordSubstitutionConfig] = None
|
|
subtitle_input: Optional[SubtitleInputConfig] = None
|
|
epub3_export: Optional[Epub3ExportConfig] = None
|
|
pronunciation: Optional[PronunciationConfig] = None
|
|
chapter_chunk: Optional[ChapterChunkConfig] = None
|
|
|
|
def __post_init__(self) -> None:
|
|
"""Resolve None → default, then validate and clamp."""
|
|
_apply_none_defaults(self)
|
|
if not self.tts_provider:
|
|
self.tts_provider = "kokoro"
|
|
_clamp_numerics(self)
|
|
|
|
|
|
def _apply_none_defaults(obj: ConversionRequest) -> None:
|
|
"""Replace None values with field defaults from dataclass declaration."""
|
|
for f in dataclasses.fields(obj):
|
|
if getattr(obj, f.name) is not None:
|
|
continue
|
|
if f.default is not dataclasses.MISSING:
|
|
setattr(obj, f.name, f.default)
|
|
elif f.default_factory is not dataclasses.MISSING:
|
|
setattr(obj, f.name, f.default_factory())
|
|
|
|
|
|
def _clamp_numerics(obj: ConversionRequest) -> None:
|
|
"""Clamp numeric fields to valid ranges."""
|
|
for attr, (min_v, max_v) in _NUMERIC_CONSTRAINTS.items():
|
|
val = getattr(obj, attr)
|
|
if val is None:
|
|
continue
|
|
if not isinstance(val, (int, float)):
|
|
raise ConversionRequestError(
|
|
f"{attr} must be a number, got {type(val).__name__}"
|
|
)
|
|
clamped = max(min_v, float(val))
|
|
if max_v is not None:
|
|
clamped = min(max_v, clamped)
|
|
setattr(obj, attr, clamped)
|