ComfyUI Model Download Manager

A robust, feature-rich Java Swing application designed to automate the process of identifying and downloading missing models for ComfyUI workflows. It bridges the gap between sharing a JSON workflow and having a fully functional local setup.

🚀 Key Features

🔍 Advanced Workflow Extraction

• JSON & PNG Support: Directly load .json workflows or drop ComfyUI-generated .png files to extract embedded workflow and prompt metadata (tEXt, iTXt, zTXt chunks).
• Recursive Traversal: Scans deeply nested structures, including subgraphs and custom node definitions.
• Smart Detection: Uses regex and node-type analysis to find model files (.safetensors, .ckpt, etc.) and infer their target directory (e.g., checkpoints, loras, vae).

🤖 AI-Powered Discovery

• Local AI Heuristics: Built-in scoring engine to predict model providers (e.g., StabilityAI, Black Forest Labs) offline based on filenames.
• Gemini Deep Search: Integrated "Deep Search" using the Google Gemini API to find official Hugging Face repositories and direct download links for obscure models.
• Context Awareness: Disambiguates generic filenames (like ae.safetensors) by analyzing the global workflow context (e.g., detecting FLUX or SD3 usage).

📥 Powerful Download Manager

• Resume Capability: Implements HTTP Range requests to continue interrupted downloads exactly where they left off.
• Queue Control: Start, Pause, Resume, and Stop functionality for full control over your bandwidth.
• Overwrite Protection: Interactive dialogs prevent accidental data loss by asking to Resume, Overwrite, or Skip existing files.
• Live Metrics: Real-time display of download speed (MB/s), percentage progress, and a global progress bar.

📂 Intelligent Sorting

• Automatic Path Mapping: Infers the correct subdirectory (e.g., checkpoints, loras, vae, controlnet) based on node types like LoraLoader or VAELoader.
• Customizable Base Path: Easily set your ComfyUI models root directory.

🛠️ Requirements

• Java 11 or higher
• Maven (for building from source)
• API Key (Optional): A Google Gemini API key for "Deep Search" functionality.

🏗️ Build & Run

1. Build the Fat JAR:

   mvn clean package
   

2. Launch:

   java -jar target/comfymodeldownloader-1.0-SNAPSHOT.jar
   

🏁 First Start & Model List Setup

Upon your first launch, it is highly recommended to import a master model list. This list helps the application immediately recognize models and their correct download sources.

1. Prepare your JSON: Create or obtain a model-list.json file. It should follow this structure:

   {
     "models": [
       {
         "name": "SDXL Base 1.0",
         "filename": "sd_xl_base_1.0.safetensors",
         "type": "checkpoints",
         "url": "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors"
       }
     ]
   }
   

2. Import: Click the Import Model List (JSON)... button in the workflow panel.
3. Verification: Once imported, the application will use this list as a primary reference for all future workflow analyses.

📖 How to Use

1. Setup Path: Enter or browse to your local ComfyUI models directory.
2. Provide Workflow: Paste JSON content, load a .json file, or load a .png with embedded metadata.
3. Analyze: Click Analyze Models. The application will identify missing models and categorize them.
4. Deep Search: For unknown models, use the "Deep Search" feature to let the AI find the source.
5. Download: Select models and click Start Queue.

📜 License

This project is licensed under the MIT License.

---

Developed to make ComfyUI workflow sharing seamless.