"""
Historical 2D Studio - Multilingual Voice, Audio & Subtitle Engine (2026)
Optimized for Low-Spec / Multi-Core Hardware. Zero-cost, 100% Copyright-Safe.
"""

import asyncio
import os
import subprocess
import sys
import edge_tts

# Danh mục giọng đọc tài liệu đỉnh cao (Documentary & War Room Voices)
DOCUMENTARY_VOICES = {
    "en": {"voice": "en-US-ChristopherNeural", "rate": "-4%", "pitch": "-3Hz", "desc": "Authoritative Baritone (BBC / History Style)"},
    "vi": {"voice": "vi-VN-NamMinhNeural", "rate": "-2%", "pitch": "-2Hz", "desc": "Hào hùng, trầm ấm sử thi"},
    "de": {"voice": "de-DE-ConradNeural", "rate": "-3%", "pitch": "-2Hz", "desc": "Deep German Military Chronicle"},
    "fr": {"voice": "fr-FR-HenriNeural", "rate": "-3%", "pitch": "-2Hz", "desc": "Classical French Narrative"},
    "es": {"voice": "es-ES-AlvaroNeural", "rate": "-3%", "pitch": "-2Hz", "desc": "Dramatic Castilian Voice"},
    "ja": {"voice": "ja-JP-KeitaNeural", "rate": "-2%", "pitch": "-1Hz", "desc": "Historical Samurai Chronicle"}
}

async def generate_narration_and_subtitles(text, lang="en", output_audio="output/narration.mp3", output_srt="output/subtitles.srt"):
    """
    Sinh file thuyết minh MP3 và file phụ đề SRT đồng bộ tuyệt đối về mặt thời gian.
    Không tốn GPU, chạy siêu tốc trên mọi cấu hình máy.
    """
    voice_config = DOCUMENTARY_VOICES.get(lang, DOCUMENTARY_VOICES["en"])
    communicate = edge_tts.Communicate(
        text=text,
        voice=voice_config["voice"],
        rate=voice_config["rate"],
        pitch=voice_config["pitch"]
    )
    
    submaker = edge_tts.SubMaker()
    
    os.makedirs(os.path.dirname(os.path.abspath(output_audio)), exist_ok=True)
    os.makedirs(os.path.dirname(os.path.abspath(output_srt)), exist_ok=True)
    
    with open(output_audio, "wb") as f_audio:
        async for chunk in communicate.stream():
            if chunk["type"] == "audio":
                f_audio.write(chunk["data"])
            elif chunk["type"] in ["SentenceBoundary", "WordBoundary"]:
                submaker.feed(chunk)
                
    with open(output_srt, "w", encoding="utf-8") as f_srt:
        f_srt.write(submaker.get_srt())
        
    print(f"[OK] Voice ({lang.upper()} - {voice_config['voice']}) saved: {output_audio}")
    print(f"[OK] Subtitles (SRT) saved: {output_srt}")
    return output_audio, output_srt

def generate_procedural_war_drum(output_wav, duration_sec=6.0, bpm=60):
    """
    Tạo tiếng trống trận dồn dập (War Drum / Heartbeat rhythm) hoàn toàn bằng thuật toán FFmpeg,
    100% không lo vấn đề bản quyền âm thanh (Public Domain / Procedural).
    """
    cmd = [
        "ffmpeg", "-y",
        "-f", "lavfi",
        "-i", f"sine=frequency=55:duration={duration_sec}",
        "-af", f"volume=1.8, lowpass=f=120, aecho=0.8:0.88:60:0.4",
        "-ac", "2",
        output_wav
    ]
    subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, check=True)
    return output_wav

def merge_video_audio_subtitles(video_path, voice_audio_path, drum_audio_path, output_final_mp4):
    """
    Hòa trộn video sa bàn, giọng thuyết minh tài liệu, và tiếng trống trận nền.
    Sử dụng GPU NVENC stream hoặc stream copy siêu nhanh.
    """
    cmd = [
        "ffmpeg", "-y",
        "-i", video_path,
        "-i", voice_audio_path,
        "-i", drum_audio_path,
        "-filter_complex", "[1:a]volume=1.0[v_voice]; [2:a]volume=0.35[v_drum]; [v_voice][v_drum]amix=inputs=2:duration=first[a_out]",
        "-map", "0:v",
        "-map", "[a_out]",
        "-c:v", "copy",
        "-c:a", "aac",
        "-b:a", "192k",
        "-shortest",
        output_final_mp4
    ]
    subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
    print(f"[OK] Master Video with Audio & Soundscape saved: {output_final_mp4}")
    return output_final_mp4

if __name__ == "__main__":
    test_script_en = (
        "In 216 BC, on the dust-swept plains of Cannae, "
        "Hannibal Barca executed the greatest tactical double encirclement in human history."
    )
    asyncio.run(generate_narration_and_subtitles(
        test_script_en, 
        lang="en", 
        output_audio="output/cannae_voice_en.mp3", 
        output_srt="output/cannae_sub_en.srt"
    ))
    
    drum_path = "output/war_drum_ambient.wav"
    generate_procedural_war_drum(drum_path, duration_sec=6.0)
    
    video_source = "output/cannae_tactical_demo.mp4"
    if os.path.exists(video_source):
        merge_video_audio_subtitles(
            video_source,
            "output/cannae_voice_en.mp3",
            drum_path,
            "output/cannae_master_with_audio.mp4"
        )