[main] bringing back dfif stuff

app.py

@@ -30,8 +30,8 @@ DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
 
 
 #-------------------------------------------------------------------------------------------------------------------------------------
-# This structure allows commands to work instantly (instead of needing to sync global commands for up to an hour)
 class MyClient(discord.Client):
+	"""This structure allows commands to work instantly (instead of needing to sync global commands for up to an hour)"""    
     def __init__(self, *, intents: discord.Intents):
         super().__init__(intents=intents)
         self.tree = app_commands.CommandTree(self)
@@ -41,8 +41,7 @@ class MyClient(discord.Client):
         self.tree.copy_global_to(guild=MY_GUILD)
         await self.tree.sync(guild=MY_GUILD)
 
-intents = discord.Intents.all() 
-client = MyClient(intents=intents)
+client = MyClient(intents=discord.Intents.default())
 #-------------------------------------------------------------------------------------------------------------------------------------
 @client.event
 async def on_ready():
@@ -53,6 +52,7 @@ async def on_ready():
 @app_commands.describe(
     prompt='Enter some text to chat with the bot! Like this: /falcon Hello, how are you?')
 async def falcon(interaction: discord.Interaction, prompt: str):
+    """Command that begins a new conversation with Falcon"""
     try:  
         await try_falcon(interaction, prompt)
 
@@ -61,6 +61,7 @@ async def falcon(interaction: discord.Interaction, prompt: str):
 #-------------------------------------------------------------------------------------------------------------------------------------
 @client.event
 async def on_message(message):
+    """Checks channel and continues Falcon conversation if it's the right Discord Thread"""
     try:
         await continue_falcon(message)
         await message.remove_reaction('🔁', client.user) # test=🔁  hf=🔁  
@@ -68,7 +69,26 @@ async def on_message(message):
     except Exception as e:
         print(f"Error: {e}")
 #------------------------------------------------------------------------------------------------------------------------------------- 
-# running in thread
+@client.tree.command()
+@app_commands.describe(
+    prompt='Enter a prompt to generate an image! Can generate realistic text, too!')
+async def deepfloydif(interaction: discord.Interaction, prompt: str):
+    """DeepfloydIF stage 1 generation"""
+    try:
+        await deepfloydif_stage_1(interaction, prompt, client)
+        
+    except Exception as e:
+        print(f"Error: {e}")
+#-------------------------------------------------------------------------------------------------------------------------------------   
+@client.event
+async def on_reaction_add(reaction, user): 
+    """Checks for a reaction in order to call dfif2"""
+    try:
+        await deepfloydif_stage_2_react_check(reaction, user)
+        
+    except Exception as e:
+        print(f"Error: {e} (known error, does not cause issues, low priority)")        
+#-------------------------------------------------------------------------------------------------------------------------------------        
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
 
 def run_bot():