feat: Release v0.10 - Modular Architecture & External Config
- Refactor codebase into src/ (preprocessing, translation, assembly) - Add pipeline/ scripts for individual stages - Externalize configuration to config/config.yaml - Fix Cover Image preservation - Update documentation and manuals
This commit is contained in:
@@ -0,0 +1,69 @@
|
||||
import argparse
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
|
||||
# Add project root to sys.path
|
||||
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
||||
|
||||
from src.preprocessing.epub_cleaner import EpubCleaner
|
||||
from src.preprocessing.text_extractor import FineGrainedExtractor
|
||||
from src.translation.manifest_manager import ManifestManager
|
||||
from src.common.data_model import BookStructure
|
||||
from src.common.utils import setup_logger, ensure_directory
|
||||
from src.common.exceptions import EpubTranslatorError
|
||||
from src.common.paths import get_work_dirs
|
||||
|
||||
logger = setup_logger("pipeline_preprocess")
|
||||
|
||||
def run_preprocess(args):
|
||||
input_path = Path(args.input_epub)
|
||||
if not input_path.exists():
|
||||
logger.error(f"Input file not found: {input_path}")
|
||||
sys.exit(1)
|
||||
|
||||
work = get_work_dirs(input_path)
|
||||
ensure_directory(work["root"])
|
||||
|
||||
structure_path = work["structure"]
|
||||
manifest_path = work["manifest"]
|
||||
|
||||
try:
|
||||
# 1. Clean / Load Structure
|
||||
if structure_path.exists() and not args.force:
|
||||
logger.info(f"Reusing existing structure: {structure_path}")
|
||||
structure = BookStructure.load(structure_path)
|
||||
else:
|
||||
logger.info("Cleaning EPUB...")
|
||||
cleaner = EpubCleaner(input_path, work["root"])
|
||||
structure_path = cleaner.clean()
|
||||
structure = BookStructure.load(structure_path)
|
||||
|
||||
# 2. Extract Text
|
||||
logger.info("Extracting text segments...")
|
||||
extractor = FineGrainedExtractor()
|
||||
entries = extractor.extract(structure)
|
||||
|
||||
# 3. Update Manifest
|
||||
logger.info(f"Updating manifest: {manifest_path}")
|
||||
manager = ManifestManager(manifest_path)
|
||||
manager.load()
|
||||
manager.add_entries(entries)
|
||||
manager.save()
|
||||
|
||||
logger.info("Preprocessing complete.")
|
||||
|
||||
except EpubTranslatorError as e:
|
||||
logger.error(f"Preprocessing failed: {e}")
|
||||
sys.exit(1)
|
||||
except Exception as e:
|
||||
logger.critical(f"Unexpected error: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
parser = argparse.ArgumentParser(description="Step 1: Preprocessing (Clean + Extract)")
|
||||
parser.add_argument("input_epub", help="Path to input EPUB")
|
||||
parser.add_argument("--force", action="store_true", help="Force re-clean")
|
||||
|
||||
args = parser.parse_args()
|
||||
run_preprocess(args)
|
||||
Reference in New Issue
Block a user