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

{webtools_cli-1.3.5 → webtools_cli-1.3.7}/ComfyUI/comfyu.py

@@ -10,26 +10,31 @@ WORKSPACE = os.environ.get("COMFYUI_WORKSPACE", "/content/ComfyUI")
 CHECKPOINT_URLS = os.environ.get("CHECKPOINT_URLS", "")
 UNET_DIFFUSION_URLS = os.environ.get("UNET_DIFFUSION_URLS", "")
 TEXT_ENCODER_URLS = os.environ.get("TEXT_ENCODER_URLS", "")
+TEXT_ENCODERS_URLS = os.environ.get("TEXT_ENCODERS_URLS", "")
 CLIP_VISION_URLS = os.environ.get("CLIP_VISION_URLS", "")
 VAE_URLS = os.environ.get("VAE_URLS", "")
 LORA_URLS = os.environ.get("LORA_URLS", "")
 CONTROLNET_URLS = os.environ.get("CONTROLNET_URLS", "")
 UPSCALE_MODELS_URLS = os.environ.get("UPSCALE_MODELS_URLS", "")
+LATENT_UPSCALE_URLS = os.environ.get("LATENT_UPSCALE_URLS", "")
 EMBEDDING_URLS = os.environ.get("EMBEDDING_URLS", "")
 CUSTOM_NODE_URLS = os.environ.get("CUSTOM_NODE_URLS", "")
 
 # --- Downloader Logic ---
 DIRS = {
-    "checkpoints":    os.path.join(WORKSPACE, "models/checkpoints"),
-    "unet":           os.path.join(WORKSPACE, "models/unet"),
-    "clip":           os.path.join(WORKSPACE, "models/clip"),
-    "clip_vision":    os.path.join(WORKSPACE, "models/clip_vision"),
-    "vae":            os.path.join(WORKSPACE, "models/vae"),
-    "loras":          os.path.join(WORKSPACE, "models/loras"),
-    "controlnet":     os.path.join(WORKSPACE, "models/controlnet"),
-    "upscale_models": os.path.join(WORKSPACE, "models/upscale_models"),
-    "embeddings":     os.path.join(WORKSPACE, "models/embeddings"),
-    "custom_nodes":   os.path.join(WORKSPACE, "custom_nodes")
+    "checkpoints":      os.path.join(WORKSPACE, "models/checkpoints"),
+    "unet":             os.path.join(WORKSPACE, "models/unet"),
+    "diffusion_models": os.path.join(WORKSPACE, "models/diffusion_models"),
+    "clip":             os.path.join(WORKSPACE, "models/clip"),
+    "text_encoders":    os.path.join(WORKSPACE, "models/text_encoders"),
+    "clip_vision":      os.path.join(WORKSPACE, "models/clip_vision"),
+    "vae":              os.path.join(WORKSPACE, "models/vae"),
+    "loras":            os.path.join(WORKSPACE, "models/loras"),
+    "controlnet":       os.path.join(WORKSPACE, "models/controlnet"),
+    "upscale_models":   os.path.join(WORKSPACE, "models/upscale_models"),
+    "latent_upscale":   os.path.join(WORKSPACE, "models/latent_upscale_models"),
+    "embeddings":       os.path.join(WORKSPACE, "models/embeddings"),
+    "custom_nodes":     os.path.join(WORKSPACE, "custom_nodes")
 }
 
 def get_filename(url, response):
@@ -107,13 +112,15 @@ def process_downloads(urls_str, target_dir, is_node=False):
 if __name__ == "__main__":
     # --- Execution ---
     process_downloads(CHECKPOINT_URLS,     DIRS["checkpoints"])
-    process_downloads(UNET_DIFFUSION_URLS, DIRS["unet"])
+    process_downloads(UNET_DIFFUSION_URLS, DIRS["diffusion_models"])
     process_downloads(TEXT_ENCODER_URLS,   DIRS["clip"])
