import subprocess
import sys

def install_rvc():
    print("Instalando rvc-python sin dependencias...")
    subprocess.check_call([
        sys.executable, "-m", "pip", "install", "--no-deps",
        "git+https://github.com/daswer123/rvc-python.git@main"
    ])
    print("rvc-python instalado.")

    # Downgradear pip temporalmente
    print("Downgradear pip temporalmente para instalar omegaconf 2.0.5...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "pip==23.3.2"])

    print("Instalando omegaconf 2.0.5 (compatible con fairseq/hydra)...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "omegaconf==2.0.5"])
    print("omegaconf 2.0.5 instalado.")

    # Volver a pip actual
    print("Volver a pip actual...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "pip==24.2"])

    print("Instalando fairseq sin dependencias...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-deps", "fairseq==0.12.2"])
    print("fairseq instalado.")

    # --- AQUÍ ESTABA EL ERROR DE INDENTACIÓN ---
    print("Instalando todas las dependencias de fairseq con --no-deps...")
    deps = [
        "bitarray",
        "sacrebleu",
        "editdistance",
        "portalocker",
        "cffi",
        "absl-py",
        "regex",
        "tabulate",
        "loguru",
        "colorama",
        "lxml",
    ]
    for dep in deps:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-deps", dep])
    # -------------------------------------------

    print("Dependencias de fairseq instaladas.")

    print("Instalando hydra-core sin dependencias...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-deps", "hydra-core==1.0.7"])
    print("hydra-core instalado.")

    print("Instalando antlr4-python3-runtime...")
    subprocess.check_call([sys.executable, "-m", "pip", "install", "antlr4-python3-runtime==4.9.3"])
    print("antlr4-python3-runtime instalado.")

    print("Instalación de RVC completa.")

if __name__ == "__main__":
    install_rvc()