Update app.py

app.py

@@ -5,16 +5,24 @@ from pathlib import Path
 from PIL import Image
 import gradio as gr
 from huggingface_hub import hf_hub_download
+import sys
+
+# 1. Configuração de Caminhos
+current_dir = os.path.dirname(os.path.abspath(__file__))
+comfyui_path = os.path.join(current_dir, "ComfyUI")
+sys.path.append(comfyui_path)
+
+# 2. Imports do ComfyUI
 from nodes import NODE_CLASS_MAPPINGS
 import folder_paths
 
-# Configure base and output directories
+# 3. 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)
 
-# Download models
+# 4. Download de Modelos
 def download_models():
     models = [
         ("black-forest-labs/FLUX.1-Redux-dev", "flux1-redux-dev.safetensors", "style_models"),
@@ -33,7 +41,7 @@ def download_models():
         hf_hub_download(repo_id=repo_id, filename=filename, local_dir=model_dir)
         folder_paths.add_model_folder_path(model_type, model_dir)
 
-# Load custom nodes
+# 5. Load custom nodes
 def import_custom_nodes():
     import asyncio
     import execution
@@ -47,7 +55,7 @@ def import_custom_nodes():
     execution.PromptQueue(server_instance)
     init_extra_nodes()
 
-# Main function to execute the workflow and generate an image
+# 6. Main function to execute the workflow and generate an image
 def generate_image(prompt, input_image, lora_weight, guidance, downsampling_factor, weight, seed, width, height, batch_size, steps):
     import_custom_nodes()
     
@@ -184,7 +192,7 @@ def generate_image(prompt, input_image, lora_weight, guidance, downsampling_fact
         print(f"Error during generation: {str(e)}")
         return None
 
-# Gradio Interface
+# 7. Gradio Interface
 with gr.Blocks() as app:
     gr.Markdown("# FLUX Redux Image Generator")