Metadata-Version: 2.4
Name: iflow-mcp_orion4d-comfyui_mcp
Version: 1.0.4
Summary: MCP server for ComfyUI - enables ChatGPT to control ComfyUI workflows
License-File: LICENSE
Requires-Python: >=3.8
Requires-Dist: fastapi>=0.104.0
Requires-Dist: fastmcp>=0.1.0
Requires-Dist: google-genai>=0.3.0
Requires-Dist: gradio>=4.0.0
Requires-Dist: httpx>=0.24.0
Requires-Dist: openai>=1.0.0
Requires-Dist: python-dotenv>=1.0.0
Requires-Dist: requests>=2.31.0
Requires-Dist: uvicorn[standard]>=0.24.0
Requires-Dist: websocket-client>=1.0.0
Description-Content-Type: text/markdown

# 🧠 Serveur MCP pour ComfyUI (Version Alpha-1)

<img width="2208" height="1235" alt="image" src="https://github.com/user-attachments/assets/11c256ce-f054-49b0-a8c8-52a1675968b7" />

Ce projet expose **ComfyUI** via un serveur compatible **MCP (Model Context Protocol)**.

Il permet :  
- de piloter **ComfyUI** depuis ChatGPT via un connecteur (mode dev) ;  
- de piloter l’interface **ComfyUI** dans Chrome via une extension WebSocket.

<img width="517" height="645" alt="image" src="https://github.com/user-attachments/assets/c3dbd461-626f-4ece-b336-b3367e386454" />

### Pour bientôt : en local, interfaces Gradio pour Ollama et Google CLI

---

## ⚙️ Installation

### 1️⃣ Cloner le dépôt
```bash
git clone https://github.com/orion4d/ComfyUI_mcp.git
```

### 2️⃣ Créer l’environnement virtuel
```bash
python -m venv venv
# Linux / Mac
source venv/bin/activate
# Windows
venv\Scripts\activate
```

### 3️⃣ Installer les dépendances
```bash
pip install -r requirements.txt
```

---

## 🔐 Génération des clés et configuration

```bash
python generate_key.py
```

Ce script :
- génère **MCP_API_KEY** (pour ChatGPT)  
- génère **WEBSOCKET_TOKEN** (pour l’extension Chrome)  
- crée automatiquement `.env` et `.env.example`

---

## 🚀 Démarrage du serveur

```bash
python server.py
```

Le serveur démarre par défaut sur :  
`http://127.0.0.1:8000`

---

## 🌐 Points d’accès

### 🔧 MCP Tools exposés
- `list_workflows`, `save_workflow`, `load_workflow`  
- `read_custom_node`, `write_custom_node`  
- `queue_prompt`, `get_history`  
- `create_custom_node_template`, `list_custom_subdir`  
- `ui_click_element`, `ui_fill_input`, `ui_get_current_workflow`

### 🧩 Routes Debug
- `/debug/health` → infos système, versions, outils  
- `/ws` → WebSocket pour l’extension Chrome

---

## 🧱 Exemple d’usage

### Depuis ChatGPT (Custom GPT)
- Authentification : `X-API-Key` → ta clé **MCP_API_KEY**  
- Appels possibles : `list_workflows`, `queue_prompt`, `read_custom_node`, etc.

### Depuis Chrome (Extension MCP)
- URL WebSocket : `ws://127.0.0.1:8000/ws`  
- Token : **WEBSOCKET_TOKEN**

---

## 🧠 Intégration ComfyUI

Le client (`ComfyUIClient`) communique via HTTP avec ton ComfyUI local :  
- URL : `http://127.0.0.1:8188`  
- Support des workflows UI et API  
- Conversion automatique via `_convert_ui_to_api()`

### Tester la connexion
```bash
curl http://127.0.0.1:8000/debug/health
```

---

# 📘 Commandes MCP–ComfyUI

