This is a completely free and open source AI image generator Discord bot written in Python. It has a Llama 3 based chatbot that you can ask to draw pictures! This project uses various free APIs, which allows it to run on less powerful computers.
Note: If this project is not https://github.com/mak448a/AI-Drawing-Chatbot, then it is unofficial. Please visit the official repo.
- Features
- Notes
- Prerequisites
- Windows Setup
- Docker Setup
- Linux/macOS Setup
- Manual Configuration
- Generate images faster with Stable Horde
- AI chatbot which is integrated with Stable Diffusion
- 2 slash commands for generating images
- Stable Diffusion
- SDXL Beta
- Pollinations
- Anything Diffusion
- And more!
- Llama3 is the recommended model.
- When using SDXL-Beta it sends your generated images to improve the model.
- Do not input any personal information on the
/imagine_poly
command because your generated image will be displayed on their official frontpage. - Please do not enter any personal information in the chatbot or in the image generators, because your prompts are sent to various providers.
- When you use Stable Horde to generate images, your prompts are sent to Stable Horde, as listed in their privacy policy.
- The chatbot will not work as expected if multiple users are chatting with it at once.
This project assumes that:
- Python 3.8+ is installed and is on your PATH
- Pip is installed
- Git is installed
- You know how to create a Discord Bot account
- The package python-venv is installed (only for certain Linux distros)
- Create a Discord Bot and grab the token. Make sure to switch on Message Content Intent in the
Bot
tab of the Developer Portal. - Grab an API key from Stable Horde at their register page.
- Clone the Stable Horde module:
git clone https://github.com/mak448a/horde_module helper_utils/horde_module --depth=1
- Create venv and install dependencies:
python -m venv venv
venv\Scripts\activate.bat
pip install -r requirements.txt
- Grab your Groq key from here:
https://console.groq.com/keys
. - Run
python run.py
and fill in all the information it asks for. - Invite the bot with the link it provides!
- You can chat with the bot by mentioning it and typing your message.
- Create a Discord Bot and grab the token. Make sure to switch on Message Content Intent in the
Bot
tab of the Developer Portal. - Grab an API key from Stable Horde at their register page.
- Create venv and install dependencies:
python3 -m venv venv && source venv/bin/activate && pip install -r requirements.txt
- Clone the Stable Horde module:
git clone https://github.com/mak448a/horde_module helper_utils/horde_module --depth=1
- Grab your Groq key from here:
https://console.groq.com/keys
. - Run
python3 run.py
and fill in all the information it asks for. - Invite the bot with the link it provides!
- You can chat with the bot by mentioning it and typing your message.
- Make sure you have Docker installed on your machine.
- Clone this repository to your local machine.
- Build the Docker image using the following command:
docker build -t ai-drawing-chatbot .
- Run the Docker container using the following command:
docker run -it --name chatbot ai-drawing-chatbot
- Enter in your token values. Go here to find out how to get these.
If you want to manually configure the program for any reason, here's how.
Configuring chat model
Go to config.json
and set the key "model"
to the model you want.
Available Models
- Llama3
You must write the model exactly as it is written here.
When you are done, the edited line should look like this:
"model": "Llama3",
Disable chatbot
Go to config.json
and set the key "chatbot"
to false.
It should look like this:
"chatbot": false,
Change loading GIF
Go to config.json
and set the key "loading_gif"
to any GIF on Tenor you want!
It should look like this:
"loading_gif": "https://tenor.com/your/favorite/loading/gif",
Turn off command syncing (makes bot load faster)
Please note that this will break the bot if you haven't run it for the first time.
Go to config.json
and set the key "sync"
to false
.
It should look like this:
"sync": false,
Change chatbot drawing model
Go to config.json
and set the key "image_model"
to a model that you find here.
It should look like this:
"image_model": "SDXL 1.0"
Stable Horde uses Kudos to rank users on a queue. The more Kudos you have, the higher you are on the generation queue. You can get Kudos by rating images and/or generating images for Stable Horde on your local hardware. To get more Kudos, first stick your API key here. Then, rate some images generated by other users here. After rating a few images, you should have faster generation times!
IMPORTANT: When an image says, "This rating must be x," you must rate it that. This picture is used as a Captcha to avoid spam.