Update app.py

app.py

@@ -15,8 +15,36 @@ from comfy import model_management
 # Configuração de diretórios
 BASE_DIR = os.path.dirname(os.path.realpath(__file__))
 output_dir = os.path.join(BASE_DIR, "output")
+models_dir = os.path.join(BASE_DIR, "models")
 os.makedirs(output_dir, exist_ok=True)
-folder_paths.set_output_directory(output_dir)
+os.makedirs(models_dir, exist_ok=True)
+
+# Configurar caminhos dos modelos
+for model_folder in ["style_models", "text_encoders", "vae", "diffusion_models", "clip_vision"]:
+    folder_path = os.path.join(models_dir, model_folder)
+    os.makedirs(folder_path, exist_ok=True)
+    folder_paths.add_model_folder_path(model_folder, folder_path)
+
+# Download dos modelos
+print("Baixando modelos necessários...")
+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.sft", 
+                local_dir=os.path.join(models_dir, "diffusion_models"))
+hf_hub_download(repo_id="google/siglip-so400m-patch14-384", 
+                filename="model.safetensors", 
+                local_dir=os.path.join(models_dir, "clip_vision"))
 
 # Diagnóstico CUDA
 print("Python version:", sys.version)