Update app.py

app.py

@@ -16,35 +16,14 @@ print("CUDA disponível:", torch.cuda.is_available())
 print("Quantidade de GPUs:", torch.cuda.device_count())
 if torch.cuda.is_available():
     print("GPU atual:", torch.cuda.get_device_name(0))
-else:
-    print("CUDA não está disponível. Verificando por que:")
-    try:
-        torch.cuda.init()
-    except Exception as e:
-        print("Erro ao inicializar CUDA:", str(e))
 
 # Adicionar o caminho da pasta ComfyUI ao sys.path
 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
+# Importar ComfyUI components
+from nodes import NODE_CLASS_MAPPINGS, init_extra_nodes
 from comfy import model_management
 import folder_paths
 
@@ -54,6 +33,10 @@ output_dir = os.path.join(BASE_DIR, "output")
 os.makedirs(output_dir, exist_ok=True)
 folder_paths.set_output_directory(output_dir)
 
+# Inicializar nós extras
+print("Inicializando nós extras...")
+init_extra_nodes()
+
 # Helper function
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
     try:
@@ -75,6 +58,7 @@ def download_models():
     ]
     
     for repo_id, filename, local_dir in models:
+        os.makedirs(local_dir, exist_ok=True)
         hf_hub_download(repo_id=repo_id, filename=filename, local_dir=local_dir)
 
 # Download models antes de inicializar