import subprocess
import os
import torch
import sys

print("🚀 FaceFusion HF Space - v2 (com Model Download)")

provider = "cuda" if torch.cuda.is_available() else "cpu"
print(f"🔧 Provider: {provider}")

# Clonar apenas se não existir
if not os.path.exists("fx36"):
    print("📥 Clonando repositório...")
    subprocess.run([
        "git", "clone", "https://github.com/rfs89y/fx36.git", "fx36", "--single-branch"
    ], check=True)

os.chdir("fx36")

print("📦 Instalando dependências mínimas...")
subprocess.run(["pip", "install", "--no-deps", "-r", "requirements.txt"], check=True)
subprocess.run(["pip", "install", "onnxruntime"], check=True)

print("🔄 Baixando modelos (isso pode demorar na primeira vez)...")
# Força download dos modelos
subprocess.run([sys.executable, "facefusion.py", "run", "--execution-providers", provider, "--skip-conda"], check=False)

print("🌐 Iniciando Interface...")

# Tentativa final de execução
subprocess.run([
    "python", "facefusion.py", "run",
    "--execution-providers", provider
])