import os from dotenv import load_dotenv import discord import generate import neural import itemname import random import re from discord import app_commands from discord.ext import tasks load_dotenv() client = discord.Client(intents=discord.Intents.all()) tree = app_commands.CommandTree(client) queue = [] guildID = int(os.environ['GUILD_ID']) botID = int(os.environ['BOT_ID']) magicForum = int(os.environ['MAGIC_ITEM_FORUM']) raceForum = int(os.environ['RACE_FORUM']) subclassForum = int(os.environ['SUBCLASS_FORUM']) locationForum = int(os.environ['LOCATION_FORUM']) monsterForum = int(os.environ['MONSTER_FORUM']) npcForum = int(os.environ['NPC_FORUM']) otherForum = int(os.environ['OTHER_FORUM']) logChannel = int(os.environ['LOG_CHANNEL']) autogen = True if os.environ['AUTOGEN'] == 'true' else False @client.event async def on_message(message): if message.author != client.user: pass @client.event async def on_thread_create(thread): # await thread.send("## Sample Concept Art Being Generated") await neural.Generate(f"art of the '{thread.name}', a form of {thread.parent.name}",thread) @tree.command(name = "magicitem", description = "Generate a new magic item!", guild=discord.Object(id=guildID)) async def magicitem(interaction, name:str, desc: str = ""): queue.append(["magic item", name,desc + ", and Give detailed rules for item effects and flavor accordingly. Keep non-artifact, non-legendary magic items simple.", magicForum, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "race", description = "Generate a new race!", guild=discord.Object(id=guildID)) async def race(interaction, name:str, desc: str = ""): queue.append(["race", name,desc, raceForum, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "subclass", description = "Generate a new subclass!", guild=discord.Object(id=guildID)) async def subclass(interaction, name:str, desc: str = ""): queue.append(["subclass", name,desc, subclassForum, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "location", description = "Generate a new location!", guild=discord.Object(id=guildID)) async def location(interaction, name:str, desc: str = ""): queue.append(["location", name,desc+", and Describe the location in detail. Provide key locations and any necessary information on NPCs.", locationForum, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "monster", description = "Generate a new monster!", guild=discord.Object(id=guildID)) async def monster(interaction, name:str, desc: str = ""): queue.append(["living being", name,desc + ", and Give adequate description to both the flavoring of the monster AND the stat block.", monsterForum, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "npc", description = "Generate a new npc!", guild=discord.Object(id=guildID)) async def npc(interaction, name:str, desc: str = ""): queue.append(["NPC", name,desc + ". Provide a stat block for the NPC, but also include a bond, ideal, personality trait, and flaw. Offer a potential quest involving the NPC.", npcForum, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "other", description = "Generate a new npc!", guild=discord.Object(id=guildID)) async def other(interaction, name:str, type:str = "spell", desc: str = ""): queue.append([type, name,f"Be sure to follow all rules surrounding the {type}.", 1144426051720724602, interaction.user.id]) embedVar = discord.Embed(title=f'Queue position: {len(queue)}', description=f"It won't be long until we get around to the '{name}'!", color=0xffff00) await interaction.response.send_message(embed=embedVar) @tree.command(name = "ask", description = "Use in a thread to get more details!", guild=discord.Object(id=guildID)) async def ask(interaction, question:str): if interaction.channel.type == discord.ChannelType.public_thread or interaction.channel.type == discord.ChannelType.forum: embedVar = discord.Embed(title=f'Reply inbound', description=f"Question: {question}", color=0xffffff) await interaction.response.send_message(embed=embedVar) #The important code ctx = "" async for message in interaction.channel.history(limit=100,oldest_first=True): if str(message.author.id) == str(botID): ctx += message.content await interaction.channel.send(await generate.GetText(f"You are being asked a question about a piece of DND 5E content. The piece of content is: \n{ctx} \n\nThe question is: {question}. Answer the question thoroughly, but keep it less than 200 words. You do not need to restate any of the content's material nor that this is for DND 5E.")) else : embedVar = discord.Embed(title=f'Invalid location', description=f"Do this inside of a forum post for context-specific results!", color=0xff0000) await interaction.response.send_message(embed=embedVar) @tree.command(name = "conceptart", description = "Generate concept art!", guild=discord.Object(id=guildID)) async def conceptart(interaction, prompt:str): embedVar = discord.Embed(title=f'Art inbound!', description=f"Prompt: '{prompt}'", color=0xffffff) await interaction.response.send_message(embed=embedVar) await neural.Generate(f"{prompt}", interaction.channel) intervals = 0 alreadyGenerating = False async def Log(loggedData): log = client.get_channel(int(logChannel)) print(loggedData) await log.send(loggedData) async def AddAutoGen(amount): if amount > 10 or not autogen: return for i in range(amount): v = random.choice([0,1,2,3 ]) if v == 0: queue.append(["magic item", itemname.makeItem().title(),"Give detailed rules for item effects and mechanics.", magicForum, -1]) elif v == 1: queue.append(["living being", itemname.makeItem(itemname.creatures).title(),"Give adequate description to both the flavoring of the monster AND the stat block.", monsterForum, -1]) elif v == 2: queue.append(["location", itemname.makeItem(itemname.locations).title(),"Describe the location in detail. Provide key locations and any necessary information on NPCs.", locationForum, -1]) elif v == 3: queue.append(["spell", itemname.makeItem(itemname.spells).title(),"Be sure to follow all rules surrounding the spell.", otherForum, -1]) @tasks.loop(minutes=0.4) async def FlushQueue(): global alreadyGenerating global intervals try: if alreadyGenerating: return alreadyGenerating = True if len(queue) == 0: intervals = (intervals+1)%6 if intervals == 0: await AddAutoGen(1) if len(queue) > 0: obj = queue.pop(0) obj[1] = re.sub('[^A-Za-z0-9 ]+', '', obj[1]) await Log(f"**Now generating the {obj[0]} '{obj[1]}'**") channel = client.get_channel(obj[3]) st = f"Create a DND 5e {obj[0]} named the '{obj[1]}'. {obj[2]} Keep the final text under 300 words. Use markdown text formatting." c = await generate.GetText(st) if len(c) > 1900: c3 = c[0:1899] thread = await channel.create_thread(name = obj[1], content = c3 + " (Truncated)") for i in range((len(c)//1900)): f = 1899*(i+1) c2 = c[f:min(len(c)-1,f+1900)] await thread.thread.send(c2) if (obj[4]!=-1): await thread.thread.send("<@" + str(obj[4]) + ">") else: await thread.thread.send("Autogenerated by the bot") else: thread = await channel.create_thread(name = obj[1], content = c) if (obj[4]!=-1): await thread.thread.send("<@" + str(obj[4]) + ">") else: await thread.thread.send("Autogenerated by the bot") alreadyGenerating = False except Exception as e: await Log("I'm just as confused as you are, there was an error somehow! The next time it should work though.") await Log("Error message: " + str(e)) alreadyGenerating = False @client.event async def on_ready(): print("I'm in") print(client.user) await tree.sync(guild=discord.Object(id=guildID)) await Log("Bot now online") FlushQueue.start() bot_token = os.environ['BOT_TOKEN'] client.run(bot_token)