Update app.py

app.py

@@ -71,24 +71,28 @@ def verify_file_exists(folder: str, filename: str) -> bool:
 
 # 7. Download de Modelos
 logger.info("Baixando modelos necessários...")
-MODELS_TO_DOWNLOAD = [
-   ("black-forest-labs/FLUX.1-Redux-dev", "flux1-redux-dev.safetensors", "style_models"),
-   ("comfyanonymous/flux_text_encoders", "t5xxl_fp16.safetensors", "text_encoders"),
-   ("zer0int/CLIP-GmP-ViT-L-14", "ViT-L-14-TEXT-detail-improved-hiT-GmP-TE-only-HF.safetensors", "text_encoders"),
-   ("black-forest-labs/FLUX.1-dev", "ae.safetensors", "vae"),
-   ("black-forest-labs/FLUX.1-dev", "flux1-dev.safetensors", "unet"),
-   ("Comfy-Org/sigclip_vision_384", "sigclip_vision_patch14_384.safetensors", "clip_vision")
-]
-
-for repo_id, filename, folder in MODELS_TO_DOWNLOAD:
-   try:
-       logger.info(f"Baixando {filename} para {folder}...")
-       hf_hub_download(repo_id=repo_id, filename=filename, local_dir=os.path.join(models_dir, folder))
-       if not verify_file_exists(folder, filename):
-           raise FileNotFoundError(f"Arquivo não encontrado após download: {filename}")
-   except Exception as e:
-       logger.error(f"Erro ao baixar {filename} de {repo_id}: {str(e)}")
-       raise
+try:
+   hf_hub_download(repo_id="black-forest-labs/FLUX.1-Redux-dev", 
+                   filename="flux1-redux-dev.safetensors", 
+                   local_dir=os.path.join(models_dir, "style_models"))
+   hf_hub_download(repo_id="comfyanonymous/flux_text_encoders", 
+                   filename="t5xxl_fp16.safetensors", 
+                   local_dir=os.path.join(models_dir, "text_encoders"))
+   hf_hub_download(repo_id="zer0int/CLIP-GmP-ViT-L-14", 
+                   filename="ViT-L-14-TEXT-detail-improved-hiT-GmP-TE-only-HF.safetensors", 
+                   local_dir=os.path.join(models_dir, "text_encoders"))
+   hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", 
+                   filename="ae.safetensors", 
+                   local_dir=os.path.join(models_dir, "vae"))
+   hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", 
+                   filename="flux1-dev.safetensors", 
+                   local_dir=os.path.join(models_dir, "unet"))
+   hf_hub_download(repo_id="Comfy-Org/sigclip_vision_384", 
+                   filename="model.safetensors",
+                   local_dir=os.path.join(models_dir, "clip_vision"))
+except Exception as e:
+   logger.error(f"Erro ao baixar modelos: {str(e)}")
+   raise
 
 # 8. Inicialização dos Modelos
 logger.info("Inicializando modelos...")
@@ -102,20 +106,26 @@ try:
            clip_name2="ViT-L-14-TEXT-detail-improved-hiT-GmP-TE-only-HF.safetensors",
            type="flux"
        )
+       if CLIP_MODEL is None:
+           raise ValueError("Falha ao carregar CLIP model")
 
        # CLIP Vision
        logger.info("Carregando CLIP Vision...")
        clipvisionloader = NODE_CLASS_MAPPINGS["CLIPVisionLoader"]()
        CLIP_VISION = clipvisionloader.load_clip(
-           clip_name="sigclip_vision_patch14_384.safetensors"
+           clip_name="model.safetensors"
        )
+       if CLIP_VISION is None:
+           raise ValueError("Falha ao carregar CLIP Vision model")
 
-       # Style Model
+       # Style Model (Carregado separadamente)
        logger.info("Carregando Style Model...")
        stylemodelloader = NODE_CLASS_MAPPINGS["StyleModelLoader"]()
        STYLE_MODEL = stylemodelloader.load_style_model(
            style_model_name="flux1-redux-dev.safetensors"
        )
+       if STYLE_MODEL is None:
+           raise ValueError("Falha ao carregar Style Model")
 
        # VAE
        logger.info("Carregando VAE...")
@@ -123,6 +133,8 @@ try:
        VAE_MODEL = vaeloader.load_vae(
            vae_name="ae.safetensors"
        )
+       if VAE_MODEL is None:
+           raise ValueError("Falha ao carregar VAE model")
 
        # UNET
        logger.info("Carregando UNET...")
@@ -131,13 +143,16 @@ try:
            unet_name="flux1-dev.safetensors",
            weight_dtype="fp8_e4m3fn"
        )
+       if UNET_MODEL is None:
+           raise ValueError("Falha ao carregar UNET model")
 
        logger.info("Carregando modelos na GPU...")
-       model_loaders = [CLIP_MODEL, VAE_MODEL, STYLE_MODEL, CLIP_VISION, UNET_MODEL]
+       model_loaders = [CLIP_MODEL, VAE_MODEL, CLIP_VISION, UNET_MODEL]
        model_management.load_models_gpu([
            loader[0].patcher if hasattr(loader[0], 'patcher') else loader[0] 
            for loader in model_loaders
        ])
+       logger.info("Modelos carregados com sucesso")
 except Exception as e:
    logger.error(f"Erro ao inicializar modelos: {str(e)}")
    raise