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

big ol if else to make this only work in the stable-diffusion channel

Browse Source
This commit is contained in:
Luke Robles 2022-10-12 17:38:35 -07:00
parent 650a23c2c7
commit 608202b259
1 changed files with 42 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
app/cogs/actual_utils.py
View File

@ -97,48 +97,51 @@ class ActualUtils(commands.Cog):
required=False,
),
):
import socket
import stable_diffusion
port = "7860"
ip = "192.168.1.80"
steps = 20
if ctx.channel.name == "stable-diffusion":
import socket
import stable_diffusion
# Send my requests to my gaming computer with the 3080 (if its up)
if ctx.author.id == 144986109804412928:
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"
port = "7860"
ip = "192.168.1.80"
steps = 20
try:
await ctx.defer()
original_message = await ctx.followup.send(
"Please be patient, I'm generating your image"
)
file_path = await stable_diffusion.generate_image(
ip=ip,
port=port,
positives=positive_prompt,
negatives=negative_prompt or None,
steps=steps,
)
await original_message.edit(
content="",
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,
)
# Send my requests to my gaming computer with the 3080 (if its up)
if ctx.author.id == 144986109804412928:
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 = await stable_diffusion.generate_image(
ip=ip,
port=port,
positives=positive_prompt,
negatives=negative_prompt or None,
steps=steps,
)
await original_message.edit(
content="",
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.slash_command(