## 🗂️ Arborescence des commandes
```text
ComfyUI
│
├── 🧠 Exécution (moteur)
│   ├─ /queue_prompt
│   ├─ /get_queue_status
│   ├─ /cancel_prompt
│   ├─ /get_history
│   └─ /interrupt_execution
│
├── ⚙️ Système & Modèles
│   ├─ /get_system_stats
│   ├─ /list_models
│   └─ /model_info
│
├── 🧩 Workflows
│   ├─ /save_workflow
│   ├─ /load_workflow
│   ├─ /list_workflows
│   └─ /inspect_workflow
│
├── 🔧 Custom Nodes (→ ComfyUI/custom_nodes/)
│   ├─ /create_custom_node_template
│   ├─ /write_custom_node
│   ├─ /read_custom_node
│   ├─ /list_custom_subdir
│   └─ /autodoc_nodes
│
├── 🖼️ Images
│   ├─ /upload_image
│   ├─ /get_image
│   └─ /list_output_images
│
└── 📂 MCP_exchange (→ output/MCP_exchange/)
    ├─ /list_exchange
    ├─ /read_exchange
    ├─ /write_exchange
    └─ /delete_exchange
```

---

## 🧠 Exécution & File
- **/queue_prompt** → exécuter un workflow  
- **/get_queue_status** → état de la file  
- **/get_history** → historique d’un prompt  
- **/cancel_prompt** → annuler un prompt  
- **/interrupt_execution** → stopper tout en cours  

## ⚙️ Système & Modèles
- **/get_system_stats** → infos GPU, RAM, versions  
- **/list_models** → lister les modèles disponibles  
- **/model_info** → détails d’un modèle  

## 🧩 Workflows
- **/save_workflow** → enregistrer un workflow  
- **/load_workflow** → charger un workflow  
- **/list_workflows** → lister tous les workflows  
- **/inspect_workflow** → analyser la structure  

## 🖼️ Images & Fichiers
- **/list_output_images** → voir les images produites  
- **/get_image** → récupérer une image  
- **/upload_image** → envoyer une image d’entrée  

## 🔧 Custom Nodes
- **/create_custom_node_template** → créer un squelette de node  
- **/write_custom_node** → écrire un fichier node  
- **/read_custom_node** → lire le code d’un node  
- **/list_custom_subdir** → explorer un dossier custom  
- **/autodoc_nodes** → générer la doc de tous les custom nodes  

## 🖥️ Interface (Chrome UI)
- **/ui_click_element** → simuler un clic  
- **/ui_fill_input** → remplir un champ texte  
- **/ui_get_current_workflow** → récupérer le workflow affiché  

##

## 📂 Structure MCP_exchange

Ce répertoire sert d’espace d’échange entre **MCP–ComfyUI** et ton environnement local.  
Toutes les commandes ci-dessous interagissent uniquement avec le dossier :  
`output/MCP_exchange/`

## 🔍 Lister les fichiers
```bash
call_tool /MCP-ComfyUI/.../list_exchange {"limit": 200, "exts": "png,jpg,jpeg,webp,bmp,tif,tiff,txt,md,html,htm,json,js,py,css"}
```

## 📖 Lire un fichier
```bash
call_tool /MCP-ComfyUI/.../read_exchange {"name": "nom_du_fichier.txt", "as_data_url": true}
```

## ✏️ Écrire un fichier
```bash
call_tool /MCP-ComfyUI/.../write_exchange {"name": "nouveau_fichier.md", "content": "contenu du fichier", "mode": "text", "overwrite": true}
```
Modes disponibles : `text`, `base64`, `data_url`.

## ❌ Supprimer un fichier
```bash
call_tool /MCP-ComfyUI/.../delete_exchange {"name": "fichier_a_supprimer.json"}
```

## 🧭 Usages typiques
- Exporter un résultat ou une image générée pour inspection.  
- Importer un script, un JSON de workflow ou un dataset.  
- Automatiser des échanges entre MCP et ComfyUI.  

**Chemin complet :** `ComfyUI/output/MCP_exchange/`  
> Les autres commandes (workflows, modèles, nœuds) agissent ailleurs ; ce groupe-ci se limite à la gestion des fichiers d’échange.
---
