This repo contains the code from my YouTube tutorial on building a Python API to connect Gradio and Comfy UI for AI image generation with Stable Diffusion models.
Here is the video walking through the code in this repo:
YouTube link to tutorial video
This project shows:
- How to connect a Gradio front-end interface to a Comfy UI backend
- Sending workflow data as API requests
- Updating generation parameters dynamically
- Displaying generated images in Gradio
- Adding text and image inputs
- Using a smartphone camera for image inputs
By the end, you'll understand the basics of building a Python API and connecting a user interface with an AI workflow.
To run the code:
- Clone the repo
- Install dependencies (
pip install requests Pillow gradio numpy
) - Modify the Comfy UI installation path
- Open
python app.py
and modify theINPUT_DIR
andOUTPUT_DIR
folder path - Run
python app.py
- Open the localhost link to view in Gradio interface
The main files:
app.py
- Contains Gradio UI and API logicworkflow_api.json
- Saved Comfy UI workflow
Let me know if you have any other questions!
Credit to my YouTube channel Code Crafters Corner for the original tutorial. Please subscribe there for more AI/coding videos.