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 folder_paths
 
 # Configuração inicial e diagnóstico CUDA
 print("Python version:", sys.version)
@@ -22,19 +23,26 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 comfyui_path = os.path.join(current_dir, "ComfyUI")
 sys.path.append(comfyui_path)
 
-# Importar ComfyUI components
-sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "ComfyUI/comfy"))
-
-import comfy.diffusers_load
-import comfy.samplers
-import comfy.sample
-import comfy.sd
-import comfy.utils
-from comfy.cli_args import args
-import folder_paths
+# Inicializar o ComfyUI
+def init_comfyui():
+    import execution
+    from nodes import NODE_CLASS_MAPPINGS, init_custom_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_custom_nodes()
+    
+    return NODE_CLASS_MAPPINGS
 
-# Importar nós do ComfyUI
-from nodes import CLIPTextEncode, VAEDecode, EmptyLatentImage, VAEEncode
+print("Inicializando ComfyUI...")
+NODE_CLASS_MAPPINGS = init_comfyui()
 
 # Configuração de diretórios
 BASE_DIR = os.path.dirname(os.path.realpath(__file__))
@@ -76,10 +84,25 @@ download_models()
 # Inicializar modelos
 print("Inicializando modelos...")
 with torch.inference_mode():
-    clip_text_encode = CLIPTextEncode()
-    vae_decode = VAEDecode()
-    vae_encode = VAEEncode()
-    empty_latent = EmptyLatentImage()
+    # CLIP
+    dualcliploader = NODE_CLASS_MAPPINGS["DualCLIPLoader"]()
+    dualcliploader_357 = dualcliploader.load_clip(
+        clip_name1="models/text_encoders/t5xxl_fp16.safetensors",
+        clip_name2="models/text_encoders/ViT-L-14-TEXT-detail-improved-hiT-GmP-HF.safetensors",
+        type="flux",
+    )
+
+    # Style Model
+    stylemodelloader = NODE_CLASS_MAPPINGS["StyleModelLoader"]()
+    stylemodelloader_441 = stylemodelloader.load_style_model(
+        style_model_name="models/style_models/flux1-redux-dev.safetensors"
+    )
+
+    # VAE
+    vaeloader = NODE_CLASS_MAPPINGS["VAELoader"]()
+    vaeloader_359 = vaeloader.load_vae(
+        vae_name="models/vae/ae.safetensors"
+    )
 
 @spaces.GPU
 def generate_image(prompt, input_image, lora_weight, guidance, downsampling_factor, weight, seed, width, height, batch_size, steps, progress=gr.Progress(track_tqdm=True)):