+    process_downloads(TEXT_ENCODERS_URLS,  DIRS["text_encoders"])
     process_downloads(CLIP_VISION_URLS,    DIRS["clip_vision"])
     process_downloads(VAE_URLS,            DIRS["vae"])
     process_downloads(LORA_URLS,           DIRS["loras"])
     process_downloads(CONTROLNET_URLS,     DIRS["controlnet"])
     process_downloads(UPSCALE_MODELS_URLS, DIRS["upscale_models"])
+    process_downloads(LATENT_UPSCALE_URLS,   DIRS["latent_upscale"])
     process_downloads(EMBEDDING_URLS,      DIRS["embeddings"])
     process_downloads(CUSTOM_NODE_URLS,    DIRS["custom_nodes"], is_node=True)
 

{webtools_cli-1.3.5 → webtools_cli-1.3.7}/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.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: webtools-cli
-Version: 1.3.5
+Version: 1.3.7
 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.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "webtools-cli"
-version = "1.3.5"
+version = "1.3.7"
 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.7}/webtools/core.py

@@ -3545,6 +3545,7 @@ def run_comfyui_mode():
         print(f"\n{Fore.CYAN}--- ComfyUI Launcher ---{Style.RESET_ALL}")
         print(f"  {Fore.CYAN}[1]{Style.RESET_ALL} Launch Standard ComfyUI")
         print(f"  {Fore.CYAN}[2]{Style.RESET_ALL} Launch Anime ComfyUI (Anima Model)")
+        print(f"  {Fore.CYAN}[3]{Style.RESET_ALL} Launch Text-to-Video (LTX-2.3)")
         print(f"  {Fore.RED}[b]{Style.RESET_ALL} Back to main menu")
         
         cui_choice = input(f"\n{Fore.LIGHTGREEN_EX}> {Style.RESET_ALL}").strip().lower()
@@ -3552,53 +3553,141 @@ def run_comfyui_mode():
         if cui_choice == 'b':
             return
             
-        if cui_choice not in ['1', '2']:
+        if cui_choice not in ['1', '2', '3']:
             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)
         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"] = workspace
