Display ComfyUI client id in the connection window #2059

Author: Acly

ai_diffusion/comfy_client.py

@@ -85,7 +85,7 @@ def __init__(self, url):
         self.url = url
         self.models = ClientModels()
         self._requests = RequestManager()
-        self._id = str(uuid.uuid4())
+        self._id = settings.comfyui_client_id
         self._active: Optional[JobInfo] = None
         self._features: ClientFeatures = ClientFeatures()
         self._supported_archs: dict[Arch, list[ResourceId]] = {}

ai_diffusion/settings.py

@@ -2,6 +2,7 @@
 from dataclasses import dataclass
 import os
 import json
+import uuid
 from enum import Enum
 from pathlib import Path
 from typing import NamedTuple, Optional, Any
@@ -208,6 +209,9 @@ class Settings(QObject):
     check_server_resources: bool
     _check_server_resources = Setting("Refuse connection if nodes or models are missing", True)
 
+    comfyui_client_id: str
+    _comfyui_client_id = Setting("ComfyUI client ID", str(uuid.uuid4()))
+
     selection_grow: int
     _selection_grow = Setting(
         _("Selection Grow"), 5, _("Selection area is expanded by a fraction of its size")

ai_diffusion/ui/settings.py

@@ -19,6 +19,7 @@
     QWidget,
     QMessageBox,
     QCheckBox,
+    QToolButton,
 )
 from PyQt5.QtCore import Qt, QMetaObject, QSize, QUrl, pyqtSignal
 from PyQt5.QtGui import QDesktopServices, QGuiApplication, QCursor, QFontMetrics
@@ -259,6 +260,21 @@ def __init__(self, server: Server):
         )
         self._supported_workloads.setOpenExternalLinks(True)
 
+        self._client_id = QWidget(self._connection_widget)
+        client_id_layout = QHBoxLayout(self._client_id)
+        client_id_layout.setContentsMargins(0, 0, 0, 0)
+        self._client_id_label = QLabel(self._client_id)
+        self._client_id_label.setStyleSheet(f"font-style: italic; color: {grey};")
+        self._client_id_button = QToolButton(self._client_id)
+        self._client_id_button.setToolButtonStyle(Qt.ToolButtonStyle.ToolButtonIconOnly)
+        self._client_id_button.setIcon(Krita.instance().icon("edit-copy"))
+        self._client_id_button.setToolTip(_("Copy to clipboard"))
+        self._client_id_button.setAutoRaise(True)
+        self._client_id_button.clicked.connect(self._copy_client_id)
+        client_id_layout.addWidget(self._client_id_label)
+        client_id_layout.addWidget(self._client_id_button)
+        client_id_layout.addStretch()
+
         anchor = _("View log files")
         open_log_button = QLabel(f"<a href='file://{util.log_dir}'>{anchor}</a>", self)
         open_log_button.setToolTip(str(util.log_dir))
@@ -270,6 +286,7 @@ def __init__(self, server: Server):
 
         connection_layout.addLayout(status_layout)
         connection_layout.addWidget(self._supported_workloads)
+        connection_layout.addWidget(self._client_id)
         connection_layout.addStretch()
 
         self._layout.addWidget(self._server_managed)
@@ -336,9 +353,12 @@ def update_server_status(self):
         connection = root.connection
         self._cloud_widget.update_connection_state(connection.state)
         self._connect_button.setEnabled(connection.state != ConnectionState.connecting)
+        self._client_id.setVisible(False)
         if connection.state == ConnectionState.connected:
             self._connection_status.setText(_("Connected"))
             self._connection_status.setStyleSheet(f"color: {green}; font-weight:bold")
+            self._client_id_label.setText(f"Client ID: {settings.comfyui_client_id}")
+            self._client_id.setVisible(True)
         elif connection.state == ConnectionState.connecting:
             self._connection_status.setText(_("Connecting"))
             self._connection_status.setStyleSheet(f"color: {yellow}; font-weight:bold")
@@ -411,6 +431,10 @@ def model_name(id: ResourceId, with_file=False):
         self._supported_workloads.setStyleSheet(style)
         self._supported_workloads.setText(text)
 
+    def _copy_client_id(self):
+        if clipboard := QGuiApplication.clipboard():
+            clipboard.setText(settings.comfyui_client_id)
+
     def _open_logs(self):
         QDesktopServices.openUrl(QUrl.fromLocalFile(str(util.log_dir)))