Skip to content

An image generator Discord bot written in Python with no GPU requirement! It can be easily deployed to your favorite VPS for hosting.

License

Notifications You must be signed in to change notification settings

mak448a/AI-Drawing-Chatbot

Repository files navigation

AI Drawing Chatbot

Discord Invite

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.

Demo picture

Table of Contents

  1. Features
  2. Notes
  3. Prerequisites
  4. Windows Setup
  5. Docker Setup
  6. Linux/macOS Setup
  7. Manual Configuration
  8. Generate images faster with Stable Horde

Features

  • AI chatbot which is integrated with Stable Diffusion
  • 2 slash commands for generating images
  • Stable Diffusion
  • SDXL Beta
  • Pollinations
  • Anything Diffusion
  • And more!

Notes

  • 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.

Prerequisites

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)

Windows Setup

  1. Create a Discord Bot and grab the token. Make sure to switch on Message Content Intent in the Bot tab of the Developer Portal.
  2. Grab an API key from Stable Horde at their register page.
  3. Clone the Stable Horde module: git clone https://github.com/mak448a/horde_module helper_utils/horde_module --depth=1
  4. Create venv and install dependencies: python -m venv venv venv\Scripts\activate.bat pip install -r requirements.txt
  5. Grab your Groq key from here: https://console.groq.com/keys.
  6. Run python run.py and fill in all the information it asks for.
  7. Invite the bot with the link it provides!
  8. You can chat with the bot by mentioning it and typing your message.

Linux and macOS Setup

  1. Create a Discord Bot and grab the token. Make sure to switch on Message Content Intent in the Bot tab of the Developer Portal.
  2. Grab an API key from Stable Horde at their register page.
  3. Create venv and install dependencies: python3 -m venv venv && source venv/bin/activate && pip install -r requirements.txt
  4. Clone the Stable Horde module: git clone https://github.com/mak448a/horde_module helper_utils/horde_module --depth=1
  5. Grab your Groq key from here: https://console.groq.com/keys.
  6. Run python3 run.py and fill in all the information it asks for.
  7. Invite the bot with the link it provides!
  8. You can chat with the bot by mentioning it and typing your message.

Docker Setup

  1. Make sure you have Docker installed on your machine.
  2. Clone this repository to your local machine.
  3. Build the Docker image using the following command: docker build -t ai-drawing-chatbot .
  4. Run the Docker container using the following command: docker run -it --name chatbot ai-drawing-chatbot
  5. Enter in your token values. Go here to find out how to get these.

Manual Configuration

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"

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.