Revert "Revert "Revert "Skip re-cloning ComfyUI on restart to speed up startup"""

This reverts commit d7ff6601bb01717f0a0762f84e8bef99b46004c3.

comfy_integration/setup.py

@@ -16,34 +16,28 @@ def move_and_overwrite(src, dst):
 
 def initialize_comfyui():
     APP_DIR = sys.path[0]
-    # Check if ComfyUI already exists in the app directory to avoid re-cloning on every restart.
-    # The presence of the 'comfy' package (imported later) indicates a previous successful setup.
-    COMFYUI_PRESENT = os.path.isdir(os.path.join(APP_DIR, 'comfy'))
     COMFYUI_TEMP_DIR = "ComfyUI_temp"
 
-    if COMFYUI_PRESENT:
-        print("✅ ComfyUI already initialized – skipping clone and merge.")
+    print("--- Cloning ComfyUI Repository ---")
+    if not os.path.exists(COMFYUI_TEMP_DIR):
+        os.system(f"git clone https://github.com/comfy-Org/ComfyUI {COMFYUI_TEMP_DIR}")
+        print("✅ ComfyUI repository cloned.")
     else:
-        print("--- Cloning ComfyUI Repository ---")
-        if not os.path.exists(COMFYUI_TEMP_DIR):
-            os.system(f"git clone https://github.com/comfy-Org/ComfyUI {COMFYUI_TEMP_DIR}")
-            print("✅ ComfyUI repository cloned.")
-        else:
-            print("✅ ComfyUI repository already exists.")
-
-        print(f"--- Merging ComfyUI from '{COMFYUI_TEMP_DIR}' to '{APP_DIR}' ---")
-        for item in os.listdir(COMFYUI_TEMP_DIR):
-            src_path = os.path.join(COMFYUI_TEMP_DIR, item)
-            dst_path = os.path.join(APP_DIR, item)
-            if item == '.git':
-                continue
-            move_and_overwrite(src_path, dst_path)
-        
-        try:
-            shutil.rmtree(COMFYUI_TEMP_DIR)
-            print("✅ ComfyUI merged and temporary directory removed.")
-        except OSError as e:
-            print(f"⚠️ Could not remove temporary directory '{COMFYUI_TEMP_DIR}': {e}")
+        print("✅ ComfyUI repository already exists.")
+
+    print(f"--- Merging ComfyUI from '{COMFYUI_TEMP_DIR}' to '{APP_DIR}' ---")
+    for item in os.listdir(COMFYUI_TEMP_DIR):
+        src_path = os.path.join(COMFYUI_TEMP_DIR, item)
+        dst_path = os.path.join(APP_DIR, item)
+        if item == '.git':
+            continue
+        move_and_overwrite(src_path, dst_path)
+    
+    try:
+        shutil.rmtree(COMFYUI_TEMP_DIR)
+        print("✅ ComfyUI merged and temporary directory removed.")
+    except OSError as e:
+        print(f"⚠️ Could not remove temporary directory '{COMFYUI_TEMP_DIR}': {e}")
 
     print("--- Cloning third-party extensions for ComfyUI ---")
 
@@ -105,6 +99,7 @@ def initialize_comfyui():
     torch.cuda.current_device = lambda: 0
     torch.cuda.get_device_name = lambda device: "cpu"
 
+
     # Apply patch to the external ComfyUI model_management before import.
     external_model_mgmt_path = os.path.join(APP_DIR, "comfy", "model_management.py")
     if os.path.exists(external_model_mgmt_path):
@@ -182,6 +177,7 @@ def initialize_comfyui():
         return False
     model_mgmt.should_use_bf16 = _safe_should_use_bf16
 
+ 
     print("--- Environment Ready ---")
     
     print("✅ ComfyUI initialized with default attention mechanism.")
@@ -193,5 +189,3 @@ def initialize_comfyui():
     os.makedirs(os.path.join(APP_DIR, OUTPUT_DIR), exist_ok=True)
     
     print("✅ All required model directories are present.")
-
-    print("✅ All required model directories are present.")