AI Drawing ChatBot
This is an AI image generator Discord bot written in Python. It has a chatbot that uses an AI language model!
This project uses various APIs, which allows it to run on less powerful computers.
DISCLAIMER: ChatGPT uses your Poe account. I am not responsible if your Poe account gets banned, since using Poe in programs is against their TOS.
Broken stuff
- Using Imaginepy
- ChatGPT
Table of Contents
- Features
- Notes
- Prerequisites
- Windows Setup
- Docker Setup
- Linux/macOS Setup
- Manual Configuration
- Generate images faster with Stable Horde
Features
- ChatGPT/GPT4All chatbot that is integrated with Stable Diffusion
- 3 slash commands for generating images
- Stable Diffusion
- SDXL Beta
- Pollinations
- Anything Diffusion
- And more!
Notes
- ChatGPT is the fastest model that you can use.
- GPT4All isn't nearly as "smart" as ChatGPT.
- Using SDXL sends your generated images to improve the model.
- While generating a response with GPT4All, the Discord bot will lock up. No commands can be run until the bot finishes writing.
- 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 may not work as expected if multiple users are chatting with it at once.
- When using Replit, GPT4All cannot be used.
- When using Replit, you must input your .env variables in the
Secrets
button in the sidebar.
Prerequisites
This project assumes that:
- Python 3.8+ is installed and is on your PATH
- Pip is installed
- Python-venv is installed (only for certain Linux distros)
- Git is installed
- You know how to create a Discord Bot account
Windows Setup
- 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.
- Register for Poe and open inspect tool. Then find your token.
- Clone the Stable Horde module:
git clone https://github.com/mak448a/horde_module --depth=1
- Create venv and install dependencies:
python -m venv venv
venv\Scripts\activate.bat
pip install -r requirements.txt
- 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.
Linux and macOS Setup
- 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.
- Register for Poe and open inspect tool. Then find your token.
- 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.
Docker Setup
- 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.
Manual Configuration
Configuring chat model
Go to config.json
and set the key "model"
to the model you want.
Available Models
- ChatGPT
- GPT4All
You must write the model exactly as it is written here. When using ChatGPT, you must sign up for a Poe account.
When you are done, the edited line should look like this:
"model": "ChatGPT",
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
Generate images faster with Stable Horde
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.