Update app.py

app.py

@@ -14,8 +14,10 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 comfyui_path = os.path.join(current_dir, "ComfyUI")
 sys.path.append(comfyui_path)
 
-# Agora podemos importar os módulos do ComfyUI
+# Agora importar os módulos do ComfyUI
 import folder_paths
+from nodes import NODE_CLASS_MAPPINGS, init_extra_nodes
+import comfy.utils
 
 # Configuração inicial e diagnóstico CUDA
 print("Python version:", sys.version)
@@ -25,38 +27,16 @@ print("Quantidade de GPUs:", torch.cuda.device_count())
 if torch.cuda.is_available():
     print("GPU atual:", torch.cuda.get_device_name(0))
 
-# 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)
-
-# Inicializar o ComfyUI
-def init_comfyui():
-    import execution
-    from nodes import NODE_CLASS_MAPPINGS, init_extra_nodes
-    import server
-    import asyncio
-    
-    # Criar e configurar o loop de eventos
-    loop = asyncio.new_event_loop()
-    asyncio.set_event_loop(loop)
-    
-    # Inicializar servidor e nós
-    server_instance = server.PromptServer(loop)
-    execution.PromptQueue(server_instance)
-    init_extra_nodes()
-    
-    return NODE_CLASS_MAPPINGS
-
-print("Inicializando ComfyUI...")
-NODE_CLASS_MAPPINGS = init_comfyui()
-
 # Configuração de diretórios
 BASE_DIR = os.path.dirname(os.path.realpath(__file__))
 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 ComfyUI...")
+init_extra_nodes()
+
 # Helper function
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
     try: