#!/usr/bin/env python3
import os
import sys
import subprocess
import shutil
from pathlib import Path

# Try to load the .env file so the wizard knows what's already configured
try:
    from dotenv import load_dotenv
    # Path logic matching main.py
    script_dir = os.path.dirname(os.path.abspath(__file__))
    # Expected: .../video_transcription/../.env_files -> .../personal_development/.env_files
    env_path = os.path.abspath(os.path.join(script_dir, '../.env_files/.env.aitranscribe'))
    if os.path.exists(env_path):
        load_dotenv(env_path)
except ImportError:
    pass

def clear_screen():
    os.system('cls' if os.name == 'nt' else 'clear')

def get_input(prompt, default=None):
    """Helper to get input with a default value."""
    if default:
        user_input = input(f"{prompt} [{default}]: ").strip()
        return user_input if user_input else default
    else:
        return input(f"{prompt}: ").strip()

def get_yes_no(prompt, default="y"):
    """Helper to get boolean input."""
    display_default = "Y/n" if default.lower() in ["y", "yes"] else "y/N"
    choice = get_input(f"{prompt} ({display_default})", default).lower()
    return choice in ["y", "yes", "true", "1"]

def print_header():
    print("==========================================")
    print("   AI Video Transcriber & Translator Wizard")
    print("==========================================")
    print("")

def main():
    clear_screen()
    print_header()

    # 1. Input File/Folder
    while True:
        input_path = get_input("Enter the path to the video file or folder")
        
        # Clean up input:
        # 1. Remove surrounding quotes (common when pasting paths)
        input_path = input_path.strip('"\'')
        # 2. Handle escaped spaces (e.g., "My\ Folder" -> "My Folder")
        input_path = input_path.replace(r'\ ', ' ')
        
        # Expand user (~) and resolve absolute path
        input_path = os.path.abspath(os.path.expanduser(input_path))
        
        if os.path.exists(input_path):
            break
        print(f"Error: Path '{input_path}' does not exist. Please try again.\n")

    print(f"Selected: {input_path}\n")

    # 2. Languages
    source_lang = get_input("Source Language (e.g., French, es)", default="auto")
    target_lang = get_input("Target Language for translation", default="English")
    print("")

    # 3. Model Size
    print("Model Size Options: tiny, base, small, medium, large, auto")
    model_size = get_input("Whisper Model Size", default="auto")
    print("")

    # 4. Features
    do_cleanup = get_yes_no("Cleanup temporary audio files after processing?", default="y")
    do_embed = get_yes_no("Embed subtitles into the video file (Soft Subs)?", default="y")
    do_diarize = get_yes_no("Enable Speaker Diarization (Identify speakers)?", default="n")
    
    do_delete_source = False
    if do_embed:
        print("\n⚠️  WARNING: Using this next option will PERMANENTLY DELETE the original video files.")
        print("   It will only run if the new subtitled video is successfully created.")
        do_delete_source = get_yes_no("Delete original source files after embedding?", default="n")
    
    hf_token = None
    if do_diarize:
        if not os.getenv("HF_TOKEN"):
            print("\nSpeaker Diarization requires a HuggingFace Token.")
            hf_token = get_input("Enter your HuggingFace Token (hidden)", default="")
            # In a real app we might use getpass, but standard input is fine for this wizard level
        else:
            print("Using HF_TOKEN from environment.")

    # 5. Build Command
    # script is in ai_transcriber/main.py relative to this script
    script_dir = os.path.dirname(os.path.abspath(__file__))
    main_script = os.path.join(script_dir, "ai_transcriber", "main.py")

    cmd = [sys.executable, main_script, input_path]
    
    cmd.extend(["--lang", target_lang])
    cmd.extend(["--model", model_size])
    
    if source_lang != "auto":
        cmd.extend(["--source-lang", source_lang])
    
    if do_cleanup:
        cmd.append("--cleanup")
    
    if do_embed:
        cmd.append("--embed")
        
    if do_delete_source:
        cmd.append("--delete-source")
        
    if do_diarize:
        cmd.append("--diarize")
        if hf_token:
            cmd.extend(["--hf-token", hf_token])

    # 6. Confirmation and Execution
    clear_screen()
    print_header()
    print("Configuration Complete!")
    print("-" * 30)
    print(f"Input:        {input_path}")
    print(f"Source Lang:  {source_lang}")
    print(f"Target Lang:  {target_lang}")
    print(f"Model:        {model_size}")
    print(f"Cleanup:      {do_cleanup}")
    print(f"Embed Subs:   {do_embed}")
    print(f"Delete Src:   {do_delete_source}")
    print(f"Diarization:  {do_diarize}")
    print("-" * 30)
    
    if not get_yes_no("Run this job now?", default="y"):
        print("Aborted.")
        sys.exit(0)

    print("\nStarting Job...\n")
    
    try:
        # Pass environment variables including HF_TOKEN if set
        env = os.environ.copy()
        if hf_token:
            env["HF_TOKEN"] = hf_token
            
        subprocess.run(cmd, check=True, env=env)
        print("\n✅ Job Complete!")
    except subprocess.CalledProcessError as e:
        print(f"\n❌ Job Failed with error code {e.returncode}")
    except KeyboardInterrupt:
        print("\nJob interrupted by user.")

if __name__ == "__main__":
    main()