post_to_discord.py

import math
import os
import sys
import traceback
import requests

from modules import images
import modules.scripts as scripts
import gradio as gr

from modules.processing import Processed, process_images

class Script(scripts.Script):
    def title(self):
        return "Post to Discord"

    def ui(self, is_img2img):
        n = gr.Textbox(label="n")
        token = gr.Textbox(label="discord bot token")
        channel_id = gr.Textbox(label="channel id")
        return [n, token, channel_id]

    def run(self, p, n, token, channel_id):
        for x in range(int(n)):
            p.seed = -1
            proc = process_images(p)
            image = proc.images
            files = {f"{proc.fullfn[0]}" : open(f"{proc.fullfn[0]}", "rb").read()}
            payload = {"content":f"{proc.info}"}
            header = {"authorization": f"Bot {token}"}
            requests.post(f"https://discord.com/api/v9/channels/{channel_id}/messages", data=payload, headers=header, files=files).text
        return Processed(p, image, p.seed, proc.info)