Update app.py

app.py

@@ -62,7 +62,7 @@ def import_custom_nodes() -> None:
     asyncio.set_event_loop(loop)
     server_instance = server.PromptServer(loop)
     execution.PromptQueue(server_instance)
-    loop.run_until_complete(init_extra_nodes())
+    init_extra_nodes()
 
 # Setup ComfyUI and custom nodes
 if not os.path.exists("ComfyUI"):
@@ -81,7 +81,6 @@ os.makedirs("ComfyUI/models/diffusion_models", exist_ok=True)
 os.makedirs("ComfyUI/models/clip", exist_ok=True)
 os.makedirs("ComfyUI/models/vae", exist_ok=True)
 os.makedirs("ComfyUI/models/upscale_models", exist_ok=True)
-os.makedirs("ComfyUI/input", exist_ok=True)
 
 # Download models if not present
 diffusion_path = "ComfyUI/models/diffusion_models/flux1-dev-fp8.safetensors"
@@ -112,7 +111,8 @@ if not os.path.exists(esrgan_x4_path):
 add_comfyui_directory_to_sys_path()
 add_extra_model_paths()
 from folder_paths import add_model_folder_path
-add_model_folder_path("unet", find_path("diffusion_models"))
+comfy_dir = find_path("ComfyUI")
+add_model_folder_path("unet", os.path.join(comfy_dir, "models", "diffusion_models"))
 import_custom_nodes()
 
 from nodes import NODE_CLASS_MAPPINGS
@@ -200,7 +200,8 @@ def enhance_image(
         temp_path = tmp.name
 
     image_base = os.path.basename(temp_path)
-    input_dir = find_path("input")
+    comfy_dir = find_path("ComfyUI")
+    input_dir = os.path.join(comfy_dir, "input")
     input_image_path = os.path.join(input_dir, image_base)
     shutil.copy(temp_path, input_image_path)