formatting

.gitignore

@@ -1,3 +1,4 @@
 venv
 .env
+.idea/
 ComfyUI

install.py

@@ -31,7 +31,7 @@ def manage_git_repo(repo_url, install_path, requirements=False, submodules=False
     """
     # Save the original directory
     original_dir = os.getcwd()
-    
+
     if not os.path.exists(install_path) or not os.path.isdir(install_path) or not os.path.exists(
             os.path.join(install_path, ".git")):
         print(f"📂 Cloning {os.path.basename(install_path)}...")
@@ -50,7 +50,7 @@ def manage_git_repo(repo_url, install_path, requirements=False, submodules=False
         run_command("python -m pip install -r requirements.txt")
 
     print(f"✅ {os.path.basename(install_path)} installed and updated.")
-    
+
     # Change back to the original directory
     os.chdir(original_dir)
 
@@ -63,6 +63,7 @@ def install_comfyui():
         requirements=True
     )
 
+
 def download_huggingface_models():
     """Download required models from Hugging Face and symlink to ComfyUI models directory."""
     from huggingface_hub import hf_hub_download
@@ -72,7 +73,8 @@ def download_huggingface_models():
         {"repo_id": "Shakker-Labs/FLUX.1-dev-ControlNet-Union-Pro", "filename": "diffusion_pytorch_model.safetensors",
          "folder": "controlnet"},
         {"repo_id": "guozinan/PuLID", "filename": "pulid_flux_v0.9.1.safetensors", "folder": "pulid"},
-        {"repo_id": "comfyanonymous/flux_text_encoders", "filename": "t5xxl_fp16.safetensors", "folder": "text_encoders"},
+        {"repo_id": "comfyanonymous/flux_text_encoders", "filename": "t5xxl_fp16.safetensors",
+         "folder": "text_encoders"},
         {"repo_id": "comfyanonymous/flux_text_encoders", "filename": "clip_l.safetensors", "folder": "text_encoders"},
     ]
 
@@ -84,7 +86,7 @@ def download_huggingface_models():
     for model in hf_models:
         try:
             model_path = hf_hub_download(
-                repo_id=model["repo_id"], 
+                repo_id=model["repo_id"],
                 filename=model["filename"],
                 cache_dir=CACHE_PATH,
                 repo_type=model.get("repo_type", "model")
@@ -108,20 +110,20 @@ def download_huggingface_models():
 def download_and_extract_antelopev2():
     """Download and extract AntelopeV2 model for insightface."""
     import zipfile, requests, shutil
-    
+
     base_path = os.path.join(MODEL_PATH, "insightface/models")
     model_target_path = os.path.join(base_path, "antelopev2")
     download_url = "https://huggingface.co/MonsterMMORPG/tools/resolve/main/antelopev2.zip"
     zip_path = os.path.join(base_path, "antelopev2.zip")
     temp_extract_path = os.path.join(base_path, "temp_antelopev2")
-    
+
     os.makedirs(base_path, exist_ok=True)
 
     if not os.path.exists(model_target_path) or not os.listdir(model_target_path):
         # First, remove any existing problematic directories
         if os.path.exists(model_target_path):
             shutil.rmtree(model_target_path)
-            
+
         print(f"📥 Downloading AntelopeV2 model...")
         try:
             response = requests.get(download_url, stream=True)
@@ -133,15 +135,15 @@ def download_and_extract_antelopev2():
 
             # Create a temporary extraction directory
             os.makedirs(temp_extract_path, exist_ok=True)
-            
+
             print("📂 Extracting AntelopeV2 model...")
             with zipfile.ZipFile(zip_path, "r") as zip_ref:
                 zip_ref.extractall(temp_extract_path)
             print("✅ Extraction complete.")
-            
+
             # Create the target directory
             os.makedirs(model_target_path, exist_ok=True)
-            
+
             # Move the model files to the correct location
             # The ZIP contains a nested antelopev2 directory we need to move files from
             nested_model_dir = os.path.join(temp_extract_path, "antelopev2")
@@ -150,16 +152,16 @@ def download_and_extract_antelopev2():
                     source = os.path.join(nested_model_dir, item)
                     target = os.path.join(model_target_path, item)
                     shutil.move(source, target)
-            
+
             # Clean up
             if os.path.exists(temp_extract_path):
                 shutil.rmtree(temp_extract_path)
             if os.path.exists(zip_path):
                 os.remove(zip_path)
-                
+
             print("🗑️ Cleaned up temporary files.")
             print("✅ AntelopeV2 model installed correctly.")
-            
+
         except Exception as e:
             print(f"❌ Failed to download/extract AntelopeV2: {e}")
     else:
@@ -185,7 +187,7 @@ def install_custom_nodes():
             "name": "rgthree-comfy",
             "requirements": True
         },
-        { # we already have insightface so don't need requirements (for dlib)
+        {  # we already have insightface so don't need requirements (for dlib)
             "repo": "https://github.com/cubiq/ComfyUI_FaceAnalysis",
             "name": "ComfyUI_FaceAnalysis",
             "requirements": False

run.py

@@ -4,6 +4,7 @@ import subprocess
 COMFYUI_PATH = "./ComfyUI"
 PORT = 8000
 
+
 def run_comfyui():
     """Launch ComfyUI with external access."""
     os.chdir(COMFYUI_PATH)
@@ -11,6 +12,8 @@ def run_comfyui():
 
     subprocess.run(f"python main.py --listen 0.0.0.0 --port {PORT} --disable-auto-launch", shell=True)
 
+
 if __name__ == "__main__":
     run_comfyui()
-    print(f"Now run port-forwarding\nssh -L {PORT}:localhost:{PORT} root@[IP_ADDRESS] -p [RUNPOD_PORT] -i ~/.ssh/[PRIVATE_KEY_NAME]")
+    print(
+        f"Now run port-forwarding\nssh -L {PORT}:localhost:{PORT} root@[IP_ADDRESS] -p [RUNPOD_PORT] -i ~/.ssh/[PRIVATE_KEY_NAME]")