/discord-chatbot

Primary LanguagePythonMIT LicenseMIT

Notice: This is a fork. Get the most up-to-date version upstream: https://github.com/mak448a/AI-Drawing-Chatbot

AI Drawing ChatBot

This is a completely free AI image generator Discord bot written in Python. It has a chatbot that you can ask to draw pictures which uses Llama 2!

This project uses various APIs, which allows it to run on less powerful computers.

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

  • gpt4free is the recommended model.
  • GPT4All isn't very "smart"
  • When using SDXL-Beta it 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 will not work as expected if multiple users are chatting with it at once.
  • When hosting on a low-end server, GPT4All cannot be used.

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 ChatGPT key, which is under accessToken from here: https://chat.openai.com/api/auth/session.
  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 ChatGPT key, which is under accessToken from here: https://chat.openai.com/api/auth/session.
  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

Configuring chat model

Go to config.json and set the key "model" to the model you want.

Available Models

  • gpt4free
  • GPT4All

You must write the model exactly as it is written here.

When you are done, the edited line should look like this:

"model": "gpt4free",
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.