🎨 PicAIsso - Bring AI art to your life

PicAIsso is an open-source StableDiffusion implementation for generating AI art using an API and a Discord Bot.

👉 Imagine self-hosting your own MidJourney Discord bot, but with a different name and a different art style.

---

You can self-host the API and the Discord bot on your own machine or on a server.

OR

You can use my own self-hosted API and Discord bot. Add the bot to your Discord server and start generating art! 🎉

Click here to invite the bot to your server. Note: The bot is currently hosted on my own workstation, so it might be offline sometimes. > Note 2: I'm saving all the generated images on my S3 bucket, these images will be used to create a dataset available for free on Hugging Face.

For self-hosting instructions, keep reading the README. 👇

---

Some examples of generated images:

---

Requirements

• Docker installed
• NVIDIA GPU with at least 12GB of VRAM (with a batch size of 1)
• NVIDIA drivers installed + NVIDIA Container Toolkit installed

Contributing

If you want to contribute to the project, please read the CONTRIBUTING.md file. It will help you to setup your development environment.

Installation

1. Clone the repository

git clone https://github.com/chainyo/picaisso.git

2. Create your own .env file for the API and update the values with your own.

   Follow the .env.template comments.

3. Create your own .env file for the Discord Bot and update the values with your own.

   Follow the .env.template comments. The Discord bot application installation process is explained if you follow the link in the .env.template file.

4. (Optional) If you want to store the generated images, there is a S3 bucket configuration in the .env file. You can create your own S3 bucket and update the values with your own. Leave the values empty if you don't want to use S3.

Deployment

There are two steps:

• Deploy the API
• Deploy the Discord Bot

Warning: Be sure to be in the root folder of the project before running the Docker commands.

Create the Docker network

docker network create picaisso

Deploy the API

1. Build the Docker image

docker build -t picaisso-api:latest -f docker/api/Dockerfile .

You should see the built image in your Docker images list, you can check it with the following command:

docker images

2. Run the Docker container

docker run -d \
  --gpus all \
  --network picaisso \
  -p 7681:7681 \
  -v ${HOME}/.cache:/root/.cache \
  --restart unless-stopped \
  --name picaisso-api \
  picaisso-api:latest

3. Test the API

The API should be running on port 7681 of your machine. It can take a few minutes to start, because the model has to be downloaded and loaded on the first run. Because we are using the ${HOME}/.cache folder as a mounted volume, the model will be downloaded only once and will be reused on the next runs.

Once the API is running, you can test it by going to http://localhost:7681/ in your web browser.

You should see the landing page of the API.

Click on the Docs button to see the API documentation and test the /generate endpoint. Check the Usage section for more details.

Trouble shooting: use the docker logs picaisso-api command to see the logs of the container.

Deploy the Discord Bot

1. Build the Docker image

docker build -t picaisso-bot:latest -f docker/bot/Dockerfile .

2. Run the Docker container

docker run -d \
  --restart unless-stopped \
  --network picaisso \
  --name picaisso-bot \
  picaisso-bot:latest

If you successfully created the Discord bot application, you just need to invite it to your server. You can find the invite link in the OAuth2 section of your Discord bot application. (See the tutorial link in the .env.template file.)

Note: Server Members Intent must be enabled on the Discord bot (under the Bot section):

Mine is live! 🎉

Usage

You can use the project in three different ways:

• Use the API documentation to generate images (Starter apprentice level 🤓, be better than that)
• Use the API endpoint to generate images (Dev guru level 🧙‍♂️)
• Use the Discord bot to generate images (Discord bro level 🤙)

API documentation

For testing purposes, you can use the API documentation to generate images and validate everything is working as expected. You can find the API documentation at http://localhost:7681/docs when the API is running.

1. Click on the Authorize button and enter your API key in the Value field.

2. Use the generate endpoint to generate an image.

2.1. Click on the Try it out button.

2.2. Enter the parameters you want to use to generate the image.

• prompt: any text you want to use to generate the image.

• author: the name of the author of the image. Used for s3 storage only, you can let string.

  2.3. Enter the Execute button and wait for the image to be generated.

You can download the image by right-clicking on the image and selecting Save image as....

API endpoint

You can use the API endpoint to generate images in your own application.

You can use the same logic used by the Discord bot to generate images. You can find the code in the bot.py file.

Discord bot

You can use the Discord bot to generate images in your Discord server.

1. Start by typing /art in any channel of your Discord server.

2. The bot will ask you to enter the prompt you want to use to generate the image.

3. Enter the prompt you want to use to generate the image.

4. The bot will tell you the job is in progress and will send you the image when it's done.

Enjoy the result! 🎉

---

If you have any questions or issues, feel free to open an issue or contact me on Twitter, Discord (ChainYo#3610) or via email ([email protected]).

If you like the project, please consider giving it a ⭐️ and share it with your friends.

---

Buy me a book 📚 if you want to support my work.