import os
import shutil
from pathlib import Path
from config.config import AUDIOBOOK_ROOT
base = AUDIOBOOK_ROOT


def accept_revision(index):
    """
    Archive original chunk and replace with revised version.
    Assumes revised version is saved as: chunk_XXX_rev.wav
    """
    base = Path(AUDIO_OUTPUT_DIR)
    original = base / f"chunk_{index:03}.wav"
    revised = base / f"chunk_{index:03}_rev.wav"
    archive_dir = base / "Audio_Revisions"
    archive_dir.mkdir(exist_ok=True)

    if not revised.exists():
        print("❌ No revised file found. Cannot accept.")
        return

    # Archive original if exists
    if original.exists():
        archived = archive_dir / f"chunk_{index:03}_orig.wav"
        shutil.move(original, archived)
        print(f"📦 Original chunk archived to {archived.name}")
    else:
        print(f"⚠️ Original chunk missing — no archive created.")

    # Move revised chunk to main filename
    shutil.move(revised, original)
    print(f"✅ Revised chunk accepted as {original.name}")