import sys
import os
import logging
import subprocess
import traceback

EXT_PATH = os.path.dirname(os.path.abspath(__file__))

sys.path.insert(0, EXT_PATH)

log = logging.getLogger("AniDoc")

download_models = True

try:
    folder_paths_path = os.path.abspath(os.path.join(EXT_PATH, "..", "..", "folder_paths.py"))

    sys.path.append(os.path.dirname(folder_paths_path))

    import folder_paths
    
    DIFFUSERS_DIR = os.path.join(folder_paths.models_dir, "diffusers")
    ANIDOC_DIR = os.path.join(DIFFUSERS_DIR, "anidoc")
    SVD_I2V_DIR = os.path.join(
        DIFFUSERS_DIR,
        "stable-video-diffusion-img2vid-xt",
    )
except:
    download_models = False

COTRACKER = os.path.join(EXT_PATH, "cotracker")
    
try:
    log.info("Installing requirements")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", f"{EXT_PATH}/requirements.txt", "--no-warn-script-location"])
    
    if download_models:
        from huggingface_hub import snapshot_download
        
        log.info("Downloading Necessary models")

        try:
            log.info(f"Downloading AniDoc model to: {ANIDOC_DIR}")
            snapshot_download(
                repo_id="Yhmeng1106/anidoc",
                ignore_patterns=["*.md"],
                local_dir=DIFFUSERS_DIR,
                local_dir_use_symlinks=False,
            )
        except Exception:
            traceback.print_exc()
            log.error("Failed to download AniDoc model")
            
        try:
            log.info(f"Downloading stable diffusion video img2vid to: {SVD_I2V_DIR}")
            snapshot_download(
                repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1",
                allow_patterns=["*.json", "*fp16*"],
                ignore_patterns=["*unet*"],
                local_dir=SVD_I2V_DIR,
                local_dir_use_symlinks=False,
            )
        except Exception:
            traceback.print_exc()
            log.error("Failed to download stable diffusion video img2vid")
    
    try:
        log.info("Installing CoTracker")
        subprocess.check_call([sys.executable, "-m", "pip", "install", COTRACKER])
    except Exception:
        traceback.print_exc()
        log.error("Failed to install CoTracker")
    
    log.info("AniDoc Installation completed")
        
except Exception:
    traceback.print_exc()
    log.error("AniDoc Installation failed")