ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

move stable diffusion to its own cog

Browse Source
This commit is contained in:
Luke Robles 2022-10-18 13:18:34 -07:00
parent a63407155c
commit 74246d573d
2 changed files with 79 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
app/cogs/actual_utils.py
View File

@ -78,72 +78,6 @@ class ActualUtils(commands.Cog):
questions.open_ai(query), questions.open_ai(query),
) )
@commands.has_role("stable-diffuser")
@commands.slash_command(
guild_ids=core_utils.my_guilds,
name="sd",
description="Pass a prompt and optional negative prompt to stable diffusion",
)
@option(
"positive_prompt",
description="The positive prompt to pass to stable diffusion",
input_type="str",
requred=True,
)
@option(
"negative_prompt",
description="An optional set of negatives you want to pass to stable diffusion",
required=False,
)
async def sd(
self, ctx: commands.Context, positive_prompt: str, negative_prompt: str
):
if ctx.channel.name == "stable-diffusion":
import socket
import stable_diffusion
port = "7860"
ip = "192.168.1.80"
steps = 20
# Send my requests to my gaming computer with the 3080 (if its up)
if ctx.author.id == core_utils.my_id:
ip = "192.168.1.188"
steps = 60
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(1)
try:
s.connect((ip, int(port)))
except:
ip = "192.168.1.80"
try:
await ctx.defer()
original_message = await ctx.followup.send(
"Please be patient, I'm generating your image"
)
file_path, time_taken = await stable_diffusion.generate_image(
ip=ip,
port=port,
positives=positive_prompt,
negatives=negative_prompt or None,
steps=steps,
)
await original_message.edit(
content=time_taken,
file=discord.File(
file_path,
filename="unknown.png",
),
)
os.remove(file_path)
except Exception as e:
await ctx.reply(
"Stable diffusion isnt running right now, sorry.\n%s" % e,
)
else:
await ctx.respond("You can only do that in the stable-diffusion channel")
@commands.has_role("Track day gamers") @commands.has_role("Track day gamers")
@commands.slash_command( @commands.slash_command(
guild_ids=core_utils.my_guilds, guild_ids=core_utils.my_guilds,

79
app/cogs/stable-diffusion.py Normal file
View File

@ -0,0 +1,79 @@
from discord import option
from discord.ext import commands
import core_utils
import discord
class StableDiffusion(commands.Cog):
def __init__(self, bot):
self.bot = bot
@commands.has_role("stable-diffuser")
@commands.slash_command(
guild_ids=core_utils.my_guilds,
name="sd",
description="Pass a prompt and optional negative prompt to stable diffusion",
)
@option(
"positive_prompt",
description="The positive prompt to pass to stable diffusion",
input_type="str",
requred=True,
)
@option(
"negative_prompt",
description="An optional set of negatives you want to pass to stable diffusion",
required=False,
)
async def sd(
self, ctx: commands.Context, positive_prompt: str, negative_prompt: str
):
if ctx.channel.name == "stable-diffusion":
import socket
import stable_diffusion
port = "7860"
ip = "192.168.1.80"
steps = 20
# Send my requests to my gaming computer with the 3080 (if its up)
if ctx.author.id == core_utils.my_id:
ip = "192.168.1.188"
steps = 60
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.settimeout(1)
try:
s.connect((ip, int(port)))
except:
ip = "192.168.1.80"
try:
await ctx.defer()
original_message = await ctx.followup.send(
"Please be patient, I'm generating your image"
)
file_path, time_taken = await stable_diffusion.generate_image(
ip=ip,
port=port,
positives=positive_prompt,
negatives=negative_prompt or None,
steps=steps,
)
await original_message.edit(
content=time_taken,
file=discord.File(
file_path,
filename="unknown.png",
),
)
os.remove(file_path)
except Exception as e:
await ctx.reply(
"Stable diffusion isnt running right now, sorry.\n%s" % e,
)
else:
await ctx.respond("You can only do that in the stable-diffusion channel")
def setup(bot):
bot.add_cog(StableDiffusion(bot))