webtools-cli 1.3.4__tar.gz → 1.3.6__tar.gz

{webtools_cli-1.3.4 → webtools_cli-1.3.6}/ComfyUI/ui.py

@@ -14,6 +14,12 @@ MEMORY_PROFILE = "Standard (Auto-Detect)" #@param ["Standard (Auto-Detect)", "Lo
 #@markdown **Visual Settings:**
 LIVE_GENERATION_PREVIEWS = True #@param {type:"boolean"}
 
+# --- Session Configuration ---
+#@markdown **Performance Profile:**
+MEMORY_PROFILE = "Standard (Auto-Detect)" #@param ["Standard (Auto-Detect)", "Low VRAM (T4 GPU / Heavy Models)", "High VRAM (A100 GPU Only)"]
+#@markdown **Visual Settings:**
+LIVE_GENERATION_PREVIEWS = True #@param {type:"boolean"}
+
 # Dynamic Workspace detection
 WORKSPACE = os.environ.get("COMFYUI_WORKSPACE")
 if not WORKSPACE:
@@ -29,6 +35,17 @@ if not WORKSPACE:
 
 ARGS = []
 
+# GPU Detection & Fallback
+try:
+    import torch
+    if not torch.cuda.is_available():
+        print("[WARNING] NVIDIA GPU not detected by Torch. Switching to CPU mode...")
+        ARGS.append("--cpu")
+except ImportError:
+    # If torch is not even installed yet (unlikely on Colab), we'll assume CPU for now
+    # or let the main.py handle its own errors. But adding --cpu is safer.
+    ARGS.append("--cpu")
+
 if "Low VRAM" in MEMORY_PROFILE:
     ARGS.append("--lowvram")
 elif "High VRAM" in MEMORY_PROFILE:

{webtools_cli-1.3.4/webtools_cli.egg-info → webtools_cli-1.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: webtools-cli
-Version: 1.3.4
+Version: 1.3.6
 Summary: Advanced Web Intelligence & Scraping Toolkit with CLI and Web UI
 Author: Abhinav Adarsh
 License-Expression: MIT

{webtools_cli-1.3.4 → webtools_cli-1.3.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "webtools-cli"
-version = "1.3.4"
+version = "1.3.6"
 description = "Advanced Web Intelligence & Scraping Toolkit with CLI and Web UI"
 readme = "README.md"
 license = "MIT"

{webtools_cli-1.3.4 → webtools_cli-1.3.6}/webtools/core.py

@@ -3528,12 +3528,18 @@ def run_comfyui_mode():
         input("\nPress Enter to return to main menu...")
         return
 
-    comfy_dir = find_comfy_dir()
-    if not comfy_dir:
-        print(f"\n{Fore.RED}❌ ERROR: ComfyUI folder not found!{Style.RESET_ALL}")
+    script_dir = find_comfy_dir()
+    if not script_dir:
+        print(f"\n{Fore.RED}❌ ERROR: ComfyUI scripts not found!{Style.RESET_ALL}")
         input("\nPress Enter to return to main menu...")
         return
 
+    # Define the actual ComfyUI repository workspace
+    if on_colab:
+        workspace = "/content/ComfyUI"
+    else:
+        workspace = os.path.join(script_dir, "ComfyUI_repo")
+
     while True:
         os.system('cls' if os.name == 'nt' else 'clear')
         print(f"\n{Fore.CYAN}--- ComfyUI Launcher ---{Style.RESET_ALL}")
@@ -3550,22 +3556,39 @@ def run_comfyui_mode():
             continue
 
         if on_colab:
-            print(f"\n{Fore.YELLOW}⚠️  Google Colab Detected!{Style.RESET_ALL}")
-            print(f"{Fore.CYAN}👉 IMPORTANT: Switch to T4 GPU (Edit -> Notebook settings -> T4 GPU).{Style.RESET_ALL}")
-            time.sleep(2)
+            while True:
+                has_gpu = check_gpu()
+                if not has_gpu:
+                    os.system('cls' if os.name == 'nt' else 'clear')
+                    print(f"\n{Fore.RED}⚠️  GPU NOT DETECTED ON COLAB!{Style.RESET_ALL}")
+                    print(f"\n{Fore.CYAN}👉 PLEASE FIX THIS NOW:{Style.RESET_ALL}")
+                    print(f"   1. Go to {Fore.YELLOW}Runtime -> Change runtime type{Fore.CYAN}")
+                    print(f"   2. Select {Fore.YELLOW}T4 GPU{Fore.CYAN}")
+                    print(f"   3. Click {Fore.YELLOW}Save{Fore.CYAN}")
+                    print(f"\n{Fore.WHITE}After switching, press Enter to retry or type 'cpu' to force slow CPU mode.{Style.RESET_ALL}")
+                    
+                    user_resp = input(f"\n{Fore.LIGHTGREEN_EX}> {Style.RESET_ALL}").strip().lower()
+                    if user_resp == 'cpu':
+                        print(f"\n{Fore.YELLOW}⚠️  Forcing CPU mode... this will be extremely slow!{Style.RESET_ALL}")
+                        time.sleep(2)
+                        break
+                    continue
+                else:
+                    print(f"\n{Fore.GREEN}✅ T4 GPU Detected! Proceeding...{Style.RESET_ALL}")
+                    time.sleep(1.5)
+                    break
 
         print(f"\n{Fore.GREEN}🚀 Starting ComfyUI Setup & Session...{Style.RESET_ALL}")
         
         # 0. Run System Initialization (comfyUI.py)
-        # This ensures the ComfyUI repo is cloned and dependencies are installed
-        init_script = os.path.join(comfy_dir, "comfyUI.py")
+        init_script = os.path.join(script_dir, "comfyUI.py")
         if os.path.exists(init_script):
             print(f"\n{Fore.CYAN}--- Running System Initialization ---{Style.RESET_ALL}")
             init_env = os.environ.copy()
-            init_env["COMFYUI_WORKSPACE"] = comfy_dir
+            init_env["COMFYUI_WORKSPACE"] = workspace
             subprocess.run([sys.executable, init_script], env=init_env)
         else:
-            print(f"{Fore.YELLOW}Warning: {init_script} not found. Skipping initialization.{Style.RESET_ALL}")
+            print(f"{Fore.YELLOW}Warning: {init_script} not found.{Style.RESET_ALL}")
 
         # 1. Handle Model Downloads if Anime chosen
         if cui_choice == '2':
@@ -3574,22 +3597,19 @@ def run_comfyui_mode():
             download_env["UNET_DIFFUSION_URLS"] = "https://huggingface.co/circlestone-labs/Anima/resolve/main/split_files/diffusion_models/anima-preview3-base.safetensors"
             download_env["TEXT_ENCODER_URLS"] = "https://huggingface.co/circlestone-labs/Anima/resolve/main/split_files/text_encoders/qwen_3_06b_base.safetensors"
             download_env["VAE_URLS"] = "https://huggingface.co/circlestone-labs/Anima/resolve/main/split_files/vae/qwen_image_vae.safetensors"
-            download_env["COMFYUI_WORKSPACE"] = comfy_dir
+            download_env["COMFYUI_WORKSPACE"] = workspace
             
-            # Run comfyu.py with these env vars
-            downloader_script = os.path.join(comfy_dir, "comfyu.py")
+            downloader_script = os.path.join(script_dir, "comfyu.py")
             if os.path.exists(downloader_script):
                 subprocess.run([sys.executable, downloader_script], env=download_env)
-            else:
-                print(f"{Fore.RED}Error: {downloader_script} not found.{Style.RESET_ALL}")
 
         # 2. Launch UI
-        ui_script = os.path.join(comfy_dir, "ui.py")
+        ui_script = os.path.join(script_dir, "ui.py")
         if os.path.exists(ui_script):
             print(f"\n{Fore.CYAN}--- Initializing ComfyUI Interface ---{Style.RESET_ALL}")
             try:
                 ui_env = os.environ.copy()
-                ui_env["COMFYUI_WORKSPACE"] = comfy_dir
+                ui_env["COMFYUI_WORKSPACE"] = workspace
                 subprocess.run([sys.executable, ui_script], env=ui_env)
             except Exception as e:
                 print(f"{Fore.RED}Error running ComfyUI: {e}{Style.RESET_ALL}")

{webtools_cli-1.3.4 → webtools_cli-1.3.6/webtools_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: webtools-cli
-Version: 1.3.4
+Version: 1.3.6
 Summary: Advanced Web Intelligence & Scraping Toolkit with CLI and Web UI
 Author: Abhinav Adarsh
 License-Expression: MIT