-            subprocess.run([sys.executable, init_script], env=init_env)
-        else:
-            print(f"{Fore.YELLOW}Warning: {init_script} not found.{Style.RESET_ALL}")
+            init_env["UPDATE_COMFY_UI"] = "false"
+            with MoonSpinner("🚀 Setting up ComfyUI"):
+                subprocess.run([sys.executable, init_script], env=init_env,
+                               stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
-        # 1. Handle Model Downloads if Anime chosen
+        # 1. Handle Model Downloads if Anime or T2V chosen
         if cui_choice == '2':
-            print(f"\n{Fore.MAGENTA}✨ Configuring Anime Model (Anima)...{Style.RESET_ALL}")
             download_env = os.environ.copy()
             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"] = workspace
-            
             downloader_script = os.path.join(script_dir, "comfyu.py")
             if os.path.exists(downloader_script):
-                subprocess.run([sys.executable, downloader_script], env=download_env)
+                with MoonSpinner("✨ Downloading Anima models"):
+                    subprocess.run([sys.executable, downloader_script], env=download_env,
+                                   stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
 
-        # 2. Launch UI
+        elif cui_choice == '3':
+            download_env = os.environ.copy()
+            download_env["UNET_DIFFUSION_URLS"] = "https://huggingface.co/unsloth/LTX-2.3-GGUF/resolve/main/ltx-2.3-22b-dev-Q4_0.gguf"
+            download_env["VAE_URLS"] = "\n".join([
+                "https://huggingface.co/unsloth/LTX-2.3-GGUF/resolve/main/vae/ltx-2.3-22b-dev_audio_vae.safetensors",
+                "https://huggingface.co/unsloth/LTX-2.3-GGUF/resolve/main/vae/ltx-2.3-22b-dev_video_vae.safetensors",
+            ])
+            download_env["TEXT_ENCODERS_URLS"] = "https://huggingface.co/unsloth/gemma-3-12b-it-qat-GGUF/resolve/main/gemma-3-12b-it-qat-UD-Q4_K_XL.gguf"
+            download_env["LORA_URLS"] = "https://huggingface.co/Lightricks/LTX-2.3/resolve/main/ltx-2.3-22b-distilled-lora-384.safetensors"
+            download_env["LATENT_UPSCALE_URLS"] = "https://huggingface.co/Lightricks/LTX-2.3/resolve/main/ltx-2.3-spatial-upscaler-x2-1.0.safetensors"
+            download_env["COMFYUI_WORKSPACE"] = workspace
+            downloader_script = os.path.join(script_dir, "comfyu.py")
+            if os.path.exists(downloader_script):
+                with MoonSpinner("🎬 Downloading LTX-2.3 models"):
+                    subprocess.run([sys.executable, downloader_script], env=download_env,
+                                   stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+
+        # 2. Launch UI — show only tunnel URL, wait for /q
         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"] = 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}")
-        else:
+        if not os.path.exists(ui_script):
             print(f"{Fore.RED}Error: {ui_script} not found.{Style.RESET_ALL}")
-        
-        input("\nPress Enter to return to ComfyUI menu...")
+            input("\nPress Enter to return to ComfyUI menu...")
+            continue
+
+        # Write ComfyUI-Manager config to disable auto-update
+        manager_config_dir = os.path.join(workspace, "user", "__manager")
+        os.makedirs(manager_config_dir, exist_ok=True)
+        manager_config_path = os.path.join(manager_config_dir, "config.ini")
+        if not os.path.exists(manager_config_path):
+            with open(manager_config_path, "w") as f:
+                f.write("[default]\nskip_update_check = true\n")
+
+        ui_env = os.environ.copy()
+        ui_env["COMFYUI_WORKSPACE"] = workspace
+
+        comfy_proc = None
+        try:
+            with MoonSpinner("⚙️  Starting ComfyUI"):
+                comfy_proc = subprocess.Popen(
+                    [sys.executable, ui_script],
+                    env=ui_env,
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.STDOUT,
+                    text=True, bufsize=1
+                )
+                # Wait until tunnel URL appears
+                tunnel_url = None
+                for line in comfy_proc.stdout:
+                    if "trycloudflare.com" in line:
+                        for part in line.split():
+                            if "trycloudflare.com" in part and part.startswith("http"):
+                                tunnel_url = part.strip()
+                                break
+                        if tunnel_url:
+                            break
+
+            os.system('cls' if os.name == 'nt' else 'clear')
+            if tunnel_url:
+                print(f"\n{Fore.GREEN}{'='*60}{Style.RESET_ALL}")
+                print(f"{Fore.GREEN}🚀 ComfyUI is LIVE:{Style.RESET_ALL} {Fore.CYAN}{tunnel_url}{Style.RESET_ALL}")
+                print(f"{Fore.GREEN}{'='*60}{Style.RESET_ALL}")
+            else:
+                print(f"\n{Fore.YELLOW}⚠️  ComfyUI started but tunnel URL not found.{Style.RESET_ALL}")
+            print(f"\nType {Fore.RED}/q{Style.RESET_ALL} to stop ComfyUI and return to menu.")
+
+            # Drain stdout in background so process doesn't block
+            def _drain(proc):
+                try:
+                    for _ in proc.stdout: pass
+                except: pass
+            threading.Thread(target=_drain, args=(comfy_proc,), daemon=True).start()
+
+            while True:
+                try:
+                    user_input = input().strip().lower()
+                    if user_input in ['/q', '/quit']:
+                        break
+                except EOFError:
+                    time.sleep(1)
+        except KeyboardInterrupt:
+            pass
+        finally:
+            if comfy_proc and comfy_proc.poll() is None:
+                comfy_proc.terminate()
+                try: comfy_proc.wait(timeout=5)
+                except: comfy_proc.kill()
 
 def main_launcher():
     """Mode selection menu on startup"""

{webtools_cli-1.3.5 → webtools_cli-1.3.7/webtools_cli.egg-info}/PKG-INFO

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