JustImageViewer의 HTTP API를 ComfyUI와 함께 사용하는 방법

ComfyUI의 LoadImage에 이미지를 선택하는 기능이 있지만 느립니다. 이 기능을 개선한 CustomNode도 있습니다만 저는 다른 방법을 시도해보았습니다.

심플한 이미지 뷰어 JustImageViewer를 만들고 외부 연동을 위해 HTTP API를 추가해서 사용하는 방법입니다. 뷰어가 Rust와 imgui-rs로 만들어져 있어서 이미지를 빠르게 보고 선택할 수 있는 장점이 있습니다.

LoadImageFromURL을 사용한 ComfyUI 와 연계

JustImageViewer의 세팅 및 확인

JustImageViewer의 settings.toml 에 http_port 를 적어 웹서버 서빙 기능을 활성화 합니다. 여기서는 8001번을 세팅하도록 하겠습니다. 경로는 다음과 같습니다.

macOS / Linux: ~/.justImageViewer/settings.toml
Windows: %USERPROFILE%/.justImageViewer/settings.toml

settings.toml 중

  http_port = 8001

뷰어를 실행하고 적당한 이미지를 선택한 뒤, 웹브라우저에서 http://localhost:8001/select 를 열어봅니다.

ComfyUI의 CustomNode 세팅

https://github.com/jerrywap/ComfyUI_LoadImageFromHttpURL 의 LoadImageFromURL 노드를 사용하여 URL의 이미지를 불러올 수 있습니다. 하지만 캐싱 문제로 인해 웹 서버에서 서빙하는 이미지가 변경되어도 새로고침되지 않는 경우가 있습니다.

그래서 다음 코드를 추가하여 강제 캐시 무효화 옵션을 추가했습니다. Pull Request 하기에 너무 작은 코드이므로 해당 부분을 gist에 올려두겠습니다. 이것을 적용하면 ComfyUI에서 다음과 같이 보입니다.

위 이미지 처럼 LoadImageFromURL 에 http://localhost:8001/select를 입력하고 캐시를 무효화 한 다음 해당 노드에 주소를 적습니다. 해당 이미지에 Preview Image를 연결해서 이미지가 제대로 적용이 되는지 확인할 수 있습니다. 그리고 ComfyUI의 Workflow를 구성하고 Queue를 실행시키면 됩니다.