How to make a Stable Diffusion discord bot

pip install -r requirements.txt
# OR
python -m pip install -r requirements.txt

pipeline_api = PipelineCloud(token="...")

async def handle_generation(image_prompt: str) -> io.BytesIO:
    response = pipeline_api.run_pipeline(
        "pipeline_67d9d8ec36d54c148c70df1f404b0369",
        [[image_prompt], {"width": 512, "height": 512, "num_inference_steps": 50}],
    )
    image_b64 = response.result_preview[0][0][0]

    image_decoded = base64.decodebytes(image_b64.encode())
    buffer = io.BytesIO(image_decoded)
    new_uid = str(uuid.uuid4())
    buffer.name = new_uid
    return buffer

# For our '/paint' command we only want to take in a single text prompt, 
# this is included in the 'options' field below and the name of the option 
# is what's used as a key word argument to our function

@bot.command(
    name="paint",
    description="Paint an image that has never existed...",
    options=[
        interactions.Option(
            name="prompt",
            description="Image generation prompt",
            type=interactions.OptionType.STRING,
            required=True,
        ),
    ],
)
async def paint(ctx: interactions.CommandContext, prompt: str) -> None:
    # You have to send back a response quickly otherwise 
    # Discord thinks that the bot has died.
    sent_response = await ctx.send("Generating image...")

    try:
        image_buffer = await handle_generation(prompt)

        # Edit the original message sent to now include the image and the prompt
        await sent_response.edit(
            files=[
                interactions.api.models.misc.File(
                    filename=prompt + ".jpeg", fp=image_buffer
                )
            ],
            content=prompt
            # You can add another argument 'ephemeral=True' to only show the 
            # result to the user that sent the request.
        )
    except:
        # If the image generation (or anything else) fails 
        # for any reason it's best to let the user know
        await sent_response.edit(
            content="Generation failed, please try again!",
        )

        # With asyncio you have to call the 'flush=True' on print
        print(traceback.format_exc(), flush=True)
Finally we need to add a few things to run the bot and complete our script:

import os
import interactions
import traceback
import base64
import io
import uuid

from pipeline.api.asyncio import PipelineCloud

# The token here is the one we collected earlier from the discord bot
discord_token = os.getenv("DISCORD_TOKEN")
pipeline_token = os.getenv("PIPELINE_API_TOKEN")
bot = interactions.Client(token=discord_token)
pipeline_api = PipelineCloud(token=pipeline_token)

# As defined earlier
async def handle_generation(...) -> None:
    ...

# As defined earlier
@bot.command(...)
async def paint(...) -> None:
    ...

bot.start()  

env DISCORD_TOKEN=... PIPELINE_API_TOKEN=... python bot.py

project_dir/
    ./bot.py
    ./requirements.txt
    ./Dockerfile
    ./docker-compose.yml
    ./secrets.env

FROM python:3.9-slim

WORKDIR /code

COPY ./requirements.txt /code/

ENV PYTHONDONTWRITEBYTECODE=1

RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

COPY ./bot.py /code/

CMD ["python", "bot.py"]

sudo docker build . -t stable-diffusion-discord-bot:main
sudo docker run --env DISCORD_TOKEN=... --env PIPELINE_API_TOKEN=... stable-diffusion-discord-bot:main

version: "3.9"
services:
stable-diffusion-discord-bot:
    container_name: stable-diffusion-discord-bot
    image: 
    build:
      context: .
      dockerfile: ./Dockerfile
    env_file:
      - secrets.env

sudo docker-compose run --build -d