[main] some docstrings

app.py

@@ -25,16 +25,9 @@ from deepfloydif import deepfloydif_stage_2_react_check
 #-------------------------------------------------------------------------------------------------------------------------------------
 MY_GUILD = discord.Object(id=1077674588122648679)  # HF = 879548962464493619, test = 1077674588122648679
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
-
-
-
-
-
-DEEPFLOYD_CHANNEL_ID = 1121834257959092234 # 1121834257959092234 = test
-#df = Client("huggingface-projects/IF", HF_TOKEN)
 #-------------------------------------------------------------------------------------------------------------------------------------
 class MyClient(discord.Client):
-    """This structure allows commands to work instantly (instead of needing to sync global commands for up to an hour)"""
+    """This structure allows slash 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)
@@ -55,6 +48,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)
     except Exception as e:
@@ -62,9 +56,10 @@ 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=🔁  
+        await message.remove_reaction('🔁', client.user) # test=🔁  hf=🔁  
     except Exception as e:
         print(f"Error: {e}")
 #------------------------------------------------------------------------------------------------------------------------------------- 
@@ -72,10 +67,10 @@ async def on_message(message):
 @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}")
 #-------------------------------------------------------------------------------------------------------------------------------------   
@@ -85,12 +80,10 @@ async def on_reaction_add(reaction, user):
     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)")        
 #-------------------------------------------------------------------------------------------------------------------------------------        
-# running in thread
+"""This allows us to run the Discord bot in a Python thread"""
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
 def run_bot():
     client.run(DISCORD_TOKEN)