# setup.py
from huggingface_hub import snapshot_download

def pre_setup():
    print("🔧 Baixando modelo base para evitar erros de rede...")
    try:
        snapshot_download(repo_id="runwayml/stable-diffusion-v1-5", allow_patterns=["scheduler/*", "text_encoder/*", "tokenizer/*", "unet/*", "vae/*"])
        print("✅ Modelo pré-baixado")
    except Exception as e:
        print(f"⚠️ Falha ao pré-baixar: {e}")

if __name__ == "__main__":
    pre_setup()