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 1 year ago
parent 83cc24e6d0
commit 73390b262c
4 changed files with 293 additions and 124 deletions
Show Stats Download Patch File Download Diff File

81
cogs/commands.py
Unescape View File

@ -528,75 +528,97 @@ class Commands(discord.Cog, name="Commands"):
@discord.slash_command(
name="load",
description="Select one of your saved indexes to query from",
guild_ids=ALLOWED_GUILDS
guild_ids=ALLOWED_GUILDS,
)
@discord.guild_only()
@discord.option(name="index", description="Which file to load the index from", required=True, autocomplete=File_autocompleter.get_indexes)
async def load_index(self, ctx:discord.ApplicationContext, index: str):
@discord.option(
name="index",
description="Which file to load the index from",
required=True,
autocomplete=File_autocompleter.get_indexes,
)
async def load_index(self, ctx: discord.ApplicationContext, index: str):
await self.index_cog.load_index_command(ctx, index)
@add_to_group("index")
@discord.slash_command(
name="add",
description="Add an index to query from",
guild_ids=ALLOWED_GUILDS
name="add", description="Add an index to query from", guild_ids=ALLOWED_GUILDS
)
@discord.guild_only()
@discord.option(name="file", description="A file to create the index from", required=False, input_type=discord.SlashCommandOptionType.attachment)
@discord.option(name="link", description="A link to a file to a webpage ", required=False, input_type=str)
async def set_file(self, ctx:discord.ApplicationContext, file: discord.Attachment, link: str):
@discord.option(
name="file",
description="A file to create the index from",
required=False,
input_type=discord.SlashCommandOptionType.attachment,
)
@discord.option(
name="link",
description="A link to a file to a webpage ",
required=False,
input_type=str,
)
async def set_file(
self, ctx: discord.ApplicationContext, file: discord.Attachment, link: str
):
await self.index_cog.set_index_command(ctx, file, link)
@add_to_group("index")
@discord.slash_command(
name="reset",
description="Reset (delete) all of your saved indexes",
guild_ids=ALLOWED_GUILDS
guild_ids=ALLOWED_GUILDS,
)
@discord.guild_only()
async def reset(self, ctx:discord.ApplicationContext):
async def reset(self, ctx: discord.ApplicationContext):
await self.index_cog.reset_command(ctx)
@add_to_group("index")
@discord.slash_command(
name="compose",
description="Combine multiple indexes together",
guild_ids=ALLOWED_GUILDS
guild_ids=ALLOWED_GUILDS,
)
@discord.option(
name="name",
description="The name of the new index",
required=False,
input_type=discord.SlashCommandOptionType.string,
)
@discord.option(name="name", description="The name of the new index", required=False, input_type=discord.SlashCommandOptionType.string)
@discord.guild_only()
async def compose(self, ctx:discord.ApplicationContext, name : str):
await self.index_cog.compose_command(ctx,name)
async def compose(self, ctx: discord.ApplicationContext, name: str):
await self.index_cog.compose_command(ctx, name)
@add_to_group("index")
@discord.slash_command(
name="add_discord",
description="Set a index from a discord channel",
guild_ids=ALLOWED_GUILDS
guild_ids=ALLOWED_GUILDS,
)
@discord.guild_only()
@discord.option(name="channel", description="A channel to create the index from", required=False, input_type=discord.SlashCommandOptionType.channel)
async def set_discord(self, ctx:discord.ApplicationContext, channel: discord.TextChannel):
@discord.option(
name="channel",
description="A channel to create the index from",
required=False,
input_type=discord.SlashCommandOptionType.channel,
)
async def set_discord(
self, ctx: discord.ApplicationContext, channel: discord.TextChannel
):
await self.index_cog.set_discord_command(ctx, channel)
@add_to_group("index")
@discord.slash_command(
name="discord_backup",
description="Save an index made from the whole server",
guild_ids=ALLOWED_GUILDS
guild_ids=ALLOWED_GUILDS,
)
@discord.guild_only()
async def discord_backup(self, ctx:discord.ApplicationContext):
async def discord_backup(self, ctx: discord.ApplicationContext):
await self.index_cog.discord_backup_command(ctx)
@add_to_group("index")
@discord.slash_command(
name="query",
description="Query from your index",
guild_ids=ALLOWED_GUILDS
name="query", description="Query from your index", guild_ids=ALLOWED_GUILDS
)
@discord.guild_only()
@discord.option(name="query", description="What to query the index", required=True)
@ -606,12 +628,13 @@ class Commands(discord.Cog, name="Commands"):
guild_ids=ALLOWED_GUILDS,
required=False,
default="default",
choices=["default", "compact", "tree_summarize"]
choices=["default", "compact", "tree_summarize"],
)
async def query(self, ctx:discord.ApplicationContext, query: str, response_mode: str):
async def query(
self, ctx: discord.ApplicationContext, query: str, response_mode: str
):
await self.index_cog.query_command(ctx, query, response_mode)
#
# DALLE commands
#

55
cogs/index_service_cog.py
Unescape View File

@ -9,8 +9,10 @@ from models.index_model import Index_handler
USER_INPUT_API_KEYS = EnvService.get_user_input_api_keys()
USER_KEY_DB = EnvService.get_api_db()
class IndexService(discord.Cog, name="IndexService"):
"""Cog containing gpt-index commands"""
def __init__(
self,
bot,
@ -19,41 +21,54 @@ class IndexService(discord.Cog, name="IndexService"):
super().__init__()
self.bot = bot
self.index_handler = Index_handler(bot, usage_service)
async def set_index_command(self, ctx, file: discord.Attachment = None, link: str = None):
async def set_index_command(
self, ctx, file: discord.Attachment = None, link: str = None
):
"""Command handler to set a file as your personal index"""
if not file and not link:
await ctx.respond("Please provide a file or a link")
return
if file and link:
await ctx.respond("Please provide only one file or link. Only one or the other.")
await ctx.respond(
"Please provide only one file or link. Only one or the other."
)
return
user_api_key = None
if USER_INPUT_API_KEYS:
user_api_key = await TextService.get_user_api_key(ctx.user.id, ctx, USER_KEY_DB)
user_api_key = await TextService.get_user_api_key(
ctx.user.id, ctx, USER_KEY_DB
)
if not user_api_key:
return
await ctx.defer(ephemeral=True)
if file:
await self.index_handler.set_file_index(ctx, file, user_api_key=user_api_key)
await self.index_handler.set_file_index(
ctx, file, user_api_key=user_api_key
)
elif link:
await self.index_handler.set_link_index(ctx, link, user_api_key=user_api_key)
await self.index_handler.set_link_index(
ctx, link, user_api_key=user_api_key
)
async def set_discord_command(self, ctx, channel: discord.TextChannel = None):
"""Command handler to set a channel as your personal index"""
user_api_key = None
if USER_INPUT_API_KEYS:
user_api_key = await TextService.get_user_api_key(ctx.user.id, ctx, USER_KEY_DB)
user_api_key = await TextService.get_user_api_key(
ctx.user.id, ctx, USER_KEY_DB
)
if not user_api_key:
return
await ctx.defer(ephemeral=True)
await self.index_handler.set_discord_index(ctx, channel, user_api_key=user_api_key)
await self.index_handler.set_discord_index(
ctx, channel, user_api_key=user_api_key
)
async def reset_command(self, ctx):
await ctx.defer(ephemeral=True)
@ -62,38 +77,44 @@ class IndexService(discord.Cog, name="IndexService"):
await ctx.respond("Your indexes have been reset")
except:
traceback.print_exc()
await ctx.respond("Something went wrong while resetting your indexes. Contact the server admin.")
await ctx.respond(
"Something went wrong while resetting your indexes. Contact the server admin."
)
async def discord_backup_command(self, ctx):
"""Command handler to backup the entire server"""
user_api_key = None
if USER_INPUT_API_KEYS:
user_api_key = await TextService.get_user_api_key(ctx.user.id, ctx, USER_KEY_DB)
user_api_key = await TextService.get_user_api_key(
ctx.user.id, ctx, USER_KEY_DB
)
if not user_api_key:
return
await ctx.defer(ephemeral=True)
await self.index_handler.backup_discord(ctx, user_api_key=user_api_key)
async def load_index_command(self, ctx, index):
"""Command handler to backup the entire server"""
user_api_key = None
if USER_INPUT_API_KEYS:
user_api_key = await TextService.get_user_api_key(ctx.user.id, ctx, USER_KEY_DB)
user_api_key = await TextService.get_user_api_key(
ctx.user.id, ctx, USER_KEY_DB
)
if not user_api_key:
return
await ctx.defer(ephemeral=True)
await self.index_handler.load_index(ctx, index, user_api_key)
async def query_command(self, ctx, query, response_mode):
"""Command handler to query your index"""
user_api_key = None
if USER_INPUT_API_KEYS:
user_api_key = await TextService.get_user_api_key(ctx.user.id, ctx, USER_KEY_DB)
user_api_key = await TextService.get_user_api_key(
ctx.user.id, ctx, USER_KEY_DB
)
if not user_api_key:
return
@ -104,7 +125,9 @@ class IndexService(discord.Cog, name="IndexService"):
"""Command handler to compose from your index"""
user_api_key = None
if USER_INPUT_API_KEYS:
user_api_key = await TextService.get_user_api_key(ctx.user.id, ctx, USER_KEY_DB)
user_api_key = await TextService.get_user_api_key(
ctx.user.id, ctx, USER_KEY_DB
)
if not user_api_key:
return

6
models/autocomplete_model.py
Unescape View File

@ -155,7 +155,11 @@ class File_autocompleter:
try:
return [
file
for file in os.listdir(EnvService.find_shared_file(f"indexes/{str(ctx.interaction.user.id)}/"))
for file in os.listdir(
EnvService.find_shared_file(
f"indexes/{str(ctx.interaction.user.id)}/"
)
)
if file.startswith(ctx.value.lower())
][
:25

275
models/index_model.py
Unescape View File

@ -12,9 +12,20 @@ from datetime import date, datetime
from gpt_index.readers import YoutubeTranscriptReader
from gpt_index.readers.schema.base import Document
from gpt_index import GPTSimpleVectorIndex, SimpleDirectoryReader, QuestionAnswerPrompt, BeautifulSoupWebReader, \
GPTFaissIndex, GPTListIndex, QueryMode, GPTTreeIndex, GoogleDocsReader, MockLLMPredictor, QueryConfig, \
IndexStructType
from gpt_index import (
GPTSimpleVectorIndex,
SimpleDirectoryReader,
QuestionAnswerPrompt,
BeautifulSoupWebReader,
GPTFaissIndex,
GPTListIndex,
QueryMode,
GPTTreeIndex,
GoogleDocsReader,
MockLLMPredictor,
QueryConfig,
IndexStructType,
)
from gpt_index.readers.web import DEFAULT_WEBSITE_EXTRACTOR
from gpt_index.composability import ComposableGraph
@ -23,15 +34,19 @@ from services.environment_service import EnvService, app_root_path
SHORT_TO_LONG_CACHE = {}
def get_and_query(user_id, index_storage, query, llm_predictor):
# TODO Do prediction here for token usage
index: [GPTSimpleVectorIndex, ComposableGraph] = index_storage[user_id].get_index_or_throw()
index: [GPTSimpleVectorIndex, ComposableGraph] = index_storage[
user_id
].get_index_or_throw()
if isinstance(index, GPTTreeIndex):
response = index.query(query, verbose=True, child_branch_factor=2)
else:
response = index.query(query,verbose=True)
response = index.query(query, verbose=True)
return response
class IndexData:
def __init__(self):
self.queryable_index = None
@ -40,8 +55,11 @@ class IndexData:
# A safety check for the future
def get_index_or_throw(self):
if not self.queryable():
raise Exception("An index access was attempted before an index was created. This is a programmer error, please report this to the maintainers.")
raise Exception(
"An index access was attempted before an index was created. This is a programmer error, please report this to the maintainers."
)
return self.queryable_index
def queryable(self):
return self.queryable_index is not None
@ -58,7 +76,12 @@ class IndexData:
# Create a folder called "indexes/{USER_ID}" if it doesn't exist already
Path(f"{app_root_path()}/indexes/{user_id}").mkdir(parents=True, exist_ok=True)
# Save the index to file under the user id
index.save_to_disk(app_root_path() / "indexes" / f"{str(user_id)}"/f"{file_name}_{date.today().month}_{date.today().day}.json")
index.save_to_disk(
app_root_path()
/ "indexes"
/ f"{str(user_id)}"
/ f"{file_name}_{date.today().month}_{date.today().day}.json"
)
def reset_indexes(self, user_id):
self.individual_indexes = []
@ -74,6 +97,7 @@ class IndexData:
traceback.print_exc()
pass
class Index_handler:
def __init__(self, bot, usage_service):
self.bot = bot
@ -104,7 +128,9 @@ class Index_handler:
def index_youtube_transcript(self, link):
documents = YoutubeTranscriptReader().load_data(ytlinks=[link])
index = GPTSimpleVectorIndex(documents,)
index = GPTSimpleVectorIndex(
documents,
)
return index
def index_load_file(self, file_path) -> [GPTSimpleVectorIndex, ComposableGraph]:
@ -115,23 +141,29 @@ class Index_handler:
return index
def index_discord(self, document) -> GPTSimpleVectorIndex:
index = GPTSimpleVectorIndex(document,)
index = GPTSimpleVectorIndex(
document,
)
return index
def index_webpage(self, url) -> GPTSimpleVectorIndex:
documents = BeautifulSoupWebReader(website_extractor=DEFAULT_WEBSITE_EXTRACTOR).load_data(urls=[url])
documents = BeautifulSoupWebReader(
website_extractor=DEFAULT_WEBSITE_EXTRACTOR
).load_data(urls=[url])
index = GPTSimpleVectorIndex(documents)
return index
def reset_indexes(self, user_id):
self.index_storage[user_id].reset_indexes(user_id)
async def set_file_index(self, ctx: discord.ApplicationContext, file: discord.Attachment, user_api_key):
async def set_file_index(
self, ctx: discord.ApplicationContext, file: discord.Attachment, user_api_key
):
if not user_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_key
else:
os.environ["OPENAI_API_KEY"] = user_api_key
try:
print(file.content_type)
if file.content_type.startswith("text/plain"):
@ -152,14 +184,20 @@ class Index_handler:
suffix = ".mp3"
# Catch video files
elif file.content_type.startswith("video/"):
pass # No suffix change
pass # No suffix change
else:
await ctx.respond("Only accepts text, pdf, images, spreadheets, powerpoint, and audio/video files.")
await ctx.respond(
"Only accepts text, pdf, images, spreadheets, powerpoint, and audio/video files."
)
return
async with aiofiles.tempfile.TemporaryDirectory() as temp_path:
async with aiofiles.tempfile.NamedTemporaryFile(suffix=suffix, dir=temp_path, delete=False) as temp_file:
async with aiofiles.tempfile.NamedTemporaryFile(
suffix=suffix, dir=temp_path, delete=False
) as temp_file:
await file.save(temp_file.name)
index = await self.loop.run_in_executor(None, partial(self.index_file, temp_path))
index = await self.loop.run_in_executor(
None, partial(self.index_file, temp_path)
)
file_name = file.filename
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, file_name)
@ -168,7 +206,9 @@ class Index_handler:
await ctx.respond("Failed to set index")
traceback.print_exc()
async def set_link_index(self, ctx: discord.ApplicationContext, link: str, user_api_key):
async def set_link_index(
self, ctx: discord.ApplicationContext, link: str, user_api_key
):
if not user_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_key
else:
@ -178,12 +218,26 @@ class Index_handler:
try:
# Check if the link contains youtube in it
if "youtube" in link:
index = await self.loop.run_in_executor(None, partial(self.index_youtube_transcript, link))
index = await self.loop.run_in_executor(
None, partial(self.index_youtube_transcript, link)
)
else:
index = await self.loop.run_in_executor(None, partial(self.index_webpage, link))
index = await self.loop.run_in_executor(
None, partial(self.index_webpage, link)
)
# Make the url look nice, remove https, useless stuff, random characters
file_name = link.replace("https://", "").replace("http://", "").replace("www.", "").replace("/", "_").replace("?", "_").replace("&", "_").replace("=", "_").replace("-", "_").replace(".", "_")
file_name = (
link.replace("https://", "")
.replace("http://", "")
.replace("www.", "")
.replace("/", "_")
.replace("?", "_")
.replace("&", "_")
.replace("=", "_")
.replace("-", "_")
.replace(".", "_")
)
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, file_name)
@ -193,24 +247,31 @@ class Index_handler:
await ctx.respond("Index set")
async def set_discord_index(self, ctx: discord.ApplicationContext, channel: discord.TextChannel, user_api_key):
async def set_discord_index(
self,
ctx: discord.ApplicationContext,
channel: discord.TextChannel,
user_api_key,
):
if not user_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_key
else:
os.environ["OPENAI_API_KEY"] = user_api_key
try:
document = await self.load_data(channel_ids=[channel.id], limit=1000, oldest_first=False)
index = await self.loop.run_in_executor(None, partial(self.index_discord, document))
document = await self.load_data(
channel_ids=[channel.id], limit=1000, oldest_first=False
)
index = await self.loop.run_in_executor(
None, partial(self.index_discord, document)
)
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, channel.name)
await ctx.respond("Index set")
except Exception:
await ctx.respond("Failed to set index")
traceback.print_exc()
async def load_index(self, ctx:discord.ApplicationContext, index, user_api_key):
async def load_index(self, ctx: discord.ApplicationContext, index, user_api_key):
if not user_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_key
else:
@ -218,7 +279,9 @@ class Index_handler:
try:
index_file = EnvService.find_shared_file(f"indexes/{ctx.user.id}/{index}")
index = await self.loop.run_in_executor(None, partial(self.index_load_file, index_file))
index = await self.loop.run_in_executor(
None, partial(self.index_load_file, index_file)
)
self.index_storage[ctx.user.id].queryable_index = index
await ctx.respond("Loaded index")
except Exception as e:
@ -229,19 +292,27 @@ class Index_handler:
index_objects = []
for _index in indexes:
index_file = EnvService.find_shared_file(f"indexes/{user_id}/{_index}")
index = await self.loop.run_in_executor(None, partial(self.index_load_file, index_file))
index = await self.loop.run_in_executor(
None, partial(self.index_load_file, index_file)
)
index_objects.append(index)
# For each index object, add its documents to a GPTTreeIndex
if deep_compose:
documents = []
for _index in index_objects:
[documents.append(_index.docstore.get_document(doc_id)) for doc_id in [docmeta for docmeta in _index.docstore.docs.keys()] if isinstance(_index.docstore.get_document(doc_id), Document)]
[
documents.append(_index.docstore.get_document(doc_id))
for doc_id in [docmeta for docmeta in _index.docstore.docs.keys()]
if isinstance(_index.docstore.get_document(doc_id), Document)
]
tree_index = GPTTreeIndex(documents=documents)
# Now we have a list of tree indexes, we can compose them
if not name:
name = f"composed_deep_index_{date.today().month}_{date.today().day}.json"
name = (
f"composed_deep_index_{date.today().month}_{date.today().day}.json"
)
# Save the composed index
tree_index.save_to_disk(f"indexes/{user_id}/{name}.json")
@ -250,7 +321,11 @@ class Index_handler:
else:
documents = []
for _index in index_objects:
[documents.append(_index.docstore.get_document(doc_id)) for doc_id in [docmeta for docmeta in _index.docstore.docs.keys()] if isinstance(_index.docstore.get_document(doc_id), Document)]
[
documents.append(_index.docstore.get_document(doc_id))
for doc_id in [docmeta for docmeta in _index.docstore.docs.keys()]
if isinstance(_index.docstore.get_document(doc_id), Document)
]
# Add everything into a simple vector index
simple_index = GPTSimpleVectorIndex(documents=documents)
@ -262,53 +337,71 @@ class Index_handler:
simple_index.save_to_disk(f"indexes/{user_id}/{name}.json")
self.index_storage[user_id].queryable_index = simple_index
async def backup_discord(self, ctx: discord.ApplicationContext, user_api_key):
if not user_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_key
else:
os.environ["OPENAI_API_KEY"] = user_api_key
try:
channel_ids:List[int] = []
channel_ids: List[int] = []
for c in ctx.guild.text_channels:
channel_ids.append(c.id)
document = await self.load_data(channel_ids=channel_ids, limit=3000, oldest_first=False)
index = await self.loop.run_in_executor(None, partial(self.index_discord, document))
Path(app_root_path() / "indexes").mkdir(parents = True, exist_ok=True)
index.save_to_disk(app_root_path() / "indexes" / f"{ctx.guild.name.replace(' ', '-')}_{date.today().month}_{date.today().day}.json")
document = await self.load_data(
channel_ids=channel_ids, limit=3000, oldest_first=False
)
index = await self.loop.run_in_executor(
None, partial(self.index_discord, document)
)
Path(app_root_path() / "indexes").mkdir(parents=True, exist_ok=True)
index.save_to_disk(
app_root_path()
/ "indexes"
/ f"{ctx.guild.name.replace(' ', '-')}_{date.today().month}_{date.today().day}.json"
)
await ctx.respond("Backup saved")
except Exception:
await ctx.respond("Failed to save backup")
traceback.print_exc()
async def query(self, ctx: discord.ApplicationContext, query:str, response_mode, user_api_key):
async def query(
self, ctx: discord.ApplicationContext, query: str, response_mode, user_api_key
):
if not user_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_key
else:
os.environ["OPENAI_API_KEY"] = user_api_key
try:
llm_predictor = MockLLMPredictor(max_tokens=256)
response = await self.loop.run_in_executor(None, partial(get_and_query, ctx.user.id, self.index_storage, query, llm_predictor))
response = await self.loop.run_in_executor(
None,
partial(
get_and_query, ctx.user.id, self.index_storage, query, llm_predictor
),
)
print("The last token usage was ", llm_predictor.last_token_usage)
await self.usage_service.update_usage(llm_predictor.last_token_usage)
await ctx.respond(f"**Query:**\n\n{query.strip()}\n\n**Query response:**\n\n{response.response.strip()}")
await ctx.respond(
f"**Query:**\n\n{query.strip()}\n\n**Query response:**\n\n{response.response.strip()}"
)
except Exception:
traceback.print_exc()
await ctx.respond("Failed to send query. You may not have an index set, load an index with /index load", delete_after=10)
await ctx.respond(
"Failed to send query. You may not have an index set, load an index with /index load",
delete_after=10,
)
# Extracted functions from DiscordReader
async def read_channel(self, channel_id: int, limit: Optional[int], oldest_first: bool) -> str:
async def read_channel(
self, channel_id: int, limit: Optional[int], oldest_first: bool
) -> str:
"""Async read channel."""
messages: List[discord.Message] = []
try:
channel = self.bot.get_channel(channel_id)
print(f"Added {channel.name} from {channel.guild.name}")
@ -323,9 +416,7 @@ class Index_handler:
for thread in channel.threads:
thread_dict[thread.id] = thread
async for msg in channel.history(
limit=limit, oldest_first=oldest_first
):
async for msg in channel.history(limit=limit, oldest_first=oldest_first):
if msg.author.bot:
pass
else:
@ -340,7 +431,9 @@ class Index_handler:
print("Encountered error: " + str(e))
channel = self.bot.get_channel(channel_id)
msg_txt_list = [f"user:{m.author.display_name}, content:{m.content}" for m in messages]
msg_txt_list = [
f"user:{m.author.display_name}, content:{m.content}" for m in messages
]
return ("<|endofstatement|>\n\n".join(msg_txt_list), channel.name)
@ -369,7 +462,9 @@ class Index_handler:
f"Channel id {channel_id} must be an integer, "
f"not {type(channel_id)}."
)
(channel_content, channel_name) = await self.read_channel(channel_id, limit=limit, oldest_first=oldest_first)
(channel_content, channel_name) = await self.read_channel(
channel_id, limit=limit, oldest_first=oldest_first
)
results.append(
Document(channel_content, extra_info={"channel_name": channel_name})
)
@ -386,7 +481,11 @@ class Index_handler:
await ctx.respond("You must load at least one indexes before composing")
return
await ctx.respond("Select the index(es) to compose. You can compose multiple indexes together, you can also Deep Compose a single index.", view=ComposeModal(self, ctx.user.id, name), ephemeral=True)
await ctx.respond(
"Select the index(es) to compose. You can compose multiple indexes together, you can also Deep Compose a single index.",
view=ComposeModal(self, ctx.user.id, name),
ephemeral=True,
)
class ComposeModal(discord.ui.View):
@ -400,7 +499,9 @@ class ComposeModal(discord.ui.View):
# Get all the indexes for the user
self.indexes = [
file
for file in os.listdir(EnvService.find_shared_file(f"indexes/{str(user_id)}/"))
for file in os.listdir(
EnvService.find_shared_file(f"indexes/{str(user_id)}/")
)
]
# Map everything into the short to long cache
@ -421,7 +522,6 @@ class ComposeModal(discord.ui.View):
][0:25],
max_values=len(self.indexes) if len(self.indexes) < 25 else 25,
min_values=1,
)
# Add the select menu to the modal
self.add_item(self.index_select)
@ -430,35 +530,37 @@ class ComposeModal(discord.ui.View):
self.extra_index_selects = []
if len(self.indexes) > 25:
for i in range(25, len(self.indexes), 25):
self.extra_index_selects.append(discord.ui.Select(
placeholder="Select index(es) to compose",
options=[
discord.SelectOption(label=index[:99], value=index[:99])
for index in self.indexes
][i:i+25],
max_values=len(self.indexes[i:i+25]),
min_values=1,
))
self.extra_index_selects.append(
discord.ui.Select(
placeholder="Select index(es) to compose",
options=[
discord.SelectOption(label=index[:99], value=index[:99])
for index in self.indexes
][i : i + 25],
max_values=len(self.indexes[i : i + 25]),
min_values=1,
)
)
self.add_item(self.extra_index_selects[-1])
# Add an input field for "Deep", a "yes" or "no" option, default no
self.deep_select = discord.ui.Select(
placeholder="Deep Compose",
options=[
discord.SelectOption(label="Yes", value="yes"),
discord.SelectOption(label="No", value="no")
discord.SelectOption(label="No", value="no"),
],
max_values=1,
min_values=1
min_values=1,
)
self.add_item(self.deep_select)
# Add a button to the modal called "Compose"
self.add_item(discord.ui.Button(label="Compose", style=discord.ButtonStyle.green, custom_id="compose"))
self.add_item(
discord.ui.Button(
label="Compose", style=discord.ButtonStyle.green, custom_id="compose"
)
)
# The callback for the button
async def interaction_check(self, interaction: discord.Interaction) -> bool:
@ -467,22 +569,39 @@ class ComposeModal(discord.ui.View):
# Check that the user selected at least one index
# The total list of indexes is the union of the values of all the select menus
indexes = self.index_select.values + [select.values[0] for select in self.extra_index_selects]
indexes = self.index_select.values + [
select.values[0] for select in self.extra_index_selects
]
# Remap them from the SHORT_TO_LONG_CACHE
indexes = [SHORT_TO_LONG_CACHE[index] for index in indexes]
if len(indexes) < 1:
await interaction.response.send_message("You must select at least 1 index", ephemeral=True)
await interaction.response.send_message(
"You must select at least 1 index", ephemeral=True
)
else:
composing_message = await interaction.response.send_message("Composing indexes, this may take a long time...", ephemeral=True, delete_after=120)
composing_message = await interaction.response.send_message(
"Composing indexes, this may take a long time...",
ephemeral=True,
delete_after=120,
)
# Compose the indexes
await self.index_cog.compose_indexes(self.user_id,indexes,self.name, False if not self.deep_select.values or self.deep_select.values[0] == "no" else True)
await interaction.followup.send("Composed indexes", ephemeral=True, delete_after=10)
await self.index_cog.compose_indexes(
self.user_id,
indexes,
self.name,
False
if not self.deep_select.values or self.deep_select.values[0] == "no"
else True,
)
await interaction.followup.send(
"Composed indexes", ephemeral=True, delete_after=10
)
try:
await composing_message.delete()
except:
pass
else:
await interaction.response.defer(ephemeral=True)
await interaction.response.defer(ephemeral=True)

Write Preview
Loading…
Cancel
Save