Update app.py

app.py

@@ -28,6 +28,22 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 comfyui_path = os.path.join(current_dir, "ComfyUI")
 sys.path.append(comfyui_path)
 
+# Import custom nodes primeiro
+def import_custom_nodes():
+    import asyncio
+    import execution
+    from nodes import init_extra_nodes
+    import server
+
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    server_instance = server.PromptServer(loop)
+    execution.PromptQueue(server_instance)
+    init_extra_nodes()
+
+print("Importando nós customizados...")
+import_custom_nodes()
+
 from nodes import NODE_CLASS_MAPPINGS
 from comfy import model_management
 import folder_paths
@@ -47,6 +63,7 @@ def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
 
 # Baixar modelos necessários
 def download_models():
+    print("Baixando modelos...")
     models = [
         ("black-forest-labs/FLUX.1-Redux-dev", "flux1-redux-dev.safetensors", "models/style_models"),
         ("comfyanonymous/flux_text_encoders", "t5xxl_fp16.safetensors", "models/text_encoders"),
@@ -60,6 +77,9 @@ def download_models():
     for repo_id, filename, local_dir in models:
         hf_hub_download(repo_id=repo_id, filename=filename, local_dir=local_dir)
 
+# Download models antes de inicializar
+download_models()
+
 # Inicializar modelos
 print("Inicializando modelos...")
 with torch.inference_mode():
@@ -87,22 +107,9 @@ with torch.inference_mode():
     ]
     model_management.load_models_gpu(valid_models)
 
-def import_custom_nodes():
-    import asyncio
-    import execution
-    from nodes import init_extra_nodes
-    import server
-
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    server_instance = server.PromptServer(loop)
-    execution.PromptQueue(server_instance)
-    init_extra_nodes()
-
 @spaces.GPU
 def generate_image(prompt, input_image, lora_weight, progress=gr.Progress(track_tqdm=True)):
     """Função principal de geração com monitoramento de progresso"""
-    import_custom_nodes()
     try:
         with torch.inference_mode():
             # Codificar texto
@@ -161,7 +168,4 @@ with gr.Blocks() as app:
     )
 
 if __name__ == "__main__":
-    # Download models at startup
-    download_models()
-    # Launch the app
     app.launch()