[main] syncing stage2 test changes

app.py

@@ -19,9 +19,9 @@ from falcon import try_falcon
 from falcon import continue_falcon
 
 import deepfloydif
-from deepfloydif import try_deepfloydif
-
-
+from deepfloydif import deepfloydif_stage_1
+from deepfloydif import deepfloydif_stage_2
+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)
@@ -33,17 +33,18 @@ DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
 DEEPFLOYD_CHANNEL_ID = 1121834257959092234 # 1121834257959092234 = test
 #df = Client("huggingface-projects/IF", HF_TOKEN)
 #-------------------------------------------------------------------------------------------------------------------------------------
-# 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)
+
     async def setup_hook(self):
         # This copies the global commands over to our guild
         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():
@@ -72,12 +73,22 @@ async def on_message(message):
     prompt='Enter a prompt to generate an image! Can generate realistic text, too!')
 async def deepfloydif(interaction: discord.Interaction, prompt: str):
     try:
-        await try_deepfloydif(interaction, prompt, client)
+        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)")        
 #-------------------------------------------------------------------------------------------------------------------------------------        
 # running in thread
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)