Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 import spaces
 from typing import Union, Sequence, Mapping, Any
+import asyncio
 
 # Diagnóstico CUDA
 print("Python version:", sys.version)
@@ -23,10 +24,11 @@ comfyui_path = os.path.join(current_dir, "ComfyUI")
 sys.path.append(comfyui_path)
 
 # Importar ComfyUI components
-import execution
-from nodes import NODE_CLASS_MAPPINGS
+from nodes import NODE_CLASS_MAPPINGS, init_custom_nodes
 import folder_paths
 from comfy import model_management
+import execution
+import server
 
 # Configuração de diretórios
 BASE_DIR = os.path.dirname(os.path.realpath(__file__))
@@ -34,8 +36,12 @@ output_dir = os.path.join(BASE_DIR, "output")
 os.makedirs(output_dir, exist_ok=True)
 folder_paths.set_output_directory(output_dir)
 
-# Inicializar execução
-execution.init()
+# Inicializar o servidor e os nós
+loop = asyncio.new_event_loop()
+asyncio.set_event_loop(loop)
+server_instance = server.PromptServer(loop)
+execution.PromptQueue(server_instance)
+init_custom_nodes()
 
 # Helper function
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any: