diff --git a/docs/features/image-generation-and-editing/comfyui.md b/docs/features/image-generation-and-editing/comfyui.md
index ec63a52..214dc9d 100644
--- a/docs/features/image-generation-and-editing/comfyui.md
+++ b/docs/features/image-generation-and-editing/comfyui.md
@@ -84,6 +84,8 @@ Once you have ComfyUI installed and running, you can connect it to Open WebUI fr
 5. **Save Configuration:**
     - Click the **Save** button at the bottom of the page to finalize the configuration. You can now use ComfyUI for image generation in Open WebUI.
 
+![Screenshot of an image being generated in the chat using ComfyUI.](/images/image-generation-and-editing/comfyui-create-image-in-chat.png)
+
 ## Edit Image
 
 Open WebUI also supports image editing through ComfyUI, allowing you to modify existing images.
@@ -100,6 +102,8 @@ Open WebUI also supports image editing through ComfyUI, allowing you to modify e
 
     ![Screenshot of the Open WebUI Images settings page with ComfyUI selected for image editing.](/images/image-generation-and-editing/comfyui-editing-settings.png)
 
+![Screenshot of an image being edited in the chat using ComfyUI.](/images/image-generation-and-editing/comfyui-edit-image-in-chat.png)
+
 
 ### Deeper Dive: Mapping ComfyUI Nodes to Open WebUI
 
diff --git a/static/images/image-generation-and-editing/comfyui-create-image-in-chat.png b/static/images/image-generation-and-editing/comfyui-create-image-in-chat.png
new file mode 100644
index 0000000..9152ce0
Binary files /dev/null and b/static/images/image-generation-and-editing/comfyui-create-image-in-chat.png differ
diff --git a/static/images/image-generation-and-editing/comfyui-edit-image-in-chat.png b/static/images/image-generation-and-editing/comfyui-edit-image-in-chat.png
new file mode 100644
index 0000000..4df2532
Binary files /dev/null and b/static/images/image-generation-and-editing/comfyui-edit-image-in-chat.png differ