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

{webtools_cli-1.3.5 → 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.5/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.5
+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.5 → webtools_cli-1.3.6}/pyproject.toml

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

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

@@ -3556,9 +3556,27 @@ 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}")
         

{webtools_cli-1.3.5 → 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.5
+Version: 1.3.6
 Summary: Advanced Web Intelligence & Scraping Toolkit with CLI and Web UI
 Author: Abhinav Adarsh
 License-Expression: MIT