open-source-mirrors
/
GPT3Discord
mirror of https://github.com/Kav-K/GPT3Discord
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Format Python code with psf/black push

Browse Source
github-actions 2 years ago
parent 3ea5f544e8
commit 24dce18052
4 changed files with 100 additions and 61 deletions
Show Stats Download Patch File Download Diff File

7
cogs/draw_image_generation.py
Unescape View File

@ -286,7 +286,8 @@ class VaryButton(discord.ui.Button):
)
prompt = redo_users[user_id].prompt
asyncio.ensure_future(self.cog.encapsulated_send(
asyncio.ensure_future(
self.cog.encapsulated_send(
prompt,
interaction.message,
response_message=response_message,
@ -356,4 +357,6 @@ class RedoButton(discord.ui.Button["SaveView"]):
)
self.converser_cog.users_to_interactions[user_id].append(message.id)
asyncio.ensure_future(self.cog.encapsulated_send(prompt, message, response_message))
asyncio.ensure_future(
self.cog.encapsulated_send(prompt, message, response_message)
)

50
cogs/gpt_3_commands_and_converser.py
Unescape View File

@ -444,7 +444,9 @@ class GPT3ComCon(commands.Cog, name="GPT3ComCon"):
# Paginate and send the response back to the users
if not response_message:
if len(response_text) > self.TEXT_CUTOFF:
await self.paginate_and_send(response_text, message) # No paginations for multi-messages.
await self.paginate_and_send(
response_text, message
) # No paginations for multi-messages.
else:
response_message = await message.reply(
response_text.replace("<|endofstatement|>", ""),
@ -453,8 +455,12 @@ class GPT3ComCon(commands.Cog, name="GPT3ComCon"):
self.redo_users[message.author.id] = RedoUser(
prompt, message, response_message
)
self.redo_users[message.author.id].add_interaction(response_message.id)
print(f"Added the interaction {response_message.id} to the redo user {message.author.id}")
self.redo_users[message.author.id].add_interaction(
response_message.id
)
print(
f"Added the interaction {response_message.id} to the redo user {message.author.id}"
)
original_message[message.author.id] = message.id
else:
# We have response_text available, this is the original message that we want to edit
@ -661,12 +667,15 @@ class GPT3ComCon(commands.Cog, name="GPT3ComCon"):
# This is because encapsulated_send is a coroutine, and we need to await it to get the response from the model.
# We can't await it in the main thread, so we need to create a new thread to run it in.
# We can make sure that when the thread executes it executes in an async fashion by
asyncio.run_coroutine_threadsafe(self.encapsulated_send(
asyncio.run_coroutine_threadsafe(
self.encapsulated_send(
message,
prompt
if message.author.id not in self.conversating_users
else "".join(self.conversating_users[message.author.id].history),
), asyncio.get_running_loop())
),
asyncio.get_running_loop(),
)
class RedoView(discord.ui.View):
@ -696,18 +705,27 @@ class EndConvoButton(discord.ui.Button["RedoView"]):
# Get the user
user_id = interaction.user.id
if user_id in self.converser_cog.redo_users and self.converser_cog.redo_users[user_id].in_interaction(interaction.message.id):
if user_id in self.converser_cog.redo_users and self.converser_cog.redo_users[
user_id
].in_interaction(interaction.message.id):
try:
await self.converser_cog.end_conversation(self.converser_cog.redo_users[user_id].message)
await interaction.response.send_message("Your conversation has ended!", ephemeral=True,
delete_after=10)
await self.converser_cog.end_conversation(
self.converser_cog.redo_users[user_id].message
)
await interaction.response.send_message(
"Your conversation has ended!", ephemeral=True, delete_after=10
)
except Exception as e:
print(e)
traceback.print_exc()
await interaction.response.send_message(e, ephemeral=True, delete_after=30)
await interaction.response.send_message(
e, ephemeral=True, delete_after=30
)
pass
else:
await interaction.response.send_message("This is not your conversation to end!", ephemeral=True, delete_after=10)
await interaction.response.send_message(
"This is not your conversation to end!", ephemeral=True, delete_after=10
)
class RedoButton(discord.ui.Button["RedoView"]):
@ -719,7 +737,9 @@ class RedoButton(discord.ui.Button["RedoView"]):
# Get the user
user_id = interaction.user.id
if user_id in self.converser_cog.redo_users and self.converser_cog.redo_users[user_id].in_interaction(interaction.message.id):
if user_id in self.converser_cog.redo_users and self.converser_cog.redo_users[
user_id
].in_interaction(interaction.message.id):
# Get the message and the prompt and call encapsulated_send
message = self.converser_cog.redo_users[user_id].message
prompt = self.converser_cog.redo_users[user_id].prompt
@ -733,4 +753,8 @@ class RedoButton(discord.ui.Button["RedoView"]):
message, prompt, response_message
)
else:
await interaction.response.send_message("You can only redo the most recent prompt that you sent yourself.", ephemeral=True, delete_after=10)
await interaction.response.send_message(
"You can only redo the most recent prompt that you sent yourself.",
ephemeral=True,
delete_after=10,
)

20
cogs/image_prompt_optimizer.py
Unescape View File

@ -96,8 +96,12 @@ class ImgPromptOptimizer(commands.Cog, name="ImgPromptOptimizer"):
response_message.id
)
self.converser_cog.redo_users[ctx.author.id] = RedoUser(prompt, ctx.message, response_message)
self.converser_cog.redo_users[ctx.author.id].add_interaction(response_message.id)
self.converser_cog.redo_users[ctx.author.id] = RedoUser(
prompt, ctx.message, response_message
)
self.converser_cog.redo_users[ctx.author.id].add_interaction(
response_message.id
)
await response_message.edit(
view=OptimizeView(
self.converser_cog, self.image_service_cog, self.deletion_queue
@ -140,7 +144,10 @@ class DrawButton(discord.ui.Button["OptimizeView"]):
user_id = interaction.user.id
interaction_id = interaction.message.id
if interaction_id not in self.converser_cog.users_to_interactions[user_id] or interaction_id not in self.converser_cog.redo_users[user_id].interactions:
if (
interaction_id not in self.converser_cog.users_to_interactions[user_id]
or interaction_id not in self.converser_cog.redo_users[user_id].interactions
):
await interaction.response.send_message(
content="You can only draw for prompts that you generated yourself!",
ephemeral=True,
@ -184,7 +191,9 @@ class RedoButton(discord.ui.Button["OptimizeView"]):
# Get the user
user_id = interaction.user.id
if user_id in self.converser_cog.redo_users and self.converser_cog.redo_users[user_id].in_interaction(interaction_id):
if user_id in self.converser_cog.redo_users and self.converser_cog.redo_users[
user_id
].in_interaction(interaction_id):
# Get the message and the prompt and call encapsulated_send
message = self.converser_cog.redo_users[user_id].message
prompt = self.converser_cog.redo_users[user_id].prompt
@ -198,5 +207,6 @@ class RedoButton(discord.ui.Button["OptimizeView"]):
else:
await interaction.response.send_message(
content="You can only redo for prompts that you generated yourself!",
ephemeral=True, delete_after=10
ephemeral=True,
delete_after=10,
)

2
models/user_model.py
Unescape View File

@ -3,6 +3,7 @@ Store information about a discord user, for the purposes of enabling conversatio
history, message count, and the id of the user in order to track them.
"""
class RedoUser:
def __init__(self, prompt, message, response):
self.prompt = prompt
@ -27,6 +28,7 @@ class RedoUser:
def __repr__(self):
return f"RedoUser({self.message.author.id})"
class User:
def __init__(self, id):
self.id = id

Write Preview
Loading…
Cancel